voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Let helpers.generate_*_thumbnail raise FileNotFoundError.

Browse source
This commit is contained in:
voussoir 2018-08-17 22:05:47 -07:00
parent 8fdbd49f70
commit 0ccfa74709
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
etiquette/helpers.py
View file

@ -156,6 +156,8 @@ def fit_into_bounds(image_width, image_height, frame_width, frame_height):
return (new_width, new_height) return (new_width, new_height)
def generate_image_thumbnail(filepath, width, height): def generate_image_thumbnail(filepath, width, height):
if not os.path.isfile(filepath):
raise FileNotFoundError(filepath)
image = PIL.Image.open(filepath) image = PIL.Image.open(filepath)
(image_width, image_height) = image.size (image_width, image_height) = image.size
(new_width, new_height) = fit_into_bounds( (new_width, new_height) = fit_into_bounds(
@ -183,6 +185,8 @@ def generate_image_thumbnail(filepath, width, height):
return image return image
def generate_video_thumbnail(filepath, outfile, width, height, **special): def generate_video_thumbnail(filepath, outfile, width, height, **special):
if not os.path.isfile(filepath):
raise FileNotFoundError(filepath)
probe = constants.ffmpeg.probe(filepath) probe = constants.ffmpeg.probe(filepath)
if not probe.video: if not probe.video:
return False return False