Archived
145 lines
5.7 KiB
Python
145 lines
5.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# django-ldapdb
|
|
# Copyright (c) 2009-2011, Bolloré telecom
|
|
# All rights reserved.
|
|
#
|
|
# See AUTHORS file for a full list of contributors.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without modification,
|
|
# are permitted provided that the following conditions are met:
|
|
#
|
|
# 1. Redistributions of source code must retain the above copyright notice,
|
|
# this list of conditions and the following disclaimer.
|
|
#
|
|
# 2. Redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
#
|
|
# 3. Neither the name of Bolloré telecom nor the names of its contributors
|
|
# may be used to endorse or promote products derived from this software
|
|
# without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
|
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
|
# ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
#
|
|
|
|
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, connection, prepared=False):
|
|
"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):
|
|
def from_ldap(self, value, connection):
|
|
if len(value) == 0:
|
|
return ''
|
|
else:
|
|
return value[0]
|
|
|
|
def get_db_prep_lookup(self, lookup_type, value, connection, prepared=False):
|
|
"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):
|
|
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, connection, prepared=False):
|
|
"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, connection, prepared=False):
|
|
"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 []
|
|
return value
|
|
|