notebook

11Očkování jeden list jedna látka.py

@@ -0,0 +1,162 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+from pathlib import Path
+import time
+import fdb
+import pandas as pd
+from openpyxl import load_workbook
+from openpyxl.worksheet.table import Table, TableStyleInfo
+from openpyxl.styles import Font, PatternFill, Alignment
+from openpyxl.utils import get_column_letter
+import re
+
+
+# map "Látka" names to sheet names
+sheet_mapping = {
+    "Chřipka": ["Influvac", "Vaxigrip", "Fluarix"],
+    "HepA": ["Avaxim", "Havrix"],
+    "HepB": ["Engerix"],
+    "HepA+B": ["Twinrix"],
+    "COVID": ["Comirnaty"],
+    "Klíšťovka": ["FSME"]
+}
+
+# ================== Výstupní cesta ==================
+BASE_DIR = Path(r"..")  # uprav dle potřeby
+timestamp = time.strftime("%Y-%m-%d %H-%M-%S")
+xlsx_name = f"Pacienti očkování {timestamp}.xlsx"
+xlsx_path = BASE_DIR / xlsx_name
+
+# ================== Připojení k DB ==================
+con = fdb.connect(
+    host='192.168.1.4',
+    database=r'z:\\MEDICUS 3\\data\\medicus.FDB',
+    user='sysdba',
+    password='masterkey',
+    charset='WIN1250'
+)
+
+# ================== SQL dotaz ==================
+sql = """
+SELECT
+    kar.rodcis    AS "Rodné číslo",
+    kar.prijmeni  AS "Příjmení",
+    kar.jmeno     AS "Jméno",
+    ockzaz.datum  AS "Datum očkování",
+    ockzaz.kodmz  AS "Kód MZ",
+    ockzaz.poznamka AS "Šarže",
+    ockzaz.latka  AS "Látka",
+    ockzaz.nazev  AS "Název",
+    ockzaz.expire AS "Expirace",
+    (
+      SELECT LIST(l.kod, ', ')
+      FROM lecd l
+      WHERE l.rodcis = kar.rodcis
+        AND l.datose = CAST(ockzaz.datum AS DATE)
+    ) AS "LECD kódy (ten den)",
+    (
+      SELECT LIST(d.kod, ', ')
+      FROM dokladd d
+      WHERE d.rodcis = kar.rodcis
+        AND d.datose = CAST(ockzaz.datum AS DATE)
+    ) AS "Výkony (ten den)"
+FROM registr
+JOIN kar     ON registr.idpac = kar.idpac
+JOIN ockzaz  ON registr.idpac = ockzaz.idpac
+WHERE
+    registr.datum_zruseni IS NULL
+    AND kar.vyrazen <> 'A'
+    AND kar.rodcis IS NOT NULL
+    AND idicp <> 0
+    AND EXTRACT(YEAR FROM ockzaz.datum)=2025
+ORDER BY ockzaz.datum DESC
+"""
+
+# ================== Načtení do DataFrame ==================
+df = pd.read_sql(sql, con)
+con.close()
+
+# Typy sloupců
+for col in ["Kód MZ", "Šarže", "Rodné číslo", "Látka", "Název", "Příjmení", "Jméno", "LECD kódy","Výkony"]:
+    if col in df.columns:
+        df[col] = df[col].astype("string")
+
+for dcol in ["Datum očkování", "Expirace"]:
+    if dcol in df.columns:
+        df[dcol] = pd.to_datetime(df[dcol], errors="coerce")
+
+# ================== Uložení do Excelu ==================
+with pd.ExcelWriter(xlsx_path, engine="openpyxl") as writer:
+    # --- 1️⃣ Custom groups ---
+    for sheet_name, latky in sheet_mapping.items():
+        mask = df["Látka"].isin(latky)
+        subdf = df[mask]
+        if not subdf.empty:
+            safe_name = re.sub(r'[\\/*?:[\]]', '', str(sheet_name))[:31]
+            subdf.to_excel(writer, index=False, sheet_name=safe_name)
+
+    # # --- 2️⃣ Optionally add sheet for "ostatní" ---
+    # used_latky = {l for group in sheet_mapping.values() for l in group}
+    # remaining = df[~df["Látka"].isin(used_latky)]
+    # if not remaining.empty:
+    #     remaining.to_excel(writer, index=False, sheet_name="Ostatní")
+
+# ================== Formátování ==================
+wb = load_workbook(xlsx_path)
+
+def autosize_columns(ws):
+    for col_idx in range(1, ws.max_column + 1):
+        col_letter = get_column_letter(col_idx)
+        max_len = 0
+        for cell in ws[col_letter]:
+            val = "" if cell.value is None else str(cell.value)
+            if len(val) > max_len:
+                max_len = len(val)
+        ws.column_dimensions[col_letter].width = min(max(12, max_len + 2), 60)
+
+def style_table(ws, table_name: str):
+    max_row = ws.max_row
+    max_col = ws.max_column
+    if max_col == 0:
+        return
+
+    # Hlavička vzhled
+    header_fill = PatternFill("solid", fgColor="D9E1F2")
+    for cell in ws[1]:
+        cell.font = Font(bold=True)
+        cell.fill = header_fill
+        cell.alignment = Alignment(vertical="center")
+
+    # Freeze top row
+    ws.freeze_panes = "A2"
+
+    # Tabulka jen pokud jsou data
+    if max_row < 2:
+        autosize_columns(ws)
+        return
+
+    ref = f"A1:{get_column_letter(max_col)}{max_row}"
+    tbl = Table(displayName=table_name, ref=ref)
+    tbl.tableStyleInfo = TableStyleInfo(
+        name="TableStyleMedium9", showRowStripes=True, showColumnStripes=False
+    )
+    ws.add_table(tbl)
+    autosize_columns(ws)
+
+def format_dates(ws, columns_names):
+    header = [c.value for c in ws[1]]
+    date_cols = [header.index(name) + 1 for name in columns_names if name in header]
+    for col_idx in date_cols:
+        for row in ws.iter_rows(min_row=2, min_col=col_idx, max_col=col_idx, max_row=ws.max_row):
+            row[0].number_format = "DD.MM.YYYY"
+
+# Formátuj všechny listy
+for ws in wb.worksheets:
+    table_name = re.sub(r'\W+', '_', ws.title)[:30] or "Tabulka"
+    style_table(ws, table_name)
+    format_dates(ws, ["Datum očkování", "Expirace"])
+
+wb.save(xlsx_path)
+
+print(f"Hotovo. Uloženo do: {xlsx_path.resolve()}")