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.
main
MasterofJOKers 2 years ago
parent 38fe8c0a82
commit 12c3ac9c05

@ -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

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

@ -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())
Loading…
Cancel
Save