87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
""" Export all photos to specified directory using album names as folders
|
|
If file has been edited, also export the edited version,
|
|
otherwise, export the original version
|
|
This will result in duplicate photos if photo is in more than album """
|
|
|
|
import os.path
|
|
import pathlib
|
|
import sys
|
|
|
|
import click
|
|
from pathvalidate import is_valid_filepath, sanitize_filepath
|
|
|
|
import osxphotos
|
|
|
|
|
|
@click.command()
|
|
@click.argument("export_path", type=click.Path(exists=True))
|
|
@click.option(
|
|
"--default-album",
|
|
help="Default folder for photos with no album. Defaults to 'unfiled'",
|
|
default="unfiled",
|
|
)
|
|
@click.option(
|
|
"--library-path",
|
|
help="Path to Photos library, default to last used library",
|
|
default=None,
|
|
)
|
|
@click.option(
|
|
"--edited",
|
|
help="Also export edited versions of photos (default is originals only)",
|
|
is_flag=True,
|
|
default=False,
|
|
)
|
|
def export(export_path, default_album, library_path, edited):
|
|
"""Export all photos, organized by album"""
|
|
export_path = os.path.expanduser(export_path)
|
|
library_path = os.path.expanduser(library_path) if library_path else None
|
|
|
|
if library_path is not None:
|
|
photosdb = osxphotos.PhotosDB(library_path)
|
|
else:
|
|
photosdb = osxphotos.PhotosDB()
|
|
|
|
photos = photosdb.photos()
|
|
|
|
for p in photos:
|
|
if not p.ismissing:
|
|
albums = p.albums
|
|
if not albums:
|
|
albums = [default_album]
|
|
for album in albums:
|
|
click.echo(f"exporting {p.original_filename} in album {album}")
|
|
|
|
# make sure no invalid characters in destination path (could be in album name)
|
|
album_name = sanitize_filepath(album, platform="auto")
|
|
|
|
# create destination folder, if necessary, based on album name
|
|
dest_dir = os.path.join(export_path, album_name)
|
|
|
|
# verify path is a valid path
|
|
if not is_valid_filepath(dest_dir, platform="auto"):
|
|
sys.exit(f"Invalid filepath {dest_dir}")
|
|
|
|
# create destination dir if needed
|
|
if not os.path.isdir(dest_dir):
|
|
os.makedirs(dest_dir)
|
|
|
|
filename = p.original_filename
|
|
# export the photo but only if --edited, photo has adjustments, and
|
|
# path_edited is not None (can be None if edited photo is missing)
|
|
if edited and p.hasadjustments and p.path_edited:
|
|
# export edited version
|
|
# use original filename with _edited appended but make sure suffix is
|
|
# same as edited file
|
|
edited_filename = f"{pathlib.Path(filename).stem}_edited{pathlib.Path(p.path_edited).suffix}"
|
|
exported = p.export(dest_dir, edited_filename, edited=True)
|
|
click.echo(f"Exported {edited_filename} to {exported}")
|
|
# export unedited version
|
|
exported = p.export(dest_dir, filename)
|
|
click.echo(f"Exported {filename} to {exported}")
|
|
else:
|
|
click.echo(f"Skipping missing photo: {p.original_filename}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
export() # pylint: disable=no-value-for-parameter
|