Update watcher.py
This commit is contained in:
parent
20f24d7a42
commit
79666caeeb
35
watcher.py
35
watcher.py
@ -6,6 +6,8 @@ from watchdog.observers import Observer
|
||||
from watchdog.events import FileSystemEventHandler
|
||||
|
||||
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):
|
||||
def __init__(self):
|
||||
@ -36,7 +38,14 @@ class NewBookHandler(FileSystemEventHandler):
|
||||
|
||||
m4b_files = [f for f in files if f.lower().endswith(".m4b") and not f.startswith("._")]
|
||||
if not m4b_files:
|
||||
print(f"⚠️ Skipping {path} (no .m4b files detected)")
|
||||
if DEBUG:
|
||||
print(f"⚠️ Skipping {path} (no .m4b files detected)")
|
||||
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)
|
||||
@ -47,10 +56,10 @@ class NewBookHandler(FileSystemEventHandler):
|
||||
|
||||
# Run your main.py script
|
||||
try:
|
||||
subprocess.run(
|
||||
["python3", "main.py", "-da", path],
|
||||
check=True
|
||||
)
|
||||
cmd = ["python3", "main.py", "-da", path]
|
||||
if MERGE_ALL:
|
||||
cmd.append("--all") # pass flag to main.py
|
||||
subprocess.run(cmd, check=True)
|
||||
print(f"✅ Finished processing {book_name}")
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(f"❌ Error processing {book_name}: {e}")
|
||||
@ -62,14 +71,14 @@ if __name__ == "__main__":
|
||||
observer.start()
|
||||
print(f"👀 Watching for new book folders in {WATCH_DIR} ...")
|
||||
|
||||
# --- Startup scan: just print existing folders ---
|
||||
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("._")]
|
||||
if m4b_files:
|
||||
print(f"📁 Existing book folder detected (startup scan): {root}")
|
||||
for f in m4b_files:
|
||||
print(" ", f)
|
||||
# Do NOT call process_new_dir -> prevents running main.py on existing files
|
||||
# --- Startup scan: just print existing folders if DEBUG is enabled ---
|
||||
if DEBUG:
|
||||
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("._")]
|
||||
if m4b_files:
|
||||
print(f"📁 Existing book folder detected (startup scan): {root}")
|
||||
for f in m4b_files:
|
||||
print(" ", f)
|
||||
|
||||
try:
|
||||
while True:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user