takino/ordigi
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Closes gh-4 Check for image type using imghdr module

Browse Source
This commit is contained in:
Jaisen Mathai 2015-12-04 00:54:21 -08:00
parent 40c9b20de0
commit 47627df3ad
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
elodie/media/media.py
View File

@ -17,6 +17,7 @@ import pyexiv2
import re import re
import subprocess import subprocess
import time import time
import imghdr
""" """
Media class for general video operations Media class for general video operations
@ -88,6 +89,12 @@ class Media(object):
""" """
def is_valid(self): def is_valid(self):
source = self.source source = self.source
# gh-4 This checks if the source file is an image.
# It doesn't validate against the list of supported types.
if(self.__name__ == 'Photo' and imghdr.what(source) is None):
return False;
# we can't use self.__get_extension else we'll endlessly recurse # we can't use self.__get_extension else we'll endlessly recurse
return os.path.splitext(source)[1][1:].lower() in self.get_valid_extensions() return os.path.splitext(source)[1][1:].lower() in self.get_valid_extensions()