114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
import os
|
||
import zlib
|
||
import pymysql
|
||
import re
|
||
from pathlib import Path
|
||
from datetime import datetime
|
||
|
||
# ==============================
|
||
# ⚙️ CONFIGURATION
|
||
# ==============================
|
||
DB_CONFIG = {
|
||
"host": "192.168.1.76",
|
||
"port": 3307,
|
||
"user": "root",
|
||
"password": "Vlado9674+",
|
||
"database": "medevio",
|
||
"charset": "utf8mb4",
|
||
}
|
||
|
||
BASE_DIR = Path(r"u:\Dropbox\Ordinace\Dokumentace_ke_zpracování\MP")
|
||
BASE_DIR.mkdir(parents=True, exist_ok=True)
|
||
|
||
|
||
def sanitize_name(name: str) -> str:
|
||
"""Replace invalid filename characters with underscore."""
|
||
return re.sub(r'[<>:"/\\|?*\x00-\x1F]', "_", name).strip()
|
||
|
||
|
||
# ==============================
|
||
# 📦 EXPORT WITH JOIN TO POZADAVKY
|
||
# ==============================
|
||
conn = pymysql.connect(**DB_CONFIG)
|
||
cur_meta = conn.cursor(pymysql.cursors.DictCursor)
|
||
cur_blob = conn.cursor()
|
||
|
||
# 🎯 JOIN medevio_downloads → pozadavky
|
||
cur_meta.execute("""
|
||
SELECT d.id, d.request_id, d.attachment_id, d.filename,
|
||
d.created_at, d.downloaded_at,
|
||
p.pacient_jmeno AS jmeno,
|
||
p.pacient_prijmeni AS prijmeni
|
||
FROM medevio_downloads d
|
||
JOIN pozadavky p ON d.request_id = p.id
|
||
WHERE d.file_content IS NOT NULL;
|
||
""")
|
||
|
||
rows = cur_meta.fetchall()
|
||
print(f"📋 Found {len(rows)} records to check/export")
|
||
|
||
skipped, exported = 0, 0
|
||
|
||
for r in rows:
|
||
try:
|
||
created = r["created_at"] or r["downloaded_at"] or datetime.now()
|
||
date_str = created.strftime("%Y-%m-%d")
|
||
|
||
# 👍 Now always correct from pozadavky
|
||
prijmeni = sanitize_name(r["prijmeni"] or "Unknown")
|
||
jmeno = sanitize_name(r["jmeno"] or "")
|
||
|
||
# 🔥 Full request_id for folder identification
|
||
full_req_id = sanitize_name(r["request_id"])
|
||
|
||
# Folder names (normal and triangle)
|
||
base_folder = f"{date_str} {prijmeni}, {jmeno} {full_req_id}"
|
||
tri_folder = f"{date_str}▲ {prijmeni}, {jmeno} {full_req_id}"
|
||
|
||
base_folder = sanitize_name(base_folder)
|
||
tri_folder = sanitize_name(tri_folder)
|
||
|
||
base_path = BASE_DIR / base_folder
|
||
tri_path = BASE_DIR / tri_folder
|
||
|
||
filename = sanitize_name(r["filename"] or f"unknown_{r['id']}.bin")
|
||
file_path_base = base_path / filename
|
||
file_path_tri = tri_path / filename
|
||
|
||
# 🟡 Skip if file already exists
|
||
if file_path_base.exists() or file_path_tri.exists():
|
||
skipped += 1
|
||
found_in = "▲" if file_path_tri.exists() else ""
|
||
print(f"⏭️ Skipping existing{found_in}: {filename}")
|
||
continue
|
||
|
||
# Ensure directory exists
|
||
base_path.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 2️⃣ Fetch blob content
|
||
cur_blob.execute(
|
||
"SELECT file_content FROM medevio_downloads WHERE id = %s",
|
||
(r["id"],)
|
||
)
|
||
blob = cur_blob.fetchone()[0]
|
||
|
||
if blob:
|
||
with open(file_path_base, "wb") as f:
|
||
f.write(blob)
|
||
exported += 1
|
||
print(f"✅ Saved: {file_path_base.relative_to(BASE_DIR)}")
|
||
else:
|
||
print(f"⚠️ No content for id={r['id']}")
|
||
|
||
except Exception as e:
|
||
print(f"❌ Error for id={r['id']}: {e}")
|
||
|
||
cur_blob.close()
|
||
cur_meta.close()
|
||
conn.close()
|
||
|
||
print(f"\n🎯 Export complete — {exported} new files saved, {skipped} skipped.\n")
|