manager.py 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. from django.db import models, connection
  2. from django.db.models import sql, query
  3. INET_TERMS = {
  4. 'inet_lt': '<',
  5. 'inet_lte': '<=',
  6. 'inet_exact': '=',
  7. 'inet_gte': '>=',
  8. 'inet_gt': '>',
  9. 'inet_not': '<>',
  10. 'inet_is_contained': '<<',
  11. 'inet_is_contained_or_equal': '<<=',
  12. 'inet_contains': '>>',
  13. 'inet_contains': '>>=',
  14. }
  15. class IPQuery(sql.Query):
  16. query_terms = sql.Query.query_terms.copy()
  17. query_terms.update(INET_TERMS)
  18. class IPWhere(sql.where.WhereNode):
  19. def make_atom(self, child, qn):
  20. table_alias, name, db_type, lookup_type, value_annot, params = child
  21. if db_type == 'inet' and lookup_type in INET_TERMS:
  22. return ('%s.%s %s inet %%s' % (table_alias, name, INET_TERMS[lookup_type]), params)
  23. return super(IPWhere, self).make_atom(child, qn)
  24. class IPManger(models.Manager):
  25. def get_query_set(self):
  26. q = IPQuery(self.model, connection, IPWhere)
  27. return query.QuerySet(self.model, q)
  28. class IPAddressField(models.IPAddressField):
  29. def get_db_prep_lookup(self, lookup_type, value):
  30. if lookup_type in INET_TERMS:
  31. return [value]
  32. return super(IPAddressField, self).get_db_prep_lookup(lookup_type, value)
  33. class Foo(models.Model):
  34. ip = IPAddressField()
  35. objects = IPManger()