From 57520eaa04717a103cd89078b8639f72bc6003ce Mon Sep 17 00:00:00 2001
From: Lucas Vivo <lucas@mediahost>
Date: Wed, 7 Oct 2026 12:01:20 +0200
Subject: [PATCH] update log print colours

---
 audiobook_metadata/combine_audiobookmedia_to_m4b.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/audiobook_metadata/combine_audiobookmedia_to_m4b.py b/audiobook_metadata/combine_audiobookmedia_to_m4b.py
index 13af738..996384d 100755
--- a/audiobook_metadata/combine_audiobookmedia_to_m4b.py
+++ b/audiobook_metadata/combine_audiobookmedia_to_m4b.py
@@ -9,6 +9,11 @@ from pathlib import Path
 import time
 from typing import Any
 
+RED = "\033[91m"
+GREEN = "\033[92m"
+YELLOW = "\033[93m"
+BLUE = "\033[94m"
+RESET = "\033[0m"
 logfile=Path(f"media_{time.strftime("%Y%m%d_%H%M%S")}.log")
 
 def log_message(message):
@@ -123,7 +128,7 @@ def create_m4b(root_path: Path, book_dir: Path) -> None:
                 text=True,
             )
             if result.returncode == 234:
-                log_message(f"[ERROR] {result.returncode} retrying with aac encoder")
+                log_message(f"{YELLOW}[WARNING]{RESET} {result.returncode} retrying with aac encoder")
                 result = subprocess.run(
                     cmd_backup,
                     stdout=log,
@@ -131,12 +136,12 @@ def create_m4b(root_path: Path, book_dir: Path) -> None:
                     text=True,
                 )
         if result.returncode == 0:
-            log_message("Success, delete media files")
             output_file_tmp.rename(output_file_final)
             for media in media_files:
                 media.unlink()
+            log_message(f"{GREEN}[SUCCESS]{RESET}, delete media files")
         else:
-            log_message(f"FAILED: Code: {result.returncode}, Stderr: {result.stderr}")
+            log_message(f"{RED}[ERROR]{RESET} Code: {result.returncode}, Stderr: {result.stderr}")
 
 def find_book_dirs(root: Path) -> tuple[list[Any], int]:
     """
@@ -205,7 +210,7 @@ def main() -> None:
         try:
             create_m4b(target,book_dir)
         except Exception as exc:
-            log_message(f"[ERROR] {book_dir}: {exc}")
+            log_message(f"{RED}[ERROR]{RESET} {book_dir}: {exc}")
 
 
 if __name__ == "__main__":