# -*- 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
from django.db.models import fields, SubfieldBase
+from ldapdb import escape_ldap_filter
+
class CharField(fields.CharField):
def __init__(self, *args, **kwargs):
kwargs['max_length'] = 200
super(CharField, self).__init__(*args, **kwargs)
+ def from_ldap(self, value, connection):
+ if len(value) == 0:
+ return ''
+ else:
+ return value[0].decode(connection.charset)
+
+ def get_db_prep_lookup(self, lookup_type, value):
+ "Returns field's value prepared for database lookup."
+ if lookup_type == 'endswith':
+ return ["*%s" % escape_ldap_filter(value)]
+ elif lookup_type == 'startswith':
+ return ["%s*" % escape_ldap_filter(value)]
+ elif lookup_type in ['contains', 'icontains']:
+ return ["*%s*" % escape_ldap_filter(value)]
+ elif lookup_type == 'exact':
+ return [escape_ldap_filter(value)]
+ elif lookup_type == 'in':
+ return [escape_ldap_filter(v) for v in value]
+
+ raise TypeError("CharField has invalid lookup: %s" % lookup_type)
+
+ def get_db_prep_save(self, value, connection):
+ return [value.encode(connection.charset)]
+
+ def get_prep_lookup(self, lookup_type, value):
+ "Perform preliminary non-db specific lookup checks and conversions"
+ if lookup_type == 'endswith':
+ return "*%s" % escape_ldap_filter(value)
+ elif lookup_type == 'startswith':
+ return "%s*" % escape_ldap_filter(value)
+ elif lookup_type in ['contains', 'icontains']:
+ return "*%s*" % escape_ldap_filter(value)
+ elif lookup_type == 'exact':
+ return escape_ldap_filter(value)
+ elif lookup_type == 'in':
+ return [escape_ldap_filter(v) for v in value]
+
+ raise TypeError("CharField has invalid lookup: %s" % lookup_type)
+
class ImageField(fields.Field):
- pass
+ def from_ldap(self, value, connection):
+ if len(value) == 0:
+ return ''
+ else:
+ return value[0]
+
+ 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_db_prep_save(self, value, connection):
+ return [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):
- pass
+ def from_ldap(self, value, connection):
+ if len(value) == 0:
+ return 0
+ else:
+ return int(value[0])
+
+ 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_db_prep_save(self, value, connection):
+ return [str(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 from_ldap(self, value, connection):
+ return value
+
+ 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_db_prep_save(self, value, connection):
+ return [x.encode(connection.charset) for x in 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):
if not value:
return []