From afafdd2bf9e9a0c28819a2b5bb74e091b145fb46 Mon Sep 17 00:00:00 2001 From: Carl Meyer Date: Fri, 30 Jul 2010 22:45:45 -0400 Subject: [PATCH] make manager_from Django 1.1 compatible --- model_utils/managers.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/model_utils/managers.py b/model_utils/managers.py index 311c956..33c3f6f 100644 --- a/model_utils/managers.py +++ b/model_utils/managers.py @@ -56,6 +56,11 @@ def manager_from(*mixins, **kwds): qset_cls = type('Queryset_%d' % id, tuple(bases), attrs) # create the Manager subclass bases[0] = kwds.get('manager_cls', Manager) - attrs['get_query_set'] = lambda self: qset_cls(self.model, using=self._db) + def _get_query_set(self): + if hasattr(self, '_db'): + return qset_cls(self.model, using=self._db) + else: + return qset_cls(self.model) + attrs['get_query_set'] = _get_query_set manager_cls = type('Manager_%d' % id, tuple(bases), attrs) return manager_cls()