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

Add option to configure folder

Browse source
This commit is contained in:
Antonio J. Delgado 2022-09-05 15:14:03 +03:00
parent d9fd9255cc
commit 1c8c6a7991
2 changed files with 18 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
README.md
View file

@ -4,4 +4,5 @@ Classifies images by their EXIF date (e.g. /path/YYYY.MM.DD/image.jpg).
= Install
apt install exiv2 cmake libexiv2-dev libboost-python-dev
python3 setup.py install

30
image_classifier/image_classifier.py
View file

@ -17,6 +17,7 @@ from logging.handlers import SysLogHandler
import face_recognition
import pyexiv2
import PIL
import datetime
class CustomFormatter(logging.Formatter):
@ -49,7 +50,7 @@ class CustomFormatter(logging.Formatter):
class image_classifier:
def __init__(self, debug_level, log_file, faces_directory, directory, no_move,
people_folder, recursive):
people_folder, recursive, folder_date_format):
''' Initial function called when object is created '''
self.debug_level = debug_level
if log_file is None:
@ -66,6 +67,7 @@ class image_classifier:
self.no_move = no_move
self.people_folder = people_folder
self.recursive = recursive
self.folder_date_format = folder_date_format
if self.recursive:
entries = self.recursive_scandir(directory)
@ -122,32 +124,32 @@ class image_classifier:
if not os.access(file, os.R_OK):
self._log.error(f"The file '{file}' is not readable.")
else:
file_date = "unknown-time"
if self.is_image(file):
if 'Exif.Photo.DateTimeOriginal' in self.metadata.exif_keys:
original_date = self.metadata['Exif.Photo.DateTimeOriginal'].value
self._log.debug(f"File creation time: {original_date} \
self._log.debug(f"File creation time in EXIF: {original_date} \
(type: {type(original_date)})")
try:
file_date = original_date.strftime('%Y.%m.%d')
#file_date = original_date.strftime('%Y/%Y.%m.%d')
file_date = original_date
except Exception as error:
self._log.error(f"Failed to convert EXIF information about date '{original_date}'.")
file_date = 'unknown-time'
if file_date == 'unknown-time':
file_date = None
if file_date is None:
self._log.debug('Date not stored in EXIF metadata')
match = re.search(
r'(?P<year>20[0-9]{2})[\-/\._]?(?P<month>[0-1]?[0-9])[\-/\._]?(?P<day>[0-3]?[0-9])',
filename
)
if match:
file_date = f"{match.group('year')}.{match.group('month')}.\
{match.group('day')}"
file_date = datetime.strptime(f"{match.group('year')}.{match.group('month')}.\
{match.group('day')}", '%Y.%m.%d')
else:
match = re.search(r'(?P<day>[0-3]?[0-9])[\-/\._]?\
(?P<month>[0-1]?[0-9])[\-/\._]?(?P<year>20[0-9]{2})', filename)
if match:
file_date = f"{match.group('year')}.{match.group('month')}.\
{match.group('day')}"
file_date = datetime.strptime(f"{match.group('year')}.{match.group('month')}.\
{match.group('day')}", '%Y.%m.%d')
else:
self._log.warning(f"Date not found in file's name '{filename}'.")
else:
@ -166,7 +168,8 @@ class image_classifier:
people = list()
if self.is_image(file):
self.process_metadata(file)
folder_date = self.get_file_date(file)
folder_date_time = self.get_file_date(file)
folder_date = folder_date_time.strftime(self.folder_date_format)
if folder_date:
if self.is_image(file):
people = self.find_faces(file)
@ -340,11 +343,12 @@ identify people. Filename would be used as the name for the person. Just one per
pictures to each person's folder. Be sure to have deduplication in the filesystem to avoid using \
too much storage.")
@click.option('--recursive', '-r', is_flag=True, help='Recursively search for files in the provided --directory')
@click.option('--folder-date-format', '-F', default='%Y/%Y.%m.%d', help='Format for the folder with the file date according to https://docs.python.org/3/library/datetime.html#strftime-strptime-behavior')
@click_config_file.configuration_option()
def __main__(debug_level, log_file, faces_directory, directory, no_move,
people_folder, recursive):
people_folder, recursive, folder_date_format):
return image_classifier(debug_level, log_file, faces_directory, directory, no_move,
people_folder, recursive)
people_folder, recursive, folder_date_format)
if __name__ == "__main__":