1 # -*- coding: utf-8 -*-
4 # Copyright (C) 2009-2010 Bolloré telecom
5 # See AUTHORS file for a full list of contributors.
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 from django.db.models import fields, SubfieldBase
23 from ldapdb import escape_ldap_filter
25 class CharField(fields.CharField):
26 def __init__(self, *args, **kwargs):
27 kwargs['max_length'] = 200
28 super(CharField, self).__init__(*args, **kwargs)
30 def get_db_prep_lookup(self, lookup_type, value):
31 if lookup_type == 'endswith':
32 return ["*%s" % escape_ldap_filter(value)]
33 elif lookup_type == 'startswith':
34 return ["%s*" % escape_ldap_filter(value)]
35 elif lookup_type == 'contains':
36 return ["*%s*" % escape_ldap_filter(value)]
37 elif lookup_type == 'exact':
38 return [escape_ldap_filter(value)]
39 elif lookup_type == 'in':
40 return [escape_ldap_filter(v) for v in value]
42 raise TypeError("CharField has invalid lookup: %s" % lookup_type)
44 def get_prep_lookup(self, lookup_type, value):
45 return escape_ldap_filter(value)
47 class ImageField(fields.Field):
50 class IntegerField(fields.IntegerField):
51 def get_db_prep_lookup(self, lookup_type, value):
52 if lookup_type == 'exact':
55 raise TypeError("IntegerField has invalid lookup: %s" % lookup_type)
57 class ListField(fields.Field):
58 __metaclass__ = SubfieldBase
60 def to_python(self, value):