Update for GeoDjango support

queryset_transform/__init__.py

@@ -1,12 +1,12 @@
 from django.db import models
 
-class TransformQuerySet(models.query.QuerySet):
+class TransformQuerySetMixin(object):
     def __init__(self, *args, **kwargs):
-        super(TransformQuerySet, self).__init__(*args, **kwargs)
+        super(TransformQuerySetMixin, self).__init__(*args, **kwargs)
         self._transform_fns = []
 
     def _clone(self, klass=None, setup=False, **kw):
-        c = super(TransformQuerySet, self)._clone(klass, setup, **kw)
+        c = super(TransformQuerySetMixin, self)._clone(klass, setup, **kw)
         c._transform_fns = self._transform_fns[:]
         return c
 
@@ -16,15 +16,19 @@ def transform(self, fn):
         return c
 
     def iterator(self):
-        result_iter = super(TransformQuerySet, self).iterator()
+        result_iter = super(TransformQuerySetMixin, self).iterator()
         if self._transform_fns:
             results = list(result_iter)
             for fn in self._transform_fns:
                 fn(results)
             return iter(results)
         return result_iter
 
-class TransformManager(models.Manager):
 
+class TransformQuerySet(TransformQuerySetMixin, models.query.QuerySet):
+    pass
+
+
+class TransformManager(models.Manager):
     def get_query_set(self):
         return TransformQuerySet(self.model)

queryset_transform/geo.py

@@ -0,0 +1,13 @@
+from django.contrib.gis.db.models.manager import GeoManager
+from django.contrib.gis.db.models.query import GeoQuerySet
+
+from . import TransformQuerySetMixin
+
+
+class GeoTransformQuerySet(TransformQuerySetMixin, GeoQuerySet):
+    pass
+
+
+class GeoTransformManager(GeoManager):
+    def get_query_set(self):
+        return GeoTransformQuerySet(self.model)