94 lines
2.3 KiB
Python
94 lines
2.3 KiB
Python
import logging
|
|
from argparse import ArgumentParser, Namespace
|
|
from pathlib import Path
|
|
|
|
from photoalbum.config import DEFAULT_CONFIG_PATH, Config
|
|
from photoalbum.generate import generate
|
|
|
|
logger = logging.getLogger("photoalbum.cli")
|
|
|
|
|
|
def main() -> None:
|
|
args = parse_args()
|
|
setup_logging(args.logging)
|
|
# TODO: load config from file
|
|
config = Config()
|
|
|
|
# Call the subcommand function
|
|
match args.action:
|
|
case "generate":
|
|
cmd_generate(args, config)
|
|
case "clean":
|
|
cmd_clean(args, config)
|
|
|
|
|
|
########################################
|
|
# Command functions
|
|
def cmd_init(args: Namespace, config: Config) -> None:
|
|
"""
|
|
Generate a basic config and template files
|
|
"""
|
|
|
|
|
|
def cmd_generate(args: Namespace, config: Config) -> None:
|
|
logger.debug(f"Generating in {args.path}")
|
|
generate(config, Path(args.path))
|
|
|
|
|
|
def cmd_clean(args: Namespace, config: Config) -> None:
|
|
"""
|
|
Clean the photo album by all files that photoalbum generated
|
|
"""
|
|
pass
|
|
|
|
|
|
########################################
|
|
# CLI Util functions
|
|
def parse_args() -> Namespace:
|
|
parser = ArgumentParser()
|
|
parser.add_argument(
|
|
"--config",
|
|
"-c",
|
|
default=DEFAULT_CONFIG_PATH,
|
|
help="Path to photoalbum.config.json for the album",
|
|
)
|
|
parser.add_argument(
|
|
"--logging",
|
|
default="warning",
|
|
choices=[level.lower() for level in logging.getLevelNamesMapping().keys()],
|
|
help="Log level",
|
|
)
|
|
|
|
subcommands = parser.add_subparsers(title="subcommands")
|
|
|
|
# Generate subcommand
|
|
generate_cmd = subcommands.add_parser(
|
|
"generate",
|
|
help="Generate the HTML photo album",
|
|
)
|
|
generate_cmd.set_defaults(action="generate")
|
|
generate_cmd.add_argument(
|
|
"path",
|
|
nargs="?",
|
|
default=".",
|
|
help="Path to dir with photos in it",
|
|
)
|
|
|
|
# Clean subcommand
|
|
clean_cmd = subcommands.add_parser(
|
|
"clean",
|
|
help="Remove all generated content from the photo album directory",
|
|
)
|
|
clean_cmd.set_defaults(action="clean")
|
|
|
|
return parser.parse_args()
|
|
|
|
|
|
def setup_logging(level: str) -> None:
|
|
levels = logging.getLevelNamesMapping()
|
|
# TODO: Set up a better formatter with date/time and stuff
|
|
logging.basicConfig(level=levels[level.upper()])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|