X-Git-Url: https://git.stderr.nl/gitweb?p=matthijs%2Fupstream%2Fdjango-ldapdb.git;a=blobdiff_plain;f=ldapdb%2Fmodels%2Ffields.py;fp=ldapdb%2Fmodels%2Ffields.py;h=5e85062407847b2a7735ec0b83eece3c38c7a427;hp=921923c052469ac826248bde9743b57d48c66ed6;hb=089876639ebce60a85a285d7ab076797fa96c20c;hpb=6baded55b00e883c2612c013b884e8135baeaebe diff --git a/ldapdb/models/fields.py b/ldapdb/models/fields.py index 921923c..5e85062 100644 --- a/ldapdb/models/fields.py +++ b/ldapdb/models/fields.py @@ -59,19 +59,35 @@ class CharField(fields.CharField): class ImageField(fields.Field): def get_db_prep_lookup(self, lookup_type, value): + "Returns field's value prepared for database lookup." + return [self.get_prep_lookup(lookup_type, value)] + + def get_prep_lookup(self, lookup_type, value): + "Perform preliminary non-db specific lookup checks and conversions" raise TypeError("ImageField has invalid lookup: %s" % lookup_type) class IntegerField(fields.IntegerField): def get_db_prep_lookup(self, lookup_type, value): - if lookup_type in ('exact', 'gte', 'lte'): - return [value] + "Returns field's value prepared for database lookup." + return [self.get_prep_lookup(lookup_type, value)] + def get_prep_lookup(self, lookup_type, value): + "Perform preliminary non-db specific lookup checks and conversions" + if lookup_type in ('exact', 'gte', 'lte'): + return value raise TypeError("IntegerField has invalid lookup: %s" % lookup_type) class ListField(fields.Field): __metaclass__ = SubfieldBase def get_db_prep_lookup(self, lookup_type, value): + "Returns field's value prepared for database lookup." + return [self.get_prep_lookup(lookup_type, value)] + + def get_prep_lookup(self, lookup_type, value): + "Perform preliminary non-db specific lookup checks and conversions" + if lookup_type == 'contains': + return escape_ldap_filter(value) raise TypeError("ListField has invalid lookup: %s" % lookup_type) def to_python(self, value):