Z230
This commit is contained in:
@@ -0,0 +1,188 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
import time
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
import pymysql
|
||||
from pymysql.cursors import DictCursor
|
||||
from playwright.sync_api import sync_playwright, TimeoutError as PWTimeout
|
||||
|
||||
# ========= CONFIG =========
|
||||
MYSQL_CFG = dict(
|
||||
host="192.168.1.76",
|
||||
port=3307,
|
||||
user="root",
|
||||
password="Vlado9674+",
|
||||
database="medevio",
|
||||
cursorclass=DictCursor,
|
||||
autocommit=False, # we commit in batches
|
||||
)
|
||||
|
||||
# Column in patients_extracted that stores Medevio UUID used in the URL:
|
||||
UUID_COLUMN = "rid" # <-- change if your column name differs
|
||||
|
||||
# Output columns (will be created if missing; MySQL 8.0+ supports IF NOT EXISTS):
|
||||
REGISTERED_COL = "medevio_registered" # TINYINT(1) NULL/0/1
|
||||
CHECKED_AT_COL = "medevio_checked_at" # DATETIME NULL
|
||||
ERROR_COL = "medevio_check_error" # TEXT NULL (optional)
|
||||
|
||||
# Medevio routing
|
||||
PATIENT_URL_TMPL = "https://my.medevio.cz/mudr-buzalkova/klinika/pacienti?pacient={uuid}"
|
||||
|
||||
# Login session (created earlier with your script)
|
||||
STATE_FILE = Path("../medevio_storage.json")
|
||||
|
||||
# Batch/pace
|
||||
BATCH_LIMIT = 5 # how many patients per run
|
||||
SLEEP_SECONDS = 3 # wait between patients (requested)
|
||||
NAV_TIMEOUT = 20_000 # ms
|
||||
TEXT_TIMEOUT = 15_000 # ms (for main area/heading)
|
||||
|
||||
# Texts indicating NOT registered:
|
||||
NOT_REGISTERED_STRINGS = [
|
||||
"Pacientka zatím nemá Medevio účet.",
|
||||
"Pacient zatím nemá Medevio účet.",
|
||||
]
|
||||
# ==========================
|
||||
|
||||
|
||||
|
||||
|
||||
SELECT_SQL = f"""
|
||||
SELECT {UUID_COLUMN} AS uuid, jmeno, prijmeni, rc
|
||||
FROM patients_extracted
|
||||
WHERE {UUID_COLUMN} IS NOT NULL
|
||||
AND {UUID_COLUMN} <> ''
|
||||
AND {REGISTERED_COL} IS NULL
|
||||
LIMIT {BATCH_LIMIT};
|
||||
"""
|
||||
|
||||
UPDATE_OK_SQL = f"""
|
||||
UPDATE patients_extracted
|
||||
SET {REGISTERED_COL}=%s, {CHECKED_AT_COL}=NOW(), {ERROR_COL}=NULL
|
||||
WHERE {UUID_COLUMN}=%s
|
||||
"""
|
||||
|
||||
UPDATE_ERR_SQL = f"""
|
||||
UPDATE patients_extracted
|
||||
SET {REGISTERED_COL}=NULL, {CHECKED_AT_COL}=NOW(), {ERROR_COL}=%s
|
||||
WHERE {UUID_COLUMN}=%s
|
||||
"""
|
||||
|
||||
DDL_SQLS = [
|
||||
f"ALTER TABLE patients_extracted ADD COLUMN {REGISTERED_COL} TINYINT(1) NULL",
|
||||
f"ALTER TABLE patients_extracted ADD COLUMN {CHECKED_AT_COL} DATETIME NULL",
|
||||
f"ALTER TABLE patients_extracted ADD COLUMN {ERROR_COL} TEXT NULL",
|
||||
]
|
||||
|
||||
CHECKS_FOR_DDL_SQLS=[
|
||||
f"SELECT COUNT(*) AS cnt FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'patients_extracted' AND COLUMN_NAME = '{REGISTERED_COL}'",
|
||||
f"SELECT COUNT(*) AS cnt FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'patients_extracted' AND COLUMN_NAME = '{CHECKED_AT_COL}'",
|
||||
f"SELECT COUNT(*) AS cnt FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA = DATABASE() AND TABLE_NAME = 'patients_extracted' AND COLUMN_NAME = '{ERROR_COL}'",
|
||||
]
|
||||
def ensure_columns(conn):
|
||||
with conn.cursor() as cur:
|
||||
for ddl,ddlcheck in zip(DDL_SQLS,CHECKS_FOR_DDL_SQLS):
|
||||
cur.execute(ddlcheck)
|
||||
row = cur.fetchone()
|
||||
if row["cnt"] == 0:
|
||||
print("Column missing")
|
||||
cur.execute(ddl)
|
||||
print(f"✓ Executed: {ddl}")
|
||||
else:
|
||||
print("Column exists")
|
||||
conn.commit()
|
||||
|
||||
|
||||
def pick_registered_flag(page_text: str) -> int:
|
||||
text = page_text or ""
|
||||
# If any NOT-registered phrase is present → 0; otherwise assume registered → 1
|
||||
for marker in NOT_REGISTERED_STRINGS:
|
||||
if marker in text:
|
||||
return 0
|
||||
return 1
|
||||
|
||||
|
||||
def main():
|
||||
# --- DB: fetch a batch to process ---
|
||||
conn = pymysql.connect(**MYSQL_CFG)
|
||||
try:
|
||||
ensure_columns(conn)
|
||||
with conn.cursor() as cur:
|
||||
cur.execute("SET NAMES utf8mb4 COLLATE utf8mb4_czech_ci")
|
||||
cur.execute("SET collation_connection = 'utf8mb4_czech_ci'")
|
||||
|
||||
cur.execute(SELECT_SQL)
|
||||
rows = cur.fetchall()
|
||||
|
||||
if not rows:
|
||||
print("No patients to check (all have medevio_registered filled).")
|
||||
return
|
||||
|
||||
print(f"Will process {len(rows)} patients…")
|
||||
|
||||
# --- Playwright session ---
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=False, slow_mo=0)
|
||||
context = browser.new_context(storage_state=str(STATE_FILE))
|
||||
page = context.new_page()
|
||||
page.set_default_timeout(NAV_TIMEOUT)
|
||||
|
||||
processed = ok = errs = 0
|
||||
|
||||
for r in rows:
|
||||
processed += 1
|
||||
# pid = r["id"]
|
||||
uuid = r["uuid"]
|
||||
name = f"{r.get('prijmeni','')}, {r.get('jmeno','')}"
|
||||
rc = r.get("rc","")
|
||||
|
||||
url = PATIENT_URL_TMPL.format(uuid=uuid)
|
||||
print(f"URL pro otevření pacienta je: {url}0")
|
||||
print(f"[{processed:>3}] {name} | RC {rc} | {uuid} → {url}")
|
||||
|
||||
try:
|
||||
page.goto(url, wait_until="domcontentloaded")
|
||||
# Optionally wait for a stable anchor; fallback to sleep
|
||||
try:
|
||||
# A stable bit we saw earlier
|
||||
page.get_by_text("Historie požadavků").wait_for(timeout=TEXT_TIMEOUT)
|
||||
except PWTimeout:
|
||||
pass
|
||||
|
||||
# Wait the requested 3 seconds for the UI to settle
|
||||
time.sleep(SLEEP_SECONDS)
|
||||
|
||||
# Get full text and detect
|
||||
full_text = page.content() # HTML; safer to check visible text too:
|
||||
vis_text = page.inner_text("body")
|
||||
|
||||
registered = pick_registered_flag(full_text) if full_text else pick_registered_flag(vis_text)
|
||||
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(UPDATE_OK_SQL, (registered, pid))
|
||||
conn.commit()
|
||||
ok += 1
|
||||
state = "REGISTERED" if registered == 1 else "NOT REGISTERED"
|
||||
print(f" → {state}")
|
||||
|
||||
except Exception as e:
|
||||
conn.rollback()
|
||||
errs += 1
|
||||
msg = f"{type(e).__name__}: {e}"
|
||||
with conn.cursor() as cur:
|
||||
cur.execute(UPDATE_ERR_SQL, (msg[:1000], pid))
|
||||
conn.commit()
|
||||
print(f" ! ERROR → {msg}")
|
||||
|
||||
browser.close()
|
||||
|
||||
print(f"Done. processed={processed}, ok={ok}, errors={errs}")
|
||||
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user