adelgado/image_classifier
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

check first for access

Browse source
This commit is contained in:
Antonio J. Delgado 2022-05-17 11:15:04 +03:00
parent 6380a1404d
commit 23011aee63
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
image_classifier/image_classifier.py
View file

@ -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,7 +67,6 @@ 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:
@ -73,8 +75,6 @@ class image_classifier:
for entry in entries:
if not entry.name.startswith('.') and entry.is_file():
self.process_file(entry.path)
else:
self._log.error(f"Unable to access directory '{directory}'")
def recursive_scandir(path, ignore_hidden_files=True):
files = []