690c42c787710229f12f75cd7a64a8b827a57212
[wolnelektury.git] / apps / sorl / thumbnail / management / commands / thumbnail_cleanup.py
1 import os
2 import re
3 from django.db import models
4 from django.conf import settings
5 from django.core.management.base import NoArgsCommand
6 from sorl.thumbnail.main import get_thumbnail_setting
7
8
9 try:
10     set
11 except NameError:
12     from sets import Set as set     # For Python 2.3
13
14 thumb_re = re.compile(r'^%s(.*)_\d{1,}x\d{1,}_[-\w]*q([1-9]\d?|100)\.jpg' %
15                       get_thumbnail_setting('PREFIX'))
16
17
18 def get_thumbnail_path(path):
19     basedir = get_thumbnail_setting('BASEDIR')
20     subdir = get_thumbnail_setting('SUBDIR')
21     return os.path.join(basedir, path, subdir)
22
23
24 def clean_up():
25     paths = set()
26     for app in models.get_apps():
27         model_list = models.get_models(app)
28         for model in model_list:
29             for field in model._meta.fields:
30                 if isinstance(field, models.ImageField):
31                     #TODO: take care of date formatted and callable upload_to.
32                     if (not callable(field.upload_to) and
33                             field.upload_to.find("%") == -1):
34                         paths = paths.union((field.upload_to,))
35     paths = list(paths)
36     for path in paths:
37         thumbnail_path = get_thumbnail_path(path)
38         try:
39             file_list = os.listdir(os.path.join(settings.MEDIA_ROOT,
40                                                 thumbnail_path))
41         except OSError:
42             continue # Dir doesn't exists, no thumbnails here.
43         for fn in file_list:
44             m = thumb_re.match(fn)
45             if m:
46                 # Due to that the naming of thumbnails replaces the dot before
47                 # extension with an underscore we have 2 possibilities for the
48                 # original filename. If either present we do not delete
49                 # suspected thumbnail.
50                 # org_fn is the expected original filename w/o extension
51                 # org_fn_alt is the expected original filename with extension
52                 org_fn = m.group(1)
53                 org_fn_exists = os.path.isfile(
54                             os.path.join(settings.MEDIA_ROOT, path, org_fn))
55
56                 usc_pos = org_fn.rfind("_")
57                 if usc_pos != -1:
58                     org_fn_alt = "%s.%s" % (org_fn[0:usc_pos],
59                                             org_fn[usc_pos+1:])
60                     org_fn_alt_exists = os.path.isfile(
61                         os.path.join(settings.MEDIA_ROOT, path, org_fn_alt))
62                 else:
63                     org_fn_alt_exists = False
64                 if not org_fn_exists and not org_fn_alt_exists:
65                     del_me = os.path.join(settings.MEDIA_ROOT,
66                                           thumbnail_path, fn)
67                     os.remove(del_me)
68
69
70 class Command(NoArgsCommand):
71     help = "Deletes thumbnails that no longer have an original file."
72     requires_model_validation = False
73
74     def handle_noargs(self, **options):
75         clean_up()