import django.db.models
-from granadilla.db import connection as ldap_connection
-from granadilla.db.query import QuerySet
+import ldapdb
+from ldapdb.models.query import QuerySet
class ModelBase(django.db.models.base.ModelBase):
"""
Metaclass for all LDAP models.
"""
def __new__(cls, name, bases, attrs):
- attr_meta = attrs.get('Meta', None)
- if attr_meta:
- dn = attr_meta._dn
- object_classes = attr_meta._object_classes
+ attr_meta = attrs.pop('Ldap', None)
super_new = super(ModelBase, cls).__new__
new_class = super_new(cls, name, bases, attrs)
new_class._default_manager.get_query_set = get_query_set
if attr_meta:
- new_class._meta.dn = dn
- new_class._meta.object_classes = attr_meta._object_classes
+ new_class._meta.dn = attr_meta.dn
+ new_class._meta.object_classes = attr_meta.object_classes
return new_class
"""
__metaclass__ = ModelBase
- def __init__(self, dn=None, *args, **kwargs):
- self.dn = dn
+ dn = django.db.models.fields.CharField(max_length=200)
+
+ def __init__(self, *args, **kwargs):
super(Model, self).__init__(*args, **kwargs)
+ self.saved_pk = self.pk
- def build_dn(self):
+ def build_rdn(self):
"""
- Build the Distinguished Name for this entry.
+ Build the Relative Distinguished Name for this entry.
"""
+ bits = []
for field in self._meta.local_fields:
if field.primary_key:
- return "%s=%s,%s" % (field.db_column, getattr(self, field.name), self._meta.dn)
+ bits.append("%s=%s" % (field.db_column, getattr(self, field.name)))
+ if not len(bits):
+ raise Exception("Could not build Distinguished Name")
+ return '+'.join(bits)
+
+ def build_dn(self):
+ """
+ Build the Distinguished Name for this entry.
+ """
+ return "%s,%s" % (self.build_rdn(), self._meta.dn)
raise Exception("Could not build Distinguished Name")
def delete(self):
Delete this entry.
"""
logging.debug("Deleting LDAP entry %s" % self.dn)
- ldap_connection.delete_s(self.dn)
+ ldapdb.connection.delete_s(self.dn)
def save(self):
# create a new entry
entry.append((field.db_column, value))
logging.debug("Creating new LDAP entry %s" % new_dn)
- ldap_connection.add_s(new_dn, entry)
+ ldapdb.connection.add_s(new_dn, entry)
# update object
self.dn = new_dn
+ self.saved_pk = self.pk
return
# update an existing entry
modlist = []
- orig = self.__class__.objects.get(pk=self.pk)
+ orig = self.__class__.objects.get(pk=self.saved_pk)
for field in self._meta.local_fields:
if not field.db_column:
continue
elif old_value:
modlist.append((ldap.MOD_DELETE, field.db_column, None))
- if len(modlist):
- logging.debug("Modifying existing LDAP entry %s" % self.dn)
- ldap_connection.modify_s(self.dn, modlist)
- else:
+ if not len(modlist):
logging.debug("No changes to be saved to LDAP entry %s" % self.dn)
+ return
+
+ # handle renaming
+ new_dn = self.build_dn()
+ if new_dn != self.dn:
+ logging.debug("Renaming LDAP entry %s to %s" % (self.dn, new_dn))
+ ldapdb.connection.rename_s(self.dn, self.build_rdn())
+ self.dn = new_dn
+
+ logging.debug("Modifying existing LDAP entry %s" % self.dn)
+ ldapdb.connection.modify_s(self.dn, modlist)
+ self.saved_pk = self.pk