This commit is contained in:
michaela.buzalkova
2025-12-03 22:45:00 +01:00
parent 718bb5131c
commit 25e033f0b4
4 changed files with 414 additions and 14 deletions

View File

@@ -0,0 +1,168 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from pathlib import Path
import time
import firebirdsql as fb
import pandas as pd
import re
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
# from Functions import get_medicus_connection
# ================== Výstupní cesta ==================
BASE_DIR = Path(r"z:\Dropbox\Ordinace\Reporty") # uprav dle potřeby
timestamp = time.strftime("%Y-%m-%d %H-%M-%S")
xlsx_name = f"{timestamp} Očkování report.xlsx"
xlsx_path = BASE_DIR / xlsx_name
# ================== Smazání starých souborů ==================
for old_file in BASE_DIR.glob("*očkování report.xlsx"):
try:
if old_file != xlsx_path: # skip the file were about to create
old_file.unlink()
print(f"Smazán starý soubor: {old_file.name}")
except Exception as e:
print(f"⚠️ Nelze smazat {old_file.name}: {e}")
# ================== Definice skupin vakcín ==================
SHEETS = {
"COVID-19": ["commirnaty", "spikevax", "nuvaxovid"],
"Chřipka": ["vaxigrip", "influvac", "fluarix", "afluria"],
"Klíšťová encefalitida": ["fsme", "encepur"],
"Tetanus": ["tetavax", "boostrix", "adacel"],
"HepA": ["avaxim", "havrix","vaqta"],
"HepB": ["engerix"],
"HepA+B": ["twinrix"],
}
# ================== Připojení k DB ==================
# con = get_medicus_connection()
con = fb.connect(
host="192.168.1.10",
port=3050,
database=r"m:\Medicus\data\MEDICUS.FDB", # raw string for backslashes
user="SYSDBA",
password="masterkey",
charset="WIN1250", # adjust if needed
)
# ================== 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()
# ================== Datové typy ==================
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 více listů ==================
with pd.ExcelWriter(xlsx_path, engine="openpyxl") as writer:
for sheet_name, vakciny in SHEETS.items():
pattern = "|".join(re.escape(v) for v in vakciny if v)
mask = df["Látka"].astype(str).str.contains(pattern, case=False, na=False)
df_filtered = df[mask]
if not df_filtered.empty:
df_filtered.to_excel(writer, index=False, sheet_name=sheet_name)
# navíc celkový přehled všech očkování
df.to_excel(writer, index=False, sheet_name="Vše")
# ================== 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 safe_table_name(sheet_name):
"""Return an Excel-safe, unique table name."""
name = re.sub(r"[^0-9A-Za-z_]", "_", sheet_name)
return f"tbl_{name[:25]}"
def style_table(ws):
max_row = ws.max_row
max_col = ws.max_column
if max_col == 0:
return
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")
ws.freeze_panes = "A2"
if max_row < 2:
autosize_columns(ws)
return
ref = f"A1:{get_column_letter(max_col)}{max_row}"
tbl = Table(displayName=safe_table_name(ws.title), 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"
for ws in wb.worksheets:
style_table(ws)
format_dates(ws, ["Datum očkování", "Expirace"])
wb.save(xlsx_path)
print(f"✅ Hotovo. Uloženo do: {xlsx_path.resolve()}")