wpsrt/wpsrt/cli.py

from pathlib import Path

import click

from .wallpapers import move_wallpaper, scan_directory


@click.command()
@click.argument(
    "source",
    type=click.Path(exists=True, file_okay=False, dir_okay=True),
    default=Path("~/Pictures/wallpapers").expanduser(),
)
@click.argument(
    "target",
    type=click.Path(exists=False, file_okay=False, dir_okay=True),
    default=Path("~/Pictures/wallpapers/by-resolution").expanduser(),
)
def sort(source: Path, target: Path):
    source = Path(source)
    target = Path(target)
    if not target.exists():
        target.mkdir()
    for wallpaper in scan_directory(source):
        filename, (xres, yres) = wallpaper
        if filename.is_relative_to(target):
            continue
        new_filename = move_wallpaper(
            filename, target / f"{xres}x{yres}/{filename.name}"
        )
        click.echo(new_filename)


if __name__ == "__main__":
    sort(None, None)