tyler/AudioBook-Compile
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update watcher.py

Browse Source
This commit is contained in:
TylerCG 2025-09-13 21:10:08 -04:00
parent 20f24d7a42
commit 79666caeeb
1 changed files with 22 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
watcher.py
View File

@ -6,6 +6,8 @@ from watchdog.observers import Observer
from watchdog.events import FileSystemEventHandler from watchdog.events import FileSystemEventHandler
WATCH_DIR = "/data" # Mount your host Books folder here WATCH_DIR = "/data" # Mount your host Books folder here
DEBUG = os.getenv("DEBUG", "false").lower() == "true"
MERGE_ALL = os.getenv("MERGE_ALL", "true").lower() == "true"
class NewBookHandler(FileSystemEventHandler): class NewBookHandler(FileSystemEventHandler):
def __init__(self): def __init__(self):
@ -36,9 +38,16 @@ class NewBookHandler(FileSystemEventHandler):
m4b_files = [f for f in files if f.lower().endswith(".m4b") and not f.startswith("._")] m4b_files = [f for f in files if f.lower().endswith(".m4b") and not f.startswith("._")]
if not m4b_files: if not m4b_files:
if DEBUG:
print(f"⚠️ Skipping {path} (no .m4b files detected)") print(f"⚠️ Skipping {path} (no .m4b files detected)")
return return
# Skip if only 1 file and MERGE_ALL is enabled
if MERGE_ALL and len(m4b_files) == 1:
if DEBUG:
print(f"⏭️ Skipping {path} (only 1 .m4b file, nothing to merge)")
return
book_name = os.path.basename(path) book_name = os.path.basename(path)
print(f"📕 New book detected: {book_name} at {path}") print(f"📕 New book detected: {book_name} at {path}")
print("Contains .m4b files:") print("Contains .m4b files:")
@ -47,10 +56,10 @@ class NewBookHandler(FileSystemEventHandler):
# Run your main.py script # Run your main.py script
try: try:
subprocess.run( cmd = ["python3", "main.py", "-da", path]
["python3", "main.py", "-da", path], if MERGE_ALL:
check=True cmd.append("--all") # pass flag to main.py
) subprocess.run(cmd, check=True)
print(f"✅ Finished processing {book_name}") print(f"✅ Finished processing {book_name}")
except subprocess.CalledProcessError as e: except subprocess.CalledProcessError as e:
print(f"❌ Error processing {book_name}: {e}") print(f"❌ Error processing {book_name}: {e}")
@ -62,14 +71,14 @@ if __name__ == "__main__":
observer.start() observer.start()
print(f"👀 Watching for new book folders in {WATCH_DIR} ...") print(f"👀 Watching for new book folders in {WATCH_DIR} ...")
# --- Startup scan: just print existing folders --- # --- Startup scan: just print existing folders if DEBUG is enabled ---
if DEBUG:
for root, dirs, files in os.walk(WATCH_DIR): for root, dirs, files in os.walk(WATCH_DIR):
m4b_files = [f for f in files if f.lower().endswith(".m4b") and not f.startswith("._")] m4b_files = [f for f in files if f.lower().endswith(".m4b") and not f.startswith("._")]
if m4b_files: if m4b_files:
print(f"📁 Existing book folder detected (startup scan): {root}") print(f"📁 Existing book folder detected (startup scan): {root}")
for f in m4b_files: for f in m4b_files:
print(" ", f) print(" ", f)
# Do NOT call process_new_dir -> prevents running main.py on existing files
try: try:
while True: while True: