Show More
@@ -1,5 +1,7 b'' | |||
|
1 | 1 | import re |
|
2 | 2 | |
|
3 | from PIL import Image | |
|
4 | ||
|
3 | 5 | from django.contrib.staticfiles import finders |
|
4 | 6 | from django.contrib.staticfiles.templatetags.staticfiles import static |
|
5 | 7 | from django.core.files.images import get_image_dimensions |
@@ -145,7 +147,15 b' class ImageViewer(AbstractViewer):' | |||
|
145 | 147 | def get_format_view(self): |
|
146 | 148 | metadata = '{}, {}'.format(self.file.name.split('.')[-1], |
|
147 | 149 | filesizeformat(self.file.size)) |
|
148 | width, height = get_image_dimensions(self.file.path) | |
|
150 | ||
|
151 | Image.warnings.simplefilter('error', Image.DecompressionBombWarning) | |
|
152 | try: | |
|
153 | width, height = get_image_dimensions(self.file.path) | |
|
154 | except Exception: | |
|
155 | # If the image is a decompression bomb, treat it as just a regular | |
|
156 | # file | |
|
157 | return super().get_format_view() | |
|
158 | ||
|
149 | 159 | preview_path = self.file.path.replace('.', '.200x150.') |
|
150 | 160 | pre_width, pre_height = get_image_dimensions(preview_path) |
|
151 | 161 |
General Comments 0
You need to be logged in to leave comments.
Login now