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

Check first for path to exists

Browse source
This commit is contained in:
Antonio J. Delgado 2022-05-02 14:13:03 +03:00
parent 80a1e50912
commit b44c274924
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
image_classifier/image_classifier.py
View file

@ -110,10 +110,13 @@ class image_classifier:
else: else:
folder = 'unknown-time' folder = 'unknown-time'
new_path = os.path.join(dirname, folder, filename) new_path = os.path.join(dirname, folder, filename)
os.makedirs(os.path.dirname(new_path), exist_ok=True) if not os.path.exists(os.path.join(dirname, folder)):
os.makedirs(os.path.dirname(new_path))
if self.people_folder: if self.people_folder:
for person in people: for person in people:
person_path = os.path.join(self.people_folder, person, folder, filename) person_path = os.path.join(self.people_folder, person, folder, filename)
if not os.path.exists(os.path.join(self.people_folder, person, folder)):
os.makedirs(os.path.dirname(person_path))
self._log.debug(f"Copying file '{file}' to '{person_path}'...") self._log.debug(f"Copying file '{file}' to '{person_path}'...")
shutil.copy(file, person_path) shutil.copy(file, person_path)
if not self.no_move: if not self.no_move: