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
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
if len(modlist):
logging.debug("Modifying existing LDAP entry %s" % self.dn)
- ldap_connection.modify_s(self.dn, modlist)
+ ldapdb.connection.modify_s(self.dn, modlist)
else:
logging.debug("No changes to be saved to LDAP entry %s" % self.dn)