Mercurial > public > madeira

comparison photologue/management/commands/plcache.py @ 71:e2868ad47a1e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
For Django 1.4, using the new manage.py.
author Brian Neal <bgneal@gmail.com>
date Sat, 14 Apr 2012 16:40:29 -0500
parents madeira/photologue/management/commands/plcache.py@63e4211628e1
children
comparison
equal deleted inserted replaced
70:f26cdda0ad8b 71:e2868ad47a1e
1 from django.core.management.base import BaseCommand, CommandError
2 from optparse import make_option
3 from photologue.models import PhotoSize, ImageModel
4
5 class Command(BaseCommand):
6 option_list = BaseCommand.option_list + (
7 make_option('--reset', '-r', action='store_true', dest='reset', help='Reset photo cache before generating'),
8 )
9
10 help = ('Manages Photologue cache file for the given sizes.')
11 args = '[sizes]'
12
13 requires_model_validation = True
14 can_import_settings = True
15
16 def handle(self, *args, **options):
17 return create_cache(args, options)
18
19 def create_cache(sizes, options):
20 """
21 Creates the cache for the given files
22 """
23 reset = options.get('reset', None)
24
25 size_list = [size.strip(' ,') for size in sizes]
26
27 if len(size_list) < 1:
28 sizes = PhotoSize.objects.filter(pre_cache=True)
29 else:
30 sizes = PhotoSize.objects.filter(name__in=size_list)
31
32 if not len(sizes):
33 raise CommandError('No photo sizes were found.')
34
35 print 'Caching photos, this may take a while...'
36
37 for cls in ImageModel.__subclasses__():
38 for photosize in sizes:
39 print 'Cacheing %s size images' % photosize.name
40 for obj in cls.objects.all():
41 if reset:
42 obj.remove_size(photosize)
43 obj.create_size(photosize)