Cleaning some music collection converters.

This commit is contained in:
James Grogan 2024-11-10 11:50:33 +00:00
parent b39807db44
commit c09179f5da
8 changed files with 94 additions and 80 deletions

14
src/guile/hello-test.scm Normal file
View file

@ -0,0 +1,14 @@
(use-modules (srfi srfi-64)
(hello))
(test-begin "harness")
(test-equal "test-hello"
"Hello world!\n"
(hi))
(test-equal "test-named-hello"
"Hello bob!\n"
(hi "bob"))
(test-end "harness")

16
src/guile/hello.scm Normal file
View file

@ -0,0 +1,16 @@
(define-module (hello))
(define GREETING_PREFIX "Hello ")
(define GREETING_SUFFIX "!\n")
(define DEFAULT_NAME "world")
(define-public hi
"Says hello to the named person"
(lambda* (#:optional name)
(string-append GREETING_PREFIX (addressee name) GREETING_SUFFIX)))
(define addressee
(lambda (name)
(if name
name
DEFAULT_NAME)))

View file

@ -1,48 +1,37 @@
from pathlib import Path from pathlib import Path
from functools import partial from functools import partial
from typing import NamedTuple, Callable from typing import Callable
import logging import logging
from .tasks import Task from jgutils.filesystem import get_files_recursive
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class ConversionConfig(NamedTuple): def _get_converted_path(input_path: Path,
input_dir: Path output_dir: Path,
output_dir: Path output_ext: str):
input_ext: str
output_ext: str
def _get_converted_path(input_path: Path, config: ConversionConfig):
""" """
Return the path you would obtain if you moved the file in the input Return the path you would obtain if you moved the file in the input
path to the output directory in the config and changed its extension. path to the output directory in the config and changed its extension.
""" """
relative_path = input_path.relative_to(config.input_dir) output_filename = str(input_path.stem) + f".{output_ext}"
output_filename = str(input_path.stem) + f".{config.output_ext}" return output_dir / input_path.parent / output_filename
return output_dir / relative_path.parent / output_filename
def convert(input_dir: Path,
output_dir: Path,
input_ext: str,
output_ext: str,
conversion_func: Callable):
def get_unconverted_files( logger.info("Converting files in %s to %s", input_dir, output_dir)
input_files: list[Path], config: ConversionConfig
) -> list[Path]:
output_files = [_get_converted_path(f, config) for f in input_files]
conversion_pairs = [] input_files = get_files_recursive(input_dir.resolve(), [input_ext])
for input_file, output_file in zip(input_files, output_files): output_files = [_get_converted_path(f.relative_to(input_dir),
if not output_file.exists(): output_dir,
conversion_pairs.append((input_file, output_file)) output_ext) for f in input_files]
return conversion_pairs conversion_pairs = [io_paths for io_paths in
zip(input_files, output_files) if not io_paths[1].exists()]
run_tasks(conversion_func, conversion_pairs)
def convert(config: ConversionConfig, conversion_func: Callable):
logger.info("Converting files in %s", config.input_dir)
logger.info("Writing output to: %s", config.output_dir)
input_files = get_files_recursive(config.input_dir.resolve(), [config.input_ext])
output_files = get_uncoverted_files(input_files, config)
run_tasks(conversion_func, zip(input_files, output_files))

View file

@ -7,25 +7,29 @@ def get_files_recursive(path: Path, extensions: list[str] | None) -> list[Path]:
if not extensions: if not extensions:
return list(path.rglob("*.*")) return list(path.rglob("*.*"))
else: else:
ret = [] paths_list = [list(path.rglob(f"*.{ext}")) for ext in extensions]
for ext in extensions: return [p for paths in paths_list for p in paths]
ret.append(list(search_path.rglob(f"*.{ext}")))
return ret def get_empty_dirs(path: Path) -> list[Path]:
return [curdir
for curdir, subdirs, files in os.walk(path)
if len(subdirs)==0 and len(files) == 0]
def _delete_empty_dirs(path: Path) -> bool: def _delete_empty_dirs(path: Path) -> bool:
found_empty = False empty_dirs = get_empty_dirs(path)
for curdir, subdirs, files in os.walk(path): map(os.rmdir, empty_dirs)
if len(subdirs) == 0 and len(files) == 0: return bool(empty_dirs)
found_empty = True
os.rmdir(curdir)
return found_empty
def delete_empty_dirs_recursive(path: Path): def delete_empty_dirs_recursive(path: Path):
found_empty = True found_empty = True
while found_empty: while found_empty:
found_empty = _delete_empty_dirs(path) found_empty = _delete_empty_dirs(path)
def replace_filename(path: Path, replacement: str) -> Path: def replace_filename(path: Path, replacement: str) -> Path:
return path.parent / (replacement + path.suffix) return path.parent / (replacement + path.suffix)
def move_file_new_name(src: Path, dst: Path, name: str):
os.makedirs(dst, exist_ok=True)
shutil.move(src, dst / f"{name}{src.suffix}")

View file

@ -12,7 +12,7 @@ def cli_music_read(args):
write_model(collection) write_model(collection)
def cli_music_refresh(args): def cli_music_refresh(args):
music.collection.refresh(args.collection.resolve(), music.refresh(args.collection.resolve(),
args.work_dir.resolve()) args.work_dir.resolve())
def main_cli(): def main_cli():

View file

@ -0,0 +1 @@
from .collection import refresh # NOQA

View file

@ -6,7 +6,7 @@ from pathlib import Path
from tinytag import TinyTag from tinytag import TinyTag
from pydantic import BaseModel from pydantic import BaseModel
from jgutils.filesystem import get_files_recursive from jgutils.filesystem import get_files_recursive, move_file_new_name
from jgutils import converters from jgutils import converters
from .models import Song, Album, Artist from .models import Song, Album, Artist
@ -21,12 +21,6 @@ class MusicCollection(BaseModel):
artists: list[Artist] = [] artists: list[Artist] = []
def get_artist(self, name:str) -> Artist | None:
for artist in self.artists:
if artist.name == name:
return artist
return None
_DEFAULT_EXTENSIONS = ["flac", _DEFAULT_EXTENSIONS = ["flac",
"mp3"] "mp3"]
@ -40,17 +34,18 @@ def read(input_dir: Path, extensions: list[str] = None) -> MusicCollection:
if not extensions: if not extensions:
extensions = _DEFAULT_EXTENSIONS extensions = _DEFAULT_EXTENSIONS
collection = MusicCollection() tags = [TingTag.get(f) for f in get_files_recursive(input_dir, extensions)]
for eachFile in get_files_recursive(input_dir, extensions): tags_with_titles = [t for t in tags if t.title]
tag = TinyTag.get(eachFile)
if not tag.title:
logger.warn("Found tag with no title, skipping: %s", tag)
continue
artist = collection.get_artist(tag.artist) artists = []
if not artist: for tag in tags_with_titles:
matching_artists = filter(lambda x: x.name == tag.artist, artists)
if not matching_artists:
artist = Artist(name=tag.artist) artist = Artist(name=tag.artist)
collection.artists.append(artist) artists.append(artist)
else:
artist = next(matching_artists)
if tag.album: if tag.album:
album = artist.get_album(tag.album) album = artist.get_album(tag.album)
@ -65,7 +60,15 @@ def read(input_dir: Path, extensions: list[str] = None) -> MusicCollection:
else: else:
artist.songs.append(song) artist.songs.append(song)
return collection return MusicCollection(artists=artists)
def _refresh_collection_dirs(tags: list, collection_dir: Path):
for tag, song_file in tags:
artist_dir = collection_dir / tag.artist
if tag.album:
move_file_new_name(song_file, artist_dir / tag.album, tag.title)
else:
move_file_new_name(song_file, artist_dir, tag.title)
def refresh(collection_dir: Path, work_dir: Path): def refresh(collection_dir: Path, work_dir: Path):
@ -76,29 +79,14 @@ def refresh(collection_dir: Path, work_dir: Path):
""" """
ext = "flac" ext = "flac"
files = get_files_recursive(collection_dir, [ext]) tags = [(TinyTag.get(f), f) for f in get_files_recursive(collection_dir, [ext])]
tags_with_titles = [t for t in tags if t[0].title]
for eachFile in files: _refresh_collection_dirs(tags_with_titles, collection_dir)
tag = TinyTag.get(eachFile)
if not tag.title:
logger.warn("Found tag with no title, skipping: %s", tag)
continue
artist_dir = input_dir / tag.artist
os.makedirs(artist_dir, exist_ok=True)
if tag.album:
os.makedirs(artist_dir / tag.album, exist_ok=True)
shutil.move(eachFile, artist_dir / tag.album / f"{tag.title}.{ext}")
else:
shutil.move(eachFile, artist_dir / f"{tag.title}.{ext}")
# Get any files not existing on the sync target, convert them to a suitable format # Get any files not existing on the sync target, convert them to a suitable format
# and push them to the target. # and push them to the target.
config = converters.ConversionConfig( converters.convert(collection_dir, work_dir, "flac", "mp3")
collection_dir, work_dir, "flac", "mp3"
)
converters.convert(config)

View file

@ -1,3 +1,5 @@
from pathlib import Path
from pydantic import BaseModel from pydantic import BaseModel
class Song(BaseModel): class Song(BaseModel):