ldapdb/models/fields.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # django-ldapdb
   4 # Copyright (C) 2009-2010 Bolloré telecom
   5 # See AUTHORS file for a full list of contributors.
   6 #
   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.
  11 #
  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.
  16 #
  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/>.
  19 #
  20
  21 from django.db.models import fields, SubfieldBase
  22
  23 from ldapdb import escape_ldap_filter
  24
  25 class CharField(fields.CharField):
  26     def __init__(self, *args, **kwargs):
  27         kwargs['max_length'] = 200
  28         super(CharField, self).__init__(*args, **kwargs)
  29
  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]
  41
  42         raise TypeError("CharField has invalid lookup: %s" % lookup_type)
  43
  44     def get_prep_lookup(self, lookup_type, value):
  45         return escape_ldap_filter(value)
  46
  47 class ImageField(fields.Field):
  48     def get_db_prep_lookup(self, lookup_type, value):
  49         raise TypeError("ImageField has invalid lookup: %s" % lookup_type)
  50
  51 class IntegerField(fields.IntegerField):
  52     def get_db_prep_lookup(self, lookup_type, value):
  53         if lookup_type == 'exact':
  54             return [value]
  55
  56         raise TypeError("IntegerField has invalid lookup: %s" % lookup_type)
  57
  58 class ListField(fields.Field):
  59     __metaclass__ = SubfieldBase
  60
  61     def get_db_prep_lookup(self, lookup_type, value):
  62         raise TypeError("ListField has invalid lookup: %s" % lookup_type)
  63
  64     def to_python(self, value):
  65         if not value:
  66             return []
  67         return value
  68