From 37f9c39a60af52bb80bcb7ec948916b3426081c9 Mon Sep 17 00:00:00 2001
From: UNFR <unfr@noreply.localhost>
Date: Thu, 2 Apr 2026 07:58:14 +0000
Subject: [PATCH] Ajouter main.py

Commit initila avec correction de l'affichage des URL
---
 main.py | 155 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 155 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..d8b2ac7
--- /dev/null
+++ b/main.py
@@ -0,0 +1,155 @@
+#!/usr/bin/env python3
+import os
+import sys
+import urllib.error
+import urllib.parse
+import urllib.request
+import xml.etree.ElementTree as ET
+
+# NZBGet Exit Codes
+POSTPROCESS_SUCCESS = 93
+POSTPROCESS_ERROR = 94
+POSTPROCESS_NONE = 95
+
+# Configuration API
+API_BASE_URL = "https://unfr.pw/api_verif_hs.php"
+
+def log_info(msg):
+    print(f"[INFO] {msg}")
+
+
+def log_warn(msg):
+    print(f"[WARNING] {msg}")
+
+
+def log_error(msg):
+    print(f"[ERROR] {msg}")
+
+
+def extract_info(nzb_path):
+    log_info(f"Lecture NZB: {nzb_path}")
+
+    with open(nzb_path, "rb") as f:
+        xml_data = f.read()
+
+    root = ET.fromstring(xml_data)
+    title = None
+    randomid = None
+    ns = {"nzb": "http://www.newzbin.com/DTD/2003/nzb"}
+    meta_nodes = root.findall(".//nzb:head/nzb:meta", ns)
+    if not meta_nodes:
+        meta_nodes = root.findall(".//head/meta")
+
+    for meta in meta_nodes:
+        meta_type = (meta.get("type") or "").strip().lower()
+        meta_value = (meta.text or "").strip()
+        if not meta_value:
+            continue
+        if meta_type == "title":
+            title = meta_value
+        elif meta_type == "tag":
+            randomid = meta_value
+
+    return title, randomid
+
+
+def call_markhs(api_key, randomid, extra_info=""):
+    params = {"apikey": api_key, "action": "markhs", "randomid": randomid}
+    if extra_info:
+        params["extra_info"] = extra_info
+
+    nzbget_version = os.environ.get("NZBOP_VERSION", "unknown")
+    user_agent = f"nzbget/{nzbget_version}"
+    query = urllib.parse.urlencode(params, quote_via=urllib.parse.quote)
+    url = f"{API_BASE_URL}?{query}"
+    log_url = url.replace(api_key, "***")
+    
+    # Décodage de l'URL uniquement pour l'affichage dans les logs
+    log_info(f"Appel API: {urllib.parse.unquote(log_url)}")
+
+    try:
+        req = urllib.request.Request(
+            url,
+            method="GET",
+            headers={
+                "User-Agent": user_agent,
+                "Accept": "*/*",
+                "Connection": "close",
+            },
+        )
+        with urllib.request.urlopen(req, timeout=30) as resp:
+            body = resp.read().decode("utf-8", errors="replace")
+            log_info(f"Reponse API ({resp.status}): {body[:500]}")
+            return resp.status == 200
+    except urllib.error.HTTPError as e:
+        body = e.read().decode("utf-8", errors="replace")
+        log_error(f"Erreur HTTP {e.code}: {e.reason} | body={body[:500]}")
+    except urllib.error.URLError as e:
+        log_error(f"Erreur reseau: {e.reason}")
+    except Exception as e:
+        log_error(f"Erreur inattendue: {e}")
+    return False
+
+
+def main():
+    if "NZBOP_EXTENSIONS" not in os.environ:
+        print("[ERROR] Ce script necessite au moins NZBGet v18.0")
+        return POSTPROCESS_ERROR
+    
+    print("[INFO] Script démarré - python " + str(sys.version_info))
+    sys.stdout.flush()
+
+    # Check nzbget.conf options
+    required_options = (
+        "NZBPO_UNFRAPIKEY",
+    )
+
+    for optname in required_options:
+      if optname not in os.environ:
+          print(
+              "[ERROR] Option %s est manquante dans la configuration du script. Veuillez vérifier"
+              % optname[6:]
+          )
+          return POSTPROCESS_ERROR
+
+    nzb_status = os.environ.get("NZBPP_STATUS", "")
+    nzb_path = os.environ.get("NZBPP_QUEUEDFILE", "")
+    api_key = os.environ.get("NZBPO_UNFRAPIKEY")
+
+    log_info(f"NZBStatus: {nzb_status}")
+    log_info(f"NZBPath: {nzb_path}")
+
+    if nzb_status not in ("FAILURE/PAR", "FAILURE/HEALTH"):
+        log_info("NZBStatus non concerné, rien à signaler.")
+        return POSTPROCESS_NONE
+
+    if not (nzb_path and os.path.isfile(nzb_path)):
+        log_error("Impossible de trouver le fichier NZB dans NZBPP_QUEUEDFILE.")
+        return POSTPROCESS_ERROR
+
+    try:
+        title, randomid = extract_info(nzb_path)
+    except Exception as e:
+        log_error(f"Erreur parsing NZB: {e}")
+        return POSTPROCESS_ERROR
+
+    if title:
+        log_info(f"Title: {title}")
+
+    if not randomid:
+        log_info('Aucune balise <meta type="tag"> trouvée dans le NZB.')
+        return POSTPROCESS_NONE
+
+    log_info(f"Random ID: {randomid}")
+    extra_info = f"NZBGet: Echec verification/reparation PAR | job={title}"
+    ok = call_markhs(api_key, randomid, extra_info)
+    if ok:
+        log_info("Release marquée HS avec succes.")
+        return POSTPROCESS_SUCCESS
+    else:
+        log_warn("Impossible de marquer la release HS.")
+        return POSTPROCESS_ERROR
+
+
+if __name__ == "__main__":
+    sys.exit(main())
\ No newline at end of file