007f0e0960
* Added query_command and example * Refactored QUERY_OPTIONS, added query_command, refactored verbose, #930, #931 * Added query options to debug-dump, #966 * Refactored query, #602 * Added precedence test for --load-config * Refactored handling of query options * Refactored export_photo * Removed extraneous print * Updated API_README * Updated examples
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""albums command for osxphotos CLI"""
|
|
|
|
import json
|
|
|
|
import click
|
|
import yaml
|
|
|
|
import osxphotos
|
|
from osxphotos._constants import _PHOTOS_4_VERSION
|
|
|
|
from .cli_params import DB_ARGUMENT, DB_OPTION, JSON_OPTION
|
|
from .common import get_photos_db
|
|
from .list import _list_libraries
|
|
|
|
|
|
@click.command()
|
|
@DB_OPTION
|
|
@JSON_OPTION
|
|
@DB_ARGUMENT
|
|
@click.pass_obj
|
|
@click.pass_context
|
|
def albums(ctx, cli_obj, db, json_, photos_library):
|
|
"""Print out albums found in the Photos library."""
|
|
|
|
# below needed for to make CliRunner work for testing
|
|
cli_db = cli_obj.db if cli_obj is not None else None
|
|
db = get_photos_db(*photos_library, db, cli_db)
|
|
if db is None:
|
|
click.echo(ctx.obj.group.commands["albums"].get_help(ctx), err=True)
|
|
click.echo("\n\nLocated the following Photos library databases: ", err=True)
|
|
_list_libraries()
|
|
return
|
|
|
|
photosdb = osxphotos.PhotosDB(dbfile=db)
|
|
albums = {"albums": photosdb.albums_as_dict}
|
|
if photosdb.db_version > _PHOTOS_4_VERSION:
|
|
albums["shared albums"] = photosdb.albums_shared_as_dict
|
|
|
|
# cli_obj will be None if called from pytest
|
|
if json_ or (cli_obj and cli_obj.json):
|
|
click.echo(json.dumps(albums, ensure_ascii=False))
|
|
else:
|
|
click.echo(yaml.dump(albums, sort_keys=False, allow_unicode=True))
|