Add sorted-co.py

This is a script to copy files onto an USB stick or an MP3 player in
sorted order so stupd mp3 players or radios that play files in the order
they were added to the directory instead of by sorted filename still
play the files in the right order.
This commit is contained in:
MasterofJOKers 2022-01-25 23:33:53 +01:00
parent 38fe8c0a82
commit 12c3ac9c05
3 changed files with 130 additions and 0 deletions

15
sorted-copy/setup.cfg Normal file
View File

@ -0,0 +1,15 @@
[metadata]
name = sorted-copy
version = 0.1
description = Copy files and directories recursively sorted and one by one to the destination. Mainly useful for USB sticks/MP3 players.
[options]
scripts = sorted-co.py
install_requires =
click
[flake8]
max-line-length = 120
exclude = .git,__pycache__,*.egg-info,*lib/python*
ignore = E241,E741,W503,W504

4
sorted-copy/setup.py Normal file
View File

@ -0,0 +1,4 @@
import setuptools
setuptools.setup()

111
sorted-copy/sorted-co.py Executable file
View File

@ -0,0 +1,111 @@
#!/usr/bin/python3
from pathlib import Path
import shutil
from typing import Any, Generator, Tuple, Union
import click
def _to_pathlib_path(ctx: click.Context, param: Any, value: str) \
-> Union[Path, Tuple[Path]]:
if isinstance(value, tuple):
return tuple(Path(v) for v in value)
else:
return Path(value)
def get_directories(directory: Path) -> Generator[Path, None, None]:
"""Return all directories below given directory in sorted order
This returns first all directories of a directory before returning
subdirectories of the sorted subdirectories.
"""
directories = [directory]
for d in directories:
for p in sorted(d.iterdir()):
if p.is_dir():
directories.append(p)
yield p
def get_files(directory: Path) -> Generator[Path, None, None]:
"""Traverse the given directory and its sorted subdirectory and yield files
See `get_directories()` for a hint to the order of directories traversed.
"""
if not directory.is_dir():
msg = f"get_files() needs a directory but got {directory}"
raise RuntimeError(msg)
directories = [directory]
for d in directories:
for p in sorted(d.iterdir()):
if p.is_dir():
directories.append(p)
elif p.is_file():
yield p
def relative_path(base: Path, p: Path) -> Path:
"""Return the part of `p` relative to `base` as relative Path"""
for a, b in zip(base.parts, p.parts):
if a != b:
raise ValueError('Base {} is not the base of {}'.format(a, b))
return Path('/'.join(p.parts[len(base.parts):]))
@click.command()
@click.argument('srces', nargs=-1, type=click.Path(exists=True, file_okay=True,
dir_okay=True, resolve_path=True), callback=_to_pathlib_path,
required=True)
@click.argument('dest', type=click.Path(exists=True, dir_okay=True,
file_okay=False, resolve_path=True), callback=_to_pathlib_path,
required=True)
@click.option('--verbose', is_flag=True)
def main(srces: Tuple[Path], dest: Path, verbose: bool) -> None:
"""Copy `srces` recursively and sorted one by one to `dest
This is mainly useful for USB sticks and MP3 players that take the order of
files created in the filesystem instead of the order of the sorted names to
play files.
"""
for src in srces:
if src.is_file():
# copy this file over directly
rel_path = Path(src.name)
dest_path = dest / rel_path
if verbose:
click.echo('copy {} -> {}'.format(src, dest_path))
shutil.copy(src, dest_path)
continue
if src.is_dir():
dest_path = dest / src.name
if verbose:
click.echo('mkdir {}'.format(dest_path))
dest_path.mkdir()
# create all directories first
for p in get_directories(src):
rel_path = Path(src.name) / relative_path(src, p)
dest_path = dest / rel_path
if verbose:
click.echo('mkdir {}'.format(dest_path))
dest_path.mkdir()
# copy all files over
for p in get_files(src):
rel_path = Path(src.name) / relative_path(src, p)
dest_path = dest / rel_path
if verbose:
click.echo('copy {} -> {}'.format(p, dest_path))
shutil.copy(p, dest_path)
continue
click.echo(f"Warning: Not copying {src}. Neither file nor directory")
if __name__ == '__main__':
import sys
sys.exit(main())