selecteur de fichiers

Author: alainabbas

config.json.exemple

@@ -1,4 +1,5 @@
 {
     "source_dir": "/home/source",
-    "dest_dir": "/mnt/destination"
+    "dest_dir": "/mnt/destination",
+    "patterns": ["*.pdf"]
 }

watch_and_copy.py

@@ -6,19 +6,24 @@
 Les répertoires source/destination peuvent être fournis en ligne de commande,
 sinon ils sont lus dans config.json (à côté de ce script).
 
+Des motifs de fichiers optionnels (ex. "*.pdf") permettent de ne copier que les
+fichiers dont le nom correspond (insensible à la casse). Sans motif, tous les
+fichiers sont copiés.
+
 Usage:
-    python watch_and_copy.py run     [<source_dir> <dest_dir>]   # foreground
-    python watch_and_copy.py start   [<source_dir> <dest_dir>]   # daemon
+    python watch_and_copy.py run     [<source_dir> <dest_dir> [<motif>...]]   # foreground
+    python watch_and_copy.py start   [<source_dir> <dest_dir> [<motif>...]]   # daemon
     python watch_and_copy.py stop
     python watch_and_copy.py status
-    python watch_and_copy.py restart [<source_dir> <dest_dir>]   # daemon
+    python watch_and_copy.py restart [<source_dir> <dest_dir> [<motif>...]]   # daemon
 """
 
 import os
 import sys
 import json
 import shutil
 import signal
+import fnmatch
 import logging
 from pathlib import Path
 from watchdog.observers import Observer
@@ -45,14 +50,32 @@ def setup_logging(foreground: bool = False):
     )
 
 
+def parse_patterns(raw) -> list[str]:
+    """Accepte une chaîne ("*.pdf" ou "*.pdf,*.docx") ou une liste de motifs."""
+    if not raw:
+        return []
+    if isinstance(raw, str):
+        raw = raw.split(",")
+    return [p.strip() for p in raw if p and p.strip()]
+
+
 class CopyOnCreate(FileSystemEventHandler):
-    def __init__(self, dest: Path):
+    def __init__(self, dest: Path, patterns: list[str] | None = None):
         self.dest = dest
+        self.patterns = [p.lower() for p in (patterns or [])]
+
+    def matches(self, name: str) -> bool:
+        if not self.patterns:
+            return True
+        name = name.lower()
+        return any(fnmatch.fnmatchcase(name, p) for p in self.patterns)
 
     def on_created(self, event):
         if event.is_directory:
             return
         src = Path(event.src_path if isinstance(event.src_path, str) else event.src_path.decode())
+        if not self.matches(src.name):
+            return
         dst = self.dest / src.name
         if dst.exists():
             logging.info(f"Ignoré  {src}  :  {dst} existe déjà")
@@ -87,7 +110,7 @@ def get_pid() -> int | None:
     return None
 
 
-def do_run(source: Path, dest: Path):
+def do_run(source: Path, dest: Path, patterns: list[str]):
     """Lance la surveillance en foreground (Ctrl+C pour arrêter)."""
     if not source.is_dir():
         print(f"Erreur : répertoire source introuvable : {source}")
@@ -98,10 +121,11 @@ def do_run(source: Path, dest: Path):
 
     logging.info(f"Surveillance de : {source}")
     logging.info(f"Destination     : {dest}")
+    logging.info(f"Filtre          : {', '.join(patterns) or 'tous les fichiers'}")
     logging.info("Ctrl+C pour arrêter.")
 
     observer = Observer()
-    observer.schedule(CopyOnCreate(dest), str(source), recursive=True)
+    observer.schedule(CopyOnCreate(dest, patterns), str(source), recursive=True)
     observer.start()
 
     try:
@@ -112,7 +136,7 @@ def do_run(source: Path, dest: Path):
         logging.info("Arrêté.")
 
 
-def do_start(source: Path, dest: Path):
+def do_start(source: Path, dest: Path, patterns: list[str]):
     if get_pid() is not None:
         print("Le daemon tourne déjà. Utilisez 'restart' pour le relancer.")
         sys.exit(1)
@@ -126,6 +150,7 @@ def do_start(source: Path, dest: Path):
     print(f"Démarrage du daemon…")
     print(f"  Source      : {source}")
     print(f"  Destination : {dest}")
+    print(f"  Filtre      : {', '.join(patterns) or 'tous les fichiers'}")
     print(f"  Logs        : {LOG_FILE}")
     print(f"  PID         : {PID_FILE}")
 
@@ -135,9 +160,10 @@ def do_start(source: Path, dest: Path):
     logging.info(f"Daemon démarré (PID {os.getpid()})")
     logging.info(f"Surveillance de : {source}")
     logging.info(f"Destination     : {dest}")
+    logging.info(f"Filtre          : {', '.join(patterns) or 'tous les fichiers'}")
 
     observer = Observer()
-    observer.schedule(CopyOnCreate(dest), str(source), recursive=True)
+    observer.schedule(CopyOnCreate(dest, patterns), str(source), recursive=True)
     observer.start()
 
     def on_signal(signum, frame):
@@ -180,13 +206,15 @@ def do_status():
 
 
 USAGE = (
-    f"Usage: {sys.argv[0]} run     [<source> <dest>]   # foreground\n"
-    f"       {sys.argv[0]} start   [<source> <dest>]   # daemon\n"
+    f"Usage: {sys.argv[0]} run     [<source> <dest> [<motif>...]]   # foreground\n"
+    f"       {sys.argv[0]} start   [<source> <dest> [<motif>...]]   # daemon\n"
     f"       {sys.argv[0]} stop\n"
     f"       {sys.argv[0]} status\n"
-    f"       {sys.argv[0]} restart [<source> <dest>]   # daemon\n"
+    f"       {sys.argv[0]} restart [<source> <dest> [<motif>...]]   # daemon\n"
     f"\n"
-    f"Si <source>/<dest> sont omis, ils sont lus dans {CONFIG_FILE}."
+    f"<motif> : filtre sur le nom de fichier, ex. '*.pdf' (à mettre entre guillemets).\n"
+    f"Si <source>/<dest> sont omis, ils sont lus dans {CONFIG_FILE}\n"
+    f"(clés source_dir, dest_dir et patterns optionnel)."
 )
 
 
@@ -202,11 +230,13 @@ def main():
     elif cmd == "status":
         do_status()
     elif cmd in ("run", "start", "restart"):
-        if len(sys.argv) == 4:
+        if len(sys.argv) >= 4:
             source_raw, dest_raw = sys.argv[2], sys.argv[3]
+            patterns = parse_patterns(sys.argv[4:])
         elif len(sys.argv) == 2:
             config = load_config()
             source_raw, dest_raw = config.get("source_dir"), config.get("dest_dir")
+            patterns = parse_patterns(config.get("patterns"))
             if not source_raw or not dest_raw:
                 print(f"Erreur : source_dir/dest_dir manquants dans {CONFIG_FILE}")
                 sys.exit(1)
@@ -217,11 +247,11 @@ def main():
         source = Path(source_raw).resolve()
         dest = Path(dest_raw).resolve()
         if cmd == "run":
-            do_run(source, dest)
+            do_run(source, dest, patterns)
         else:
             if cmd == "restart":
                 do_stop()
-            do_start(source, dest)
+            do_start(source, dest, patterns)
     else:
         print(USAGE)
         sys.exit(1)