Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-05-14 10:13:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Cache the results of has_jpeg_support and has_png_support

Browse source 
This means we can use them on the admin dashboard without performing redundant file/image operations.
This commit is contained in:
Matt Westcott 2015-09-23 02:29:18 +01:00
parent a311955534
commit 1a743366fb
1 changed files with 34 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
wagtail/wagtailimages/checks.py
View file

@ -5,34 +5,48 @@ from django.core.checks import register, Warning
from willow.image import Image
_has_jpeg_support = None
_has_png_support = None
def has_jpeg_support():
wagtail_jpg = os.path.join(os.path.dirname(__file__), 'check_files', 'wagtail.jpg')
is_ok = True
f = open(wagtail_jpg, 'rb')
global _has_jpeg_support
try:
Image.open(f)
except (IOError, Image.LoaderError):
is_ok = False
finally:
f.close()
if _has_jpeg_support is None:
wagtail_jpg = os.path.join(os.path.dirname(__file__), 'check_files', 'wagtail.jpg')
succeeded = True
f = open(wagtail_jpg, 'rb')
return is_ok
try:
Image.open(f)
except (IOError, Image.LoaderError):
succeeded = False
finally:
f.close()
_has_jpeg_support = succeeded
return _has_jpeg_support
def has_png_support():
wagtail_png = os.path.join(os.path.dirname(__file__), 'check_files', 'wagtail.png')
is_ok = True
f = open(wagtail_png, 'rb')
global _has_png_support
try:
Image.open(f)
except (IOError, Image.LoaderError):
is_ok = False
finally:
f.close()
if _has_png_support is None:
wagtail_png = os.path.join(os.path.dirname(__file__), 'check_files', 'wagtail.png')
succeeded = True
f = open(wagtail_png, 'rb')
return is_ok
try:
Image.open(f)
except (IOError, Image.LoaderError):
succeeded = False
finally:
f.close()
_has_png_support = succeeded
return _has_png_support
@register()