Mercurial > public > sg101
diff gpp/potd/models.py @ 515:ae89ba801e8b
For #194, convert the POTD management command to a celery task.
Refactored to put the logic for the command into a function, and the command simply calls this function. The task can also just call this function. Added some basic tests for the new function.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Wed, 14 Dec 2011 02:41:15 +0000 |
| parents | 7dbdbb08e68c |
| children | ff67946fd4b0 |
line wrap: on
line diff
--- a/gpp/potd/models.py Tue Dec 13 02:37:50 2011 +0000 +++ b/gpp/potd/models.py Wed Dec 14 02:41:15 2011 +0000 @@ -1,5 +1,6 @@ """ Models for the Photo Of The Day (POTD) application. + """ import os from PIL import ImageFile @@ -89,7 +90,15 @@ class CurrentManager(models.Manager): + """ + Manager for the Current model. + + """ def get_current_photo(self): + """ + Retrieves the current photo object from the current record. + + """ try: c = self.get(pk=1) return c.potd @@ -97,6 +106,10 @@ return None def get_current_id(self): + """ + Returns the ID of the current POTD from the current record. + + """ potd = self.get_current_photo() if potd is not None: return potd.pk @@ -117,7 +130,16 @@ class SequenceManager(models.Manager): + """ + Manager for the Sequence model. + + """ def insert_photo(self, photo_id): + """ + Inserts the given photo_id just after the current photo so it + will appear as tomorrow's POTD. + + """ current = Current.objects.get_current_id() if current is not None: s = self.get(pk=1) @@ -129,6 +151,10 @@ s.save() def remove_photo(self, photo_id): + """ + Removes a given photo id from the sequence of photos. + + """ try: s = self.get(pk=1) except Sequence.DoesNotExist: