diff --git a/audiobook_metadata/metadata_audiobooks_library.py b/audiobook_metadata/metadata_audiobooks_library.py index 4e302fa..f437644 100755 --- a/audiobook_metadata/metadata_audiobooks_library.py +++ b/audiobook_metadata/metadata_audiobooks_library.py @@ -11,31 +11,10 @@ import sqlite3 from collections import defaultdict from concurrent.futures import ThreadPoolExecutor, as_completed from pathlib import Path - -RED = "\033[91m" -GREEN = "\033[92m" -YELLOW = "\033[93m" -BLUE = "\033[94m" -RESET = "\033[0m" -TRACE = 5 - - -class CustomLogger(logging.Logger): - def trace(self, msg: str, *args, **kwargs) -> None: - if self.isEnabledFor(TRACE): - self._log(TRACE, msg, args, **kwargs) - +from serverlibrary import CustomLogger logging.setLoggerClass(CustomLogger) logger: CustomLogger = logging.getLogger(__name__) # type: ignore[assignment] -LOG_LEVELS = { - "TRACE": TRACE, - "DEBUG": logging.DEBUG, - "INFO": logging.INFO, - "WARNING": logging.WARNING, - "ERROR": logging.ERROR, - "CRITICAL": logging.CRITICAL, -} metadata_fields = ["artist", "album", "composer", "comment", "genre", "grouping", "album artist", "description"] audio_extensions = {"*.m4b", "*.m4a", "*.mp3", "*.flac", "*.ogg", "*.opus"} @@ -427,7 +406,7 @@ class Audiobook: output_file_final = self.bookPath / f"{self.bookPath.name}.m4b" if output_file_final.exists(): - logger.info(f"{RED}[SKIP]{RESET} {output_file_final}") + logger.info(f"{red}[SKIP]{reset} {output_file_final}") logger.info("Book already present") return @@ -586,7 +565,7 @@ class Audiobook: if result.returncode == 234: logger.warning( - f"{YELLOW}[WARNING]{RESET} - {self.bookPath.name} - {result.returncode} retrying with aac encoder" + f"{yellow}[WARNING]{reset} - {self.bookPath.name} - {result.returncode} retrying with aac encoder" ) result = subprocess.run( cmd_backup, @@ -609,11 +588,11 @@ class Audiobook: self.changed = False self.size = Path(output_file_final).stat().st_size logger.info( - f"{GREEN}[SUCCESS]{RESET} - {self.bookPath.name} - delete media files" + f"{green}[SUCCESS]{reset} - {self.bookPath.name} - delete media files" ) else: logger.error( - f"{RED}[ERROR]{RESET} - {self.bookPath.name} - Code: {result.returncode}, Stderr: {result.stderr}" + f"{red}[ERROR]{reset} - {self.bookPath.name} - Code: {result.returncode}, Stderr: {result.stderr}" ) def write_metadata(self, keys=None, dry_run=False):