easy_thumbnails.management: 77 total statements, 0.0% covered

Generated: Wed 2013-03-13 10:33 CET

Source file: /media/Envs/Envs/filer-gallery/lib/python2.7/site-packages/easy_thumbnails/management/__init__.py

Stats: 0 executed, 74 missed, 3 excluded, 50 ignored

1. import os
2. import re
4. from easy_thumbnails.conf import settings
6. re_thumbnail_file = re.compile(r'(?P<source_filename>.+)_(?P<x>\d+)x(?P<y>\d+)'
7. r'(?:_(?P<options>\w+))?_q(?P<quality>\d+)'
8. r'(?:.[^.]+)?$')
11. def all_thumbnails(path, recursive=True, prefix=None, subdir=None):
12. """
13. Return a dictionary referencing all files which match the thumbnail format.
15. Each key is a source image filename, relative to path.
16. Each value is a list of dictionaries as explained in `thumbnails_for_file`.
17. """
18. if prefix is None:
19. prefix = settings.THUMBNAIL_PREFIX
20. if subdir is None:
21. subdir = settings.THUMBNAIL_SUBDIR
22. thumbnail_files = {}
23. if not path.endswith('/'):
24. path = '%s/' % path
25. len_path = len(path)
26. if recursive:
27. all = os.walk(path)
28. else:
29. files = []
30. for file in os.listdir(path):
31. if os.path.isfile(os.path.join(path, file)):
32. files.append(file)
33. all = [(path, [], files)]
34. for dir_, subdirs, files in all:
35. rel_dir = dir_[len_path:]
36. for file in files:
37. thumb = re_thumbnail_file.match(file)
38. if not thumb:
39. continue
40. d = thumb.groupdict()
41. source_filename = d.pop('source_filename')
42. if prefix:
43. source_path, source_filename = os.path.split(source_filename)
44. if not source_filename.startswith(prefix):
45. continue
46. source_filename = os.path.join(source_path,
47. source_filename[len(prefix):])
48. d['options'] = d['options'] and d['options'].split('_') or []
49. if subdir and rel_dir.endswith(subdir):
50. rel_dir = rel_dir[:-len(subdir)]
51. # Corner-case bug: if the filename didn't have an extension but did
52. # have an underscore, the last underscore will get converted to a
53. # '.'.
54. m = re.match(r'(.*)_(.*)', source_filename)
55. if m:
56. source_filename = '%s.%s' % m.groups()
57. filename = os.path.join(rel_dir, source_filename)
58. thumbnail_file = thumbnail_files.setdefault(filename, [])
59. d['filename'] = os.path.join(dir_, file)
60. thumbnail_file.append(d)
61. return thumbnail_files
64. def thumbnails_for_file(relative_source_path, root=None, basedir=None,
65. subdir=None, prefix=None):
66. """
67. Return a list of dictionaries, one for each thumbnail belonging to the
68. source image.
70. The following list explains each key of the dictionary:
72. `filename` -- absolute thumbnail path
73. `x` and `y` -- the size of the thumbnail
74. `options` -- list of options for this thumbnail
75. `quality` -- quality setting for this thumbnail
76. """
77. if root is None:
78. root = settings.MEDIA_ROOT
79. if prefix is None:
80. prefix = settings.THUMBNAIL_PREFIX
81. if subdir is None:
82. subdir = settings.THUMBNAIL_SUBDIR
83. if basedir is None:
84. basedir = settings.THUMBNAIL_BASEDIR
85. source_dir, filename = os.path.split(relative_source_path)
86. thumbs_path = os.path.join(root, basedir, source_dir, subdir)
87. if not os.path.isdir(thumbs_path):
88. return []
89. files = all_thumbnails(thumbs_path, recursive=False, prefix=prefix,
90. subdir='')
91. return files.get(filename, [])
94. def delete_thumbnails(relative_source_path, root=None, basedir=None,
95. subdir=None, prefix=None):
96. """
97. Delete all thumbnails for a source image.
98. """
99. thumbs = thumbnails_for_file(relative_source_path, root, basedir, subdir,
100. prefix)
101. return _delete_using_thumbs_list(thumbs)
104. def _delete_using_thumbs_list(thumbs):
105. deleted = 0
106. for thumb_dict in thumbs:
107. filename = thumb_dict['filename']
108. try:
109. os.remove(filename)
110. except:
111. pass
112. else:
113. deleted += 1
114. return deleted
117. def delete_all_thumbnails(path, recursive=True):
118. """
119. Delete all files within a path which match the thumbnails pattern.
121. By default, matching files from all sub-directories are also removed. To
122. only remove from the path directory, set recursive=False.
123. """
124. total = 0
125. for thumbs in all_thumbnails(path, recursive=recursive).values():
126. total += _delete_using_thumbs_list(thumbs)
127. return total