Add align_dmz_from_ayoub: tag servers.zone_id DMZ depuis colonne DMZ Excel

tools/align_dmz_from_ayoub.py

@@ -0,0 +1,120 @@
+"""Tag servers.zone_id = DMZ depuis la colonne DMZ du fichier Ayoub.
+
+Lit sheet 'Serveurs patchables 2026' colonne DMZ (booleen True/False ou oui/non).
+Cree la zone 'DMZ' si absente (is_dmz=true).
+
+Usage:
+    python tools/align_dmz_from_ayoub.py <xlsx> [--dry-run]
+"""
+import os
+import argparse
+from sqlalchemy import create_engine, text
+
+try:
+    import openpyxl
+except ImportError:
+    print("[ERR] pip install openpyxl")
+    raise
+
+DATABASE_URL = os.getenv("DATABASE_URL_DEMO") or os.getenv("DATABASE_URL") \
+    or "postgresql://patchcenter:PatchCenter2026!@localhost:5432/patchcenter_demo"
+
+
+def is_dmz_value(v):
+    if v is None:
+        return False
+    if isinstance(v, bool):
+        return v
+    s = str(v).strip().lower()
+    return s in ("true", "vrai", "oui", "yes", "1", "dmz", "x")
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("xlsx_path")
+    parser.add_argument("--sheet", default="Serveurs patchables 2026")
+    parser.add_argument("--dry-run", action="store_true")
+    args = parser.parse_args()
+
+    engine = create_engine(DATABASE_URL)
+    print(f"[INFO] DB: {DATABASE_URL.split('@')[-1]}")
+    conn = engine.connect().execution_options(isolation_level="AUTOCOMMIT")
+
+    # Recupere / cree la zone DMZ
+    dmz = conn.execute(text(
+        "SELECT id FROM zones WHERE is_dmz=true OR LOWER(name)='dmz' ORDER BY id LIMIT 1"
+    )).fetchone()
+    if dmz:
+        dmz_id = dmz.id
+        print(f"[INFO] Zone DMZ existante (id={dmz_id})")
+    else:
+        if args.dry_run:
+            print("[DRY] Zone DMZ sera creee")
+            dmz_id = -1
+        else:
+            conn.execute(text(
+                "INSERT INTO zones (name, description, is_dmz) VALUES ('DMZ', 'Zone DMZ - patching prioritaire', true)"
+            ))
+            dmz_id = conn.execute(text(
+                "SELECT id FROM zones WHERE is_dmz=true ORDER BY id LIMIT 1"
+            )).fetchone().id
+            print(f"[INFO] Zone DMZ creee (id={dmz_id})")
+
+    wb = openpyxl.load_workbook(args.xlsx_path, data_only=True)
+    if args.sheet not in wb.sheetnames:
+        print(f"[ERR] Sheet '{args.sheet}' introuvable. Sheets: {wb.sheetnames}")
+        return
+    ws = wb[args.sheet]
+
+    header = [str(c.value).strip() if c.value else "" for c in ws[1]]
+    try:
+        idx_host = header.index("Asset Name")
+    except ValueError:
+        try:
+            idx_host = header.index("Hostname")
+        except ValueError:
+            print("[ERR] Colonne Asset Name/Hostname introuvable")
+            return
+    try:
+        idx_dmz = header.index("DMZ")
+    except ValueError:
+        print("[ERR] Colonne DMZ introuvable")
+        return
+    print(f"[INFO] idx_host={idx_host} idx_dmz={idx_dmz}")
+
+    stats = {"tagged": 0, "already_dmz": 0, "not_dmz": 0, "not_found": 0}
+
+    for row in ws.iter_rows(min_row=2, values_only=True):
+        hostname = row[idx_host] if idx_host < len(row) else None
+        if not hostname:
+            continue
+        hostname = str(hostname).strip().split(".")[0].lower()
+        if not any(c.isalpha() for c in hostname):
+            continue
+        is_dmz = is_dmz_value(row[idx_dmz] if idx_dmz < len(row) else None)
+        if not is_dmz:
+            stats["not_dmz"] += 1
+            continue
+
+        srv = conn.execute(text("SELECT id, zone_id FROM servers WHERE hostname=:h"),
+                           {"h": hostname}).fetchone()
+        if not srv:
+            stats["not_found"] += 1
+            continue
+        if srv.zone_id == dmz_id:
+            stats["already_dmz"] += 1
+            continue
+        if args.dry_run:
+            print(f"  DRY: {hostname} (zone {srv.zone_id} -> DMZ)")
+        else:
+            conn.execute(text("UPDATE servers SET zone_id=:z WHERE id=:sid"),
+                         {"z": dmz_id, "sid": srv.id})
+        stats["tagged"] += 1
+
+    conn.close()
+    print(f"\n[DONE] Tagged DMZ: {stats['tagged']} | Deja DMZ: {stats['already_dmz']} "
+          f"| Non-DMZ: {stats['not_dmz']} | Hors base: {stats['not_found']}")
+
+
+if __name__ == "__main__":
+    main()