126 lines
4.2 KiB
Python
126 lines
4.2 KiB
Python
"""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()
|
|
if s in ("true", "vrai", "oui", "yes", "1", "dmz", "x"):
|
|
return True
|
|
# SANEF: 'Exposition internet', 'Exposition Internet', 'Expo indirecte'
|
|
if "exposition" in s or "expo indirecte" in s or "exposé" in s or "expose" in s:
|
|
return True
|
|
return False
|
|
|
|
|
|
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()
|