From 660db24dec5fc9b298c0a42ad48ca47be8edb046 Mon Sep 17 00:00:00 2001 From: "Antonio J. Delgado" Date: Mon, 6 Sep 2021 16:17:53 +0300 Subject: [PATCH] Add check for no image --- image_classifier/image_classifier.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index e358b8b..ca45d7f 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -66,12 +66,13 @@ class image_classifier: move it to the folder of the day''' self._log.debug(f"Processing file '{file}'...") people = self.find_faces(file) - with open(file, 'rb') as image_file: - exif_info = exif.Image(image_file) - if exif_info.has_exif: - print(json.dumps(exif_info, indent=2)) - # get date - # move to destination + if people: + with open(file, 'rb') as image_file: + exif_info = exif.Image(image_file) + if exif_info.has_exif: + print(json.dumps(exif_info, indent=2)) + # get date + # move to destination def load_known_people(self): known_people = list() @@ -99,6 +100,7 @@ class image_classifier: names.append(known_person['name']) except PIL.UnidentifiedImageError as error: self._log.debug(f"File '{file}' don't seem to be an image.") + return False return names def _init_log(self):