diff options
author | Étienne Loks <etienne.loks@iggdrasil.net> | 2016-02-19 19:22:08 +0100 |
---|---|---|
committer | Étienne Loks <etienne.loks@iggdrasil.net> | 2016-02-19 19:22:08 +0100 |
commit | c3f043e4937bd9ed7c68a38488e588866943f4c5 (patch) | |
tree | 48ecf37a9c135f85786e7ba9d1d24c75b4366c29 /chimere/managers.py | |
parent | b3058742a5acc9ca19a5edc31260b9ad0ca5ba5a (diff) | |
download | Chimère-c3f043e4937bd9ed7c68a38488e588866943f4c5.tar.bz2 Chimère-c3f043e4937bd9ed7c68a38488e588866943f4c5.zip |
Flake8
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() |