# -*- 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 get_db_prep_value(self, value):
- """Returns field's value prepared for interacting with the database
- backend.
+ def get_db_prep_lookup(self, lookup_type, value):
+ if lookup_type == 'endswith':
+ return ["*%s" % escape_ldap_filter(value)]
+ elif lookup_type == 'startswith':
+ return ["%s*" % escape_ldap_filter(value)]
+ elif lookup_type == 'contains':
+ 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]
- Used by the default implementations of ``get_db_prep_save``and
- `get_db_prep_lookup```
- """
- return value.replace('\\', '\\5c')
- .replace('*', '\\2a')
- .replace('(', '\\28')
- .replace(')', '\\29')
- .replace('\0', '\\00')
+ raise TypeError("CharField has invalid lookup: %s" % lookup_type)
+ def get_prep_lookup(self, lookup_type, value):
+ return escape_ldap_filter(value)
+
class ImageField(fields.Field):
pass
class IntegerField(fields.IntegerField):
- pass
+ def get_db_prep_lookup(self, lookup_type, value):
+ if lookup_type == 'exact':
+ return [value]
+
+ raise TypeError("IntegerField has invalid lookup: %s" % lookup_type)
class ListField(fields.Field):
__metaclass__ = SubfieldBase