X-Git-Url: https://git.stderr.nl/gitweb?p=matthijs%2Fupstream%2Fdjango-ldapdb.git;a=blobdiff_plain;f=ldapdb%2Fmodels%2Fbase.py;h=41d00e8c684b3fa7483973a5478b888808cce290;hp=cdc44e21fc4e32869dc0c49043951a88f130f102;hb=c15d0f62f8b92e9a2c69d9b93ded4cf964ef710c;hpb=81037a75f672596982475a7081946bfd9b4fd286 diff --git a/ldapdb/models/base.py b/ldapdb/models/base.py index cdc44e2..41d00e8 100644 --- a/ldapdb/models/base.py +++ b/ldapdb/models/base.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- # # django-ldapdb -# Copyright (C) 2009 Bolloré telecom +# Copyright (C) 2009-2010 Bolloré telecom # See AUTHORS file for a full list of contributors. # # This program is free software: you can redistribute it and/or modify @@ -18,8 +18,6 @@ # along with this program. If not, see . # -# -*- coding: utf-8 -*- - import ldap import logging @@ -34,20 +32,15 @@ class ModelBase(django.db.models.base.ModelBase): Metaclass for all LDAP models. """ def __new__(cls, name, bases, attrs): - attr_meta = attrs.pop('Ldap', None) - super_new = super(ModelBase, cls).__new__ new_class = super_new(cls, name, bases, attrs) # patch manager to use our own QuerySet class - def get_query_set(): - return QuerySet(new_class) - new_class.objects.get_query_set = get_query_set - new_class._default_manager.get_query_set = get_query_set - - if attr_meta: - new_class._meta.dn = attr_meta.dn - new_class._meta.object_classes = attr_meta.object_classes + if not new_class._meta.abstract: + def get_query_set(): + return QuerySet(new_class) + new_class.objects.get_query_set = get_query_set + new_class._default_manager.get_query_set = get_query_set return new_class @@ -67,6 +60,12 @@ class Model(django.db.models.base.Model): super(Model, self).__init__(*args, **kwargs) self.saved_pk = self.pk + def _collect_sub_objects(self, collector): + """ + This private API seems to be called by the admin interface in django 1.2 + """ + pass + def build_rdn(self): """ Build the Relative Distinguished Name for this entry. @@ -106,7 +105,7 @@ class Model(django.db.models.base.Model): continue value = getattr(self, field.name) if value: - entry.append((field.db_column, value)) + entry.append((field.db_column, field.get_db_prep_save(value, connection=ldapdb.connection))) logging.debug("Creating new LDAP entry %s" % new_dn) ldapdb.connection.add_s(new_dn, entry) @@ -126,7 +125,7 @@ class Model(django.db.models.base.Model): new_value = getattr(self, field.name, None) if old_value != new_value: if new_value: - modlist.append((ldap.MOD_REPLACE, field.db_column, new_value)) + modlist.append((ldap.MOD_REPLACE, field.db_column, field.get_db_prep_save(new_value, connection=ldapdb.connection))) elif old_value: modlist.append((ldap.MOD_DELETE, field.db_column, None)) @@ -154,7 +153,8 @@ class Model(django.db.models.base.Model): """ import new import re - name = "%s_%s" % (base_class.__name__, re.sub('[=,]', '_', base_dn)) + suffix = re.sub('[=,]', '_', base_dn) + name = "%s_%s" % (base_class.__name__, str(suffix)) new_class = new.classobj(name, (base_class,), {'base_dn': base_dn, '__module__': base_class.__module__}) return new_class