z230

SběrDatRůzné/SudokuKiller/Testy/22_CropFromDB.py

@@ -0,0 +1,104 @@
+"""
+Stáhne 10 puzzle z MySQL (tabulka sudoku_killer), ořízne ray-cast metodou
+a uloží do Testy/verify/ pro vizuální verifikaci.
+"""
+
+import sys
+from pathlib import Path
+import fitz
+
+sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent / "Knihovny"))
+from mysql_db import connect_mysql
+
+import pymysql.cursors
+
+sys.stdout.reconfigure(encoding="utf-8")
+sys.stderr.reconfigure(encoding="utf-8")
+
+OUT_DIR = Path(__file__).parent / "verify"
+OUT_DIR.mkdir(exist_ok=True)
+
+MARGIN = 2  # pt — minimální rámeček
+
+
+def crop_raycast(pdf_bytes: bytes) -> bytes:
+    doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+    page = doc[0]
+    paths = page.get_drawings()
+
+    ph = page.mediabox.height
+    y_mid = ph / 2
+
+    hit_h = [(p["rect"], p.get("width") or 0) for p in paths
+             if p["rect"].y0 <= y_mid <= p["rect"].y1]
+    if not hit_h:
+        raise ValueError("Horizontální paprsek nenašel žádné kresby")
+
+    rects = [r for r, _ in hit_h]
+    x_left  = min(r.x0 for r in rects)
+    x_right = max(r.x1 for r in rects)
+    top_cut = min(r.y0 for r in rects)
+    bot_cut = max(r.y1 for r in rects)
+
+    # lineWidth svislých okrajových čar — souřadnice jsou středy, ne vizuální okraje
+    lw_left  = next((lw for r, lw in hit_h if r.x0 == x_left),  0)
+    lw_right = next((lw for r, lw in hit_h if r.x1 == x_right), 0)
+
+    vis_x_left  = x_left  - lw_left  / 2
+    vis_x_right = x_right + lw_right / 2
+    # top_cut / bot_cut jsou již vnější vizuální hrany (shodují se s okrajem horizontálních čar)
+
+    clip = fitz.Rect(
+        vis_x_left  - MARGIN,
+        top_cut     - MARGIN,
+        vis_x_right + MARGIN,
+        bot_cut     + MARGIN,
+    )
+
+    doc_new = fitz.open()
+    p = doc_new.new_page(width=clip.width, height=clip.height)
+    p.show_pdf_page(fitz.Rect(0, 0, clip.width, clip.height), doc, 0, clip=clip)
+
+    out = doc_new.tobytes()
+    doc.close()
+    doc_new.close()
+    return out
+
+
+def main():
+    import pymysql.cursors
+    conn = connect_mysql(database="puzzle", cursorclass=pymysql.cursors.DictCursor)
+    cursor = conn.cursor()
+
+    cursor.execute("""
+        SELECT puzzle_number, puzzle_date, difficulty, file_puzzle
+        FROM sudoku_killer
+        WHERE file_puzzle IS NOT NULL
+        ORDER BY puzzle_number
+        LIMIT 10
+    """)
+    rows = cursor.fetchall()
+    cursor.close()
+    conn.close()
+
+    print(f"Staženo {len(rows)} záznamů z DB.")
+
+    for row in rows:
+        num  = row["puzzle_number"]
+        date = row["puzzle_date"]
+        diff = row["difficulty"]
+        pdf_bytes = bytes(row["file_puzzle"])
+
+        try:
+            cropped = crop_raycast(pdf_bytes)
+            out_path = OUT_DIR / f"{date} Puzzle SudokuKiller {num} [diff {diff}] cropped.pdf"
+            out_path.write_bytes(cropped)
+            print(f"  OK  #{num}  →  {out_path.name}")
+        except Exception as e:
+            print(f"  CHYBA  #{num}: {e}", file=sys.stderr)
+
+    print(f"\nHotovo. Soubory v: {OUT_DIR}")
+
+
+if __name__ == "__main__":
+    main()