From 23011aee63bc3bc131fbd77bd5732edf517986cb Mon Sep 17 00:00:00 2001 From: "Antonio J. Delgado" Date: Tue, 17 May 2022 11:15:04 +0300 Subject: [PATCH] check first for access --- image_classifier/image_classifier.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index 0fdf6db..d1c4db1 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -57,6 +57,9 @@ class image_classifier: log_file = os.path.join(home_path, 'log', 'image_classifier.log') self.log_file = log_file self._init_log() + if not os.access(directory, os.R_OK): + self._log.error(f"Unable to access directory '{directory}'") + exit 1 self.faces_directory = faces_directory self.directory = directory self.known_people = self.load_known_people() @@ -64,17 +67,14 @@ class image_classifier: self.people_folder = people_folder self.recursive = recursive - if os.access(directory, os.R_OK): - if self.recursive: - entries = self.recursive_scandir(directory) - else: - entries = os.scandir(directory) - self._log.debug(f"Processing {len(entries)} files...") - for entry in entries: - if not entry.name.startswith('.') and entry.is_file(): - self.process_file(entry.path) + if self.recursive: + entries = self.recursive_scandir(directory) else: - self._log.error(f"Unable to access directory '{directory}'") + entries = os.scandir(directory) + self._log.debug(f"Processing {len(entries)} files...") + for entry in entries: + if not entry.name.startswith('.') and entry.is_file(): + self.process_file(entry.path) def recursive_scandir(path, ignore_hidden_files=True): files = []