add support for "greater or equal" and "less or equal" lookups on IntegerFields

[matthijs/upstream/django-ldapdb.git] / ldapdb / models / fields.py

diff --git a/ldapdb/models/fields.py b/ldapdb/models/fields.py
index ff0ad722b8e75a8d0067933be4637170e1cf426b..d21bbd0dc258c31bc05a35b8a1ee2d7a0b69eb07 100644 (file)
--- a/ldapdb/models/fields.py
+++ b/ldapdb/models/fields.py
@@ -50,7 +50,7 @@ class ImageField(fields.Field):
 
 class IntegerField(fields.IntegerField):
     def get_db_prep_lookup(self, lookup_type, value):
-        if lookup_type == 'exact':
+        if lookup_type in ('exact', 'gte', 'lte'):
             return [value]
 
         raise TypeError("IntegerField has invalid lookup: %s" % lookup_type)