Mercurial > public > sg101
annotate gpp/potd/management/commands/pick_potd.py @ 467:b910cc1460c8
Add the ability to conditionally add model instances to the search index on update. This is not perfect, as some instances should be deleted from the index if they are updated such that they should not be in the index anymore. Will think about and address that later.
| author | Brian Neal <bgneal@gmail.com> |
|---|---|
| date | Sun, 24 Jul 2011 18:12:20 +0000 |
| parents | dbd703f7d63a |
| children | ae89ba801e8b |
| rev | line source |
|---|---|
| gremmie@1 | 1 """ |
| gremmie@1 | 2 pick_potd is a custom manage.py command for the POTD application. |
| gremmie@1 | 3 It is intended to be called from a cron job at midnight to pick the |
| gremmie@1 | 4 new POTD. |
| gremmie@1 | 5 """ |
| gremmie@1 | 6 |
| gremmie@1 | 7 import random |
| gremmie@1 | 8 from django.core.management.base import NoArgsCommand |
| gremmie@1 | 9 |
| gremmie@1 | 10 from potd.models import Current |
| gremmie@1 | 11 from potd.models import Sequence |
| gremmie@1 | 12 from potd.models import Photo |
| gremmie@1 | 13 |
| gremmie@1 | 14 def get_sequence(): |
| gremmie@1 | 15 try: |
| gremmie@1 | 16 s = Sequence.objects.get(pk=1) |
| gremmie@1 | 17 if s.seq: |
| gremmie@1 | 18 return [int(x) for x in s.seq.split(',')] |
| gremmie@1 | 19 except: |
| gremmie@1 | 20 pass |
| gremmie@1 | 21 return [] |
| gremmie@1 | 22 |
| gremmie@1 | 23 def new_sequence(): |
| gremmie@1 | 24 the_ids = Photo.objects.values_list('id', flat=True).order_by('id') |
| gremmie@1 | 25 ids = [] |
| gremmie@1 | 26 for id in the_ids.iterator(): |
| gremmie@1 | 27 ids.append(int(id)) |
| gremmie@1 | 28 |
| gremmie@1 | 29 random.shuffle(ids) |
| gremmie@1 | 30 try: |
| gremmie@1 | 31 s = Sequence.objects.get(pk=1) |
| gremmie@1 | 32 except Sequence.DoesNotExist: |
| gremmie@1 | 33 s = Sequence() |
| gremmie@1 | 34 |
| gremmie@1 | 35 s.seq = ','.join([str(id) for id in ids]) |
| gremmie@1 | 36 s.save() |
| gremmie@1 | 37 return ids |
| gremmie@1 | 38 |
| gremmie@1 | 39 class Command(NoArgsCommand): |
| gremmie@1 | 40 help = "Chooses the next POTD. Run this command at midnight to update the POTD." |
| gremmie@1 | 41 #requires_model_validation = False |
| gremmie@1 | 42 |
| gremmie@1 | 43 def handle_noargs(self, **options): |
| gremmie@1 | 44 try: |
| gremmie@1 | 45 c = Current.objects.get(pk=1) |
| gremmie@1 | 46 current = c.potd.pk |
| gremmie@1 | 47 except Current.DoesNotExist: |
| gremmie@1 | 48 c = Current() |
| gremmie@1 | 49 current = None |
| gremmie@1 | 50 |
| gremmie@1 | 51 seq = get_sequence() |
| gremmie@1 | 52 if current is None or len(seq) == 0 or current == seq[-1]: |
| gremmie@1 | 53 # time to generate a new random sequence |
| gremmie@1 | 54 seq = new_sequence() |
| gremmie@1 | 55 # set current to the first one in the sequence |
| gremmie@1 | 56 if len(seq) > 0: |
| gremmie@1 | 57 try: |
| gremmie@1 | 58 c.potd = Photo.objects.get(pk=seq[0]) |
| gremmie@1 | 59 c.potd.potd_count += 1 |
| gremmie@1 | 60 c.potd.save() |
| gremmie@1 | 61 c.save() |
| gremmie@1 | 62 except: |
| gremmie@1 | 63 pass |
| gremmie@1 | 64 else: |
| gremmie@1 | 65 # find current in the sequence, pick the next one |
| gremmie@1 | 66 try: |
| gremmie@1 | 67 i = seq.index(current) |
| gremmie@1 | 68 c.potd = Photo.objects.get(pk=seq[i + 1]) |
| gremmie@1 | 69 c.potd.potd_count += 1 |
| gremmie@1 | 70 c.potd.save() |
| gremmie@1 | 71 c.save() |
| gremmie@1 | 72 except: |
| gremmie@1 | 73 pass |
| gremmie@1 | 74 |