michael/containers
1
0
Fork 0
Code Packages

photo sort working

Browse Source
This commit is contained in:
= 2024-11-23 00:06:43 -05:00
parent 0fbb347028
commit 538600380a
1 changed files with 35 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
upload-sorter/sort.py Normal file → Executable file
View File

@ -10,8 +10,8 @@ from PIL import Image, ExifTags, UnidentifiedImageError
# Set the directories
source_directories: list[Path] = [Path(directory_string).resolve() for directory_string in getenv('WATCH_DIRS', '/sync').split(':')]
recordings_directory: Path = Path(getenv('RECORDINGS_DIRECTORY', '/recordings')).resolve()
photos_directory: Path = Path(getenv('PHOTOS_DIRECTORY', '/photos')).resolve()
recordings_directory: Path = Path(getenv('RECORDINGS_DIRECTORY', '/recordings')).resolve()
photo_extensions: list[str] = ['.jpg', '.jpeg']
video_extensions: list[str] = ['.mp4', '.mov', '.avi', '.mkv']
sleep_duration: int = 30
@ -21,15 +21,16 @@ def main():
# Check the source directories exist and create the ignore files if they don't exist
for directory in source_directories:
if not directory.exists() or not directory.is_dir():
print(f"Directory {directory} does not exist or is not a directory. Exiting.")
print(f"Source directory {directory} does not exist or is not a directory. Exiting.")
sys.exit(1)
ignore_file: Path = directory.joinpath("ignored_files.json")
if not ignore_file.exists():
with open('ignored_files.json', 'w') as f:
print(f"Ignored files list {ignore_file} does not exist. Creating.")
with open(ignore_file, 'w') as f:
json.dump([], f)
# Check the destination directories exist and are writable
for directory in [recordings_directory, photos_directory]:
for directory in [photos_directory, recordings_directory]:
if not directory.exists() or not directory.is_dir() :
print(f"Destination directory {directory} does not exist or is not a directory. Exiting.")
sys.exit(1)
@ -41,13 +42,6 @@ def main():
print(f"Destination directory {directory} is not writable")
sys.exit(1)
# Ensure folders for the current year exist
current_year: str = time.strftime("%Y")
for directory in [recordings_directory, photos_directory]:
year_directory: Path = directory.joinpath(current_year)
if not year_directory.exists():
year_directory.mkdir()
# Start the main loop
while True:
for directory in source_directories:
@ -63,7 +57,7 @@ def main():
def sort_directory(directory: Path):
# Load the ignored files list
ignore_file = directory.joinpath("ignored_files.json")
with open(file= ignore_file) as f:
with open(file = ignore_file) as f:
ignore_list: list[str] = json.load(f)
# Get all files in the directory
file_list: list[Path] = [file for file in directory.iterdir() if file.is_file()]
@ -76,21 +70,21 @@ def sort_directory(directory: Path):
print(f'''
Found {str(len(file_list))} total files in directory {directory.name}.
{str(len(ignore_list))} to ignore.
{str(len(sort_list))} to sort.
{str(len(photo_list))} photos.
{str(len(video_list))} videos.
{str(len(photo_list) + len(video_list))} to sort:
{str(len(photo_list))} photos.
{str(len(video_list))} videos.
''')
# Sort the photos
print(f"Sorting {str(len(photo_list))} photos from directory {directory.name}")
ignored_files = sort_photos(photo_list, ignored_files)
ignore_list = sort_photos(photo_list, ignore_list)
# Sort the videos
print(f"Sorting {str(len(video_list))} videos from directory {directory.name}")
ignored_files = sort_videos(video_list, ignored_files)
ignore_list = sort_videos(video_list, ignore_list)
# Update the ignore_file
with open(file= ignore_file, mode='w') as f:
json.dump(ignored_files, f)
with open(file = ignore_file, mode='w') as f:
json.dump(ignore_list, f)
def sort_photos(photo_list: list[Path], ignored_files: list[str]) -> list[str]:
def sort_photos(photo_list: list[Path], ignore_list: list[str]) -> list[str]:
# Init the previous_name for collision avoidance
previous_name: str = ""
for photo in photo_list:
@ -103,30 +97,22 @@ def sort_photos(photo_list: list[Path], ignored_files: list[str]) -> list[str]:
continue
except UnidentifiedImageError:
print(f"File {photo.name} is not a valid image file. Ignoring.")
ignored_files.append(photo.name)
ignore_list.append(photo.name)
continue
# Format the date
if exif[ExifTags.Base.DateTimeOriginal] is not None and exif[ExifTags.Base.DateTimeOriginal] != '':
image_timestamp = time.strptime(
data_string = exif[ExifTags.Base.DateTimeOriginal],
format = "%Y:%m:%d %H:%M:%S"
)
image_year: str = time.strftime(
format = "%Y",
time_tuple = image_timestamp
)
image_date_string: str = time.strftime(
format = "%Y-%m-%d",
time_tuple = image_timestamp
)
image_time_string: str = time.strftime(
format = "%H-%M-%S",
time_tuple = image_timestamp
)
if ExifTags.Base.DateTimeOriginal in exif.keys():
image_date_time_string = exif[ExifTags.Base.DateTimeOriginal]
elif ExifTags.Base.DateTime in exif.keys():
image_date_time_string = exif[ExifTags.Base.DateTime]
else:
print(f"File {photo.name} does not have a valid creation date, ignoring.")
ignored_files.append(photo.name)
image_date_time_string = None
if image_date_time_string is None or image_date_time_string == '':
print(f"File {photo.name} does not have a valid exif timestamp, ignoring.")
ignore_list.append(photo.name)
continue
image_timestamp = time.strptime(image_date_time_string, "%Y:%m:%d %H:%M:%S")
image_year: str = time.strftime("%Y", image_timestamp)
image_date_string: str = time.strftime("%Y-%m-%d", image_timestamp)
image_time_string: str = time.strftime("%H-%M-%S", image_timestamp)
# Set index_count for this iteration
index_count: int = 0
# Format the new name
@ -134,20 +120,23 @@ def sort_photos(photo_list: list[Path], ignored_files: list[str]) -> list[str]:
while new_name == previous_name:
index_count += 1
new_name = f"{image_date_string}_{image_time_string}_{str(index_count).zfill(3)}.jpg"
# Move the file
new_path: Path = photos_directory.joinpath(image_year, new_name).resolve()
# Ensure the parent year folder exists
new_path.parent.mkdir(parents=True, exist_ok=True)
# Move the file
if not new_path.exists():
print(f"Moving {photo.as_posix()} to {new_path.as_posix()}")
photo.rename(new_path)
else:
print(f"File {new_name} already exists, ignoring.")
ignored_files.append(photo.name)
ignore_list.append(photo.name)
# Update the previous_name for next iteration
previous_name = new_name
return ignored_files
return ignore_list
def sort_videos(video_list: list[Path], ignored_files: list[str]) -> list[str]:
def sort_videos(video_list: list[Path], ignore_list: list[str]) -> list[str]:
print("Video sorting not implemened yet.")
return ignored_files
return ignore_list
def signal_handler(sig, frame):
print('Shutdown signal received. Exiting gracefully.')