diff --git a/ldap3_mapper_new/attribute.py b/ldap3_mapper_new/attribute.py index 6b8376e4408a8cc6e781b08be050ed309f5822c3..9e984025d08a543e01e55a09c2648d49e1c65a20 100644 --- a/ldap3_mapper_new/attribute.py +++ b/ldap3_mapper_new/attribute.py @@ -38,10 +38,19 @@ class AttributeList(MutableSequence): self.__set(tmp) class Attribute: - def __init__(self, name, aliases=None, multi=False): + def __init__(self, name, aliases=None, multi=False, default=None): self.name = name self.aliases = aliases or [] self.multi = multi + self.default = default + + def add_hook(self, obj): + if obj.ldap_object.getattr(self.name) == []: + self.__set__(self.name, self.default() if callable(self.default) else self.default) + + def __set_name__(self, cls, name): + if self.default is not None: + cls.ldap_add_hooks = cls.ldap_add_hooks + (self.add_hook,) def __get__(self, obj, objtype=None): if obj is None: diff --git a/ldap3_mapper_new/dbutils.py b/ldap3_mapper_new/dbutils.py index 9dee2179480c39b7570e37718464923baefc4b3f..21a62321c5fa3b87a492d73b487fca7f2ab35f98 100644 --- a/ldap3_mapper_new/dbutils.py +++ b/ldap3_mapper_new/dbutils.py @@ -1,5 +1,7 @@ from collections.abc import MutableSet +from .model import add_to_session + class DBRelationshipSet(MutableSet): def __init__(self, dbobj, relattr, ldapcls): self.__dbobj = dbobj @@ -27,7 +29,7 @@ class DBRelationshipSet(MutableSet): if not isinstance(value, self.__ldapcls): raise TypeError() if value.ldap_object.session is not None: - self.__ldapcls.ldap_mapper.session.add(value) + add_to_session(value, self.__ldapcls.ldap_mapper.session) if value.ldap_object.dn not in self.__get_dns(): getattr(self.__dbobj, self.__relattr).append(self.__ldapcls(dn=value.ldap_object.dn)) diff --git a/ldap3_mapper_new/model.py b/ldap3_mapper_new/model.py index 7f65466d15ac4c3a91196456d0569b845d35af46..4a0c20d737614050a888275949bd5694f1e88380 100644 --- a/ldap3_mapper_new/model.py +++ b/ldap3_mapper_new/model.py @@ -1,4 +1,3 @@ - try: # Added in v2.5 from ldap3.utils.dn import escape_rdn @@ -14,14 +13,19 @@ except ImportError: rdn = ''.join((rdn[:-1], '\\ ')) return rdn -from . import base +from . import core + +def add_to_session(obj, session): + for func in obj.ldap_add_hooks: + func(obj) + session.add(obj.ldap_object, obj.dn, obj.ldap_object_classes) class Session: def __init__(self, get_connection): - self.ldap_session = base.Session(get_connection) + self.ldap_session = core.Session(get_connection) def add(self, obj): - self.ldap_session.add(obj.ldap_object, obj.dn, obj.ldap_object_classes) + add_to_session(obj, self.ldap_session) def delete(self, obj): self.ldap_session.delete(obj.ldap_object) @@ -76,6 +80,8 @@ class ModelQueryWrapper: class Model: # Overwritten by mapper ldap_mapper = None + query = ModelQueryWrapper() + ldap_add_hooks = tuple() # Overwritten by models ldap_search_base = None @@ -84,10 +90,8 @@ class Model: ldap_dn_base = None ldap_dn_attribute = None - query = ModelQueryWrapper() - def __init__(self, **kwargs): - self.ldap_object = base.Object() + self.ldap_object = core.Object() for key, value, in kwargs.items(): if not hasattr(self, key): raise Exception() diff --git a/ldap3_mapper_new/relationship.py b/ldap3_mapper_new/relationship.py index 486d7276a3ea4307200097a8cbfc1a65343e4362..0698626689bbba0bd7f6ead626b44241a3f77a8a 100644 --- a/ldap3_mapper_new/relationship.py +++ b/ldap3_mapper_new/relationship.py @@ -1,6 +1,6 @@ from collections.abc import MutableSet -from .model import make_modelobj, make_modelobjs +from .model import make_modelobj, make_modelobjs, add_to_session class UnboundObjectError(Exception): pass @@ -38,7 +38,7 @@ class RelationshipSet(MutableSet): def add(self, value): self.__modify_check(value) if value.ldap_object.session is None: - self.__ldap_object.session.add(value.ldap_object) + add_to_session(value, self.__ldap_object.session) assert value.ldap_object.session == self.__ldap_object.session self.__ldap_object.attradd(self.__name, value.dn) @@ -102,7 +102,7 @@ class BackreferenceSet(MutableSet): def add(self, value): self.__modify_check(value) if value.ldap_object.session is None: - self.__ldap_object.session.add(value.ldap_object) + add_to_session(value, self.__ldap_object.session) assert value.ldap_object.session == self.__ldap_object.session if self.__ldap_object.dn not in value.ldap_object.getattr(self.__name): value.ldap_object.attradd(self.__name, self.__ldap_object.dn)