57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import os
|
|
import subprocess
|
|
import time
|
|
from watchdog.observers import Observer
|
|
from watchdog.events import FileSystemEventHandler
|
|
|
|
WATCH_DIR = "/data" # Mount host Books folder here (Books:/data)
|
|
|
|
class NewBookHandler(FileSystemEventHandler):
|
|
def on_created(self, event):
|
|
if event.is_directory:
|
|
self.process_new_dir(event.src_path)
|
|
|
|
def on_moved(self, event):
|
|
if event.is_directory:
|
|
self.process_new_dir(event.dest_path)
|
|
|
|
def process_new_dir(self, path):
|
|
# Only process if directory contains at least one .m4b file
|
|
try:
|
|
m4b_files = [
|
|
f for f in os.listdir(path)
|
|
if f.lower().endswith(".m4b") and not f.startswith("._")
|
|
]
|
|
except FileNotFoundError:
|
|
return # directory might have been deleted quickly
|
|
|
|
if not m4b_files:
|
|
print(f"⚠️ Skipping {path} (no .m4b files found yet)")
|
|
return
|
|
|
|
book_name = os.path.basename(path)
|
|
print(f"📕 New book detected: {book_name} at {path}")
|
|
|
|
try:
|
|
subprocess.run(
|
|
["python3", "main.py", "-da", path],
|
|
check=True
|
|
)
|
|
print(f"✅ Finished processing {book_name}")
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"❌ Error processing {book_name}: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
observer = Observer()
|
|
event_handler = NewBookHandler()
|
|
observer.schedule(event_handler, WATCH_DIR, recursive=True)
|
|
observer.start()
|
|
print(f"👀 Watching for new book folders in {WATCH_DIR} ...")
|
|
|
|
try:
|
|
while True:
|
|
time.sleep(1)
|
|
except KeyboardInterrupt:
|
|
observer.stop()
|
|
observer.join()
|