diff options
Diffstat (limited to 'chimere/managers.py')
-rw-r--r-- | chimere/managers.py | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/chimere/managers.py b/chimere/managers.py index 89aa35f..807b493 100644 --- a/chimere/managers.py +++ b/chimere/managers.py @@ -1,14 +1,13 @@ # -*- coding: utf-8 -*- -import sys -from django.contrib.gis.db import models - -from logging import getLogger -from datetime import datetime, date, timedelta +from datetime import date, timedelta +from django.contrib.gis.db import models from django.conf import settings # BASE + + class BaseGeoQuerySet(models.query.QuerySet): """ Custom queryset for Geo objects. @@ -16,11 +15,11 @@ class BaseGeoQuerySet(models.query.QuerySet): def available(self): """ Chainable filter to retrieve only available elements. """ return self.filter(status="A") - + def submited(self): """ Chainable filter to retrieve only submited elements. """ return self.filter(status="S") - + def active(self): """ Returns objects where within the active date period of events. @@ -31,7 +30,8 @@ class BaseGeoQuerySet(models.query.QuerySet): today = date.today() after = today + timedelta(settings.CHIMERE_DAYS_BEFORE_EVENT) return self.filter(end_date__gte=today, start_date__lte=after) - + + class BaseGeoManager(models.GeoManager): """ Custom base manager for Geo objects. @@ -39,14 +39,14 @@ class BaseGeoManager(models.GeoManager): def get_query_set(self): """ Use our custom QuerySet. """ return BaseGeoQuerySet(self.model) - + # Methods defined in the queryset that we also want exposed in the # It would be nice to implement it with a magic method instead, later :) def available(self): return self.get_query_set().available() - + def submited(self): return self.get_query_set().submited() - + def active(self): return self.get_query_set().active() |