notebookVB

05 Kontrola pojištěnců/10 Čti OZP.py

@@ -0,0 +1,180 @@
+import os
+from pathlib import Path
+from blake3 import blake3
+import pymysql
+from datetime import date
+
+# =========================
+# VÝVOJOVÝ PŘEPÍNAČ
+# =========================
+
+RESET_DB = True   # !!! POZOR: smaže kapitace a pojištěnce !!!
+
+
+# =========================
+# KONFIGURACE
+# =========================
+
+BASE_PATH = Path(
+    r"U:\Dropbox\Ordinace\Dokumentace_ke_zpracování\Výpis pojištěnců"
+)
+
+DB_CONFIG = {
+    "host": "192.168.1.76",
+    "port": 3307,
+    "user": "root",
+    "password": "Vlado9674+",
+    "database": "ordinace",
+    "charset": "utf8mb4",
+    "autocommit": False,
+}
+
+# =========================
+# DB POMOCNÉ FUNKCE
+# =========================
+def reset_kapitace_tables(conn):
+    print("!!! RESET_DB=True – mažu data kapitace !!!")
+
+    with conn.cursor() as cur:
+        # kvůli FK
+        cur.execute("SET FOREIGN_KEY_CHECKS = 0")
+
+        cur.execute("TRUNCATE TABLE zp_kapitace_pojistenec")
+        cur.execute("TRUNCATE TABLE zp_kapitace_header")
+
+        cur.execute("SET FOREIGN_KEY_CHECKS = 1")
+
+    conn.commit()
+    print("✔ Kapitace resetována")
+
+def get_conn():
+    return pymysql.connect(**DB_CONFIG)
+
+def blake_exists(cur, blake):
+    cur.execute(
+        "SELECT id FROM zp_kapitace_header WHERE file_blake3 = %s",
+        (blake,),
+    )
+    return cur.fetchone()
+
+# =========================
+# PARSERY
+# =========================
+
+def parse_header(line: str):
+    return {
+        "icp_lekar": line[1:9].strip(),
+        "pocet_pojistencu": int(line[9:14]),
+        "rok": 2000 + int(line[14:16]),
+        "mesic": int(line[16:18]),
+        "den": int(line[18:20]),
+    }
+
+def parse_pojistenec(line: str):
+    return {
+        "poradi_radku": int(line[1:5]),
+        "vekova_skupina": int(line[5:7]),
+        "prijmeni": line[7:37].strip(),
+        "jmeno": line[37:61].strip(),
+        "cislo_pojistence": line[61:71].strip(),
+        "kapitace_od": date(
+            int(line[75:79]),
+            int(line[73:75]),
+            int(line[71:73]),
+        ),
+        "zp_kod": line[79:82].strip(),
+    }
+
+# =========================
+# HLAVNÍ IMPORT
+# =========================
+
+def import_file(path: Path):
+    raw = path.read_bytes()
+    blake = blake3(raw).hexdigest()
+
+    text = raw.decode("cp852")
+    lines = text.splitlines()
+
+    with get_conn() as conn:
+        cur = conn.cursor()
+
+        if blake_exists(cur, blake):
+            print(f"SKIP  {path.name} (už existuje)")
+            return
+
+        # ---- HEADER ----
+        h = parse_header(lines[0])
+        snapshot_date = date(h["rok"], h["mesic"], h["den"])
+
+        cur.execute(
+            """
+            INSERT INTO zp_kapitace_header
+            (source_file, zp_kod, icp_lekar, pocet_pojistencu,
+             rok, mesic, den, snapshot_date,
+             file_blake3, file_content, file_size, file_lines)
+            VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)
+            """,
+            (
+                path.name,
+                path.name[1:4],          # xxx ze jména Fxxx...
+                h["icp_lekar"],
+                h["pocet_pojistencu"],
+                h["rok"],
+                h["mesic"],
+                h["den"],
+                snapshot_date,
+                blake,
+                text,
+                len(raw),
+                len(lines),
+            ),
+        )
+
+        header_id = cur.lastrowid
+
+        # ---- POJISTENCI ----
+        for line in lines[1:]:
+            if not line.startswith("I"):
+                continue
+
+            p = parse_pojistenec(line)
+
+            cur.execute(
+                """
+                INSERT INTO zp_kapitace_pojistenec
+                (header_id, poradi_radku, vekova_skupina,
+                 prijmeni, jmeno, cislo_pojistence,
+                 kapitace_od, zp_kod)
+                VALUES (%s,%s,%s,%s,%s,%s,%s,%s)
+                """,
+                (
+                    header_id,
+                    p["poradi_radku"],
+                    p["vekova_skupina"],
+                    p["prijmeni"],
+                    p["jmeno"],
+                    p["cislo_pojistence"],
+                    p["kapitace_od"],
+                    p["zp_kod"],
+                ),
+            )
+
+        conn.commit()
+        print(f"IMPORTED {path.name}")
+
+# =========================
+# RUN
+# =========================
+
+def main():
+    with get_conn() as conn:
+        if RESET_DB:
+            reset_kapitace_tables(conn)
+
+    for f in sorted(BASE_PATH.glob("F*.???")):
+        import_file(f)
+
+
+if __name__ == "__main__":
+    main()