notebookvb

SběrDatRůzné/DailyKakuro/vykresli_kakuro.py

@@ -0,0 +1,161 @@
+"""
+Vykreslí Kakuro puzzle do PDF z dat v MySQL tabulce puzzles.
+"""
+
+import json
+import os
+import sys
+from pathlib import Path
+
+sys.stdout.reconfigure(encoding="utf-8")
+sys.path.insert(0, str(Path(__file__).parent.parent.parent / "Knihovny"))
+
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.units import cm
+from reportlab.pdfbase import pdfmetrics
+from reportlab.pdfbase.ttfonts import TTFont
+from reportlab.pdfgen.canvas import Canvas
+
+from mysql_db import connect_mysql
+
+_fonts_dir = os.path.join(os.environ.get("WINDIR", r"C:\Windows"), "Fonts")
+pdfmetrics.registerFont(TTFont("Arial", os.path.join(_fonts_dir, "arial.ttf")))
+pdfmetrics.registerFont(TTFont("ArialBold", os.path.join(_fonts_dir, "arialbd.ttf")))
+
+OUTPUT = Path(__file__).parent / "test_kakuro.pdf"
+
+
+def parse_grid(data_str: str) -> list[str]:
+    rows = ["0" + r for r in data_str.split(",")]
+    rows.insert(0, "0" * len(rows[0]))
+    return rows
+
+
+def sum_hor(grid, x, y):
+    s = 0
+    w = len(grid[0])
+    for i in range(x + 1, w):
+        if grid[y][i] == "0":
+            break
+        s += int(grid[y][i])
+    return s if s > 0 else 0
+
+
+def sum_vert(grid, x, y):
+    s = 0
+    h = len(grid)
+    for i in range(y + 1, h):
+        if grid[i][x] == "0":
+            break
+        s += int(grid[i][x])
+    return s if s > 0 else 0
+
+
+def draw_kakuro(c: Canvas, x0: float, y0: float, cell: float,
+                data_str: str, title: str = "", show_solution: bool = False):
+    grid = parse_grid(data_str)
+    h = len(grid)
+    w = len(grid[0])
+
+    clue_font = max(cell * 0.3, 5)
+    num_font = max(cell * 0.5, 6)
+
+    if title:
+        c.setFont("ArialBold", 12)
+        c.drawString(x0, y0 + 5, title)
+
+    for gy in range(h):
+        for gx in range(w):
+            cx = x0 + gx * cell
+            cy = y0 - (gy + 1) * cell
+            ch = grid[gy][gx]
+
+            if ch == "0":
+                # Černá buňka
+                c.setFillColor(colors.Color(0.15, 0.15, 0.15))
+                c.rect(cx, cy, cell, cell, fill=1, stroke=0)
+
+                sv = sum_vert(grid, gx, gy)
+                sh = sum_hor(grid, gx, gy)
+
+                if sv or sh:
+                    # Diagonála
+                    c.setStrokeColor(colors.white)
+                    c.setLineWidth(0.5)
+                    c.line(cx, cy + cell, cx + cell, cy)
+                    c.setStrokeColor(colors.black)
+
+                    c.setFillColor(colors.white)
+                    c.setFont("ArialBold", clue_font)
+
+                    if sh:
+                        # Součet doprava — horní pravý trojúhelník
+                        c.drawString(cx + cell * 0.52, cy + cell * 0.55, str(sh))
+
+                    if sv:
+                        # Součet dolů — dolní levý trojúhelník
+                        c.drawString(cx + cell * 0.08, cy + cell * 0.08, str(sv))
+
+            else:
+                # Bílá buňka
+                if show_solution:
+                    c.setFillColor(colors.black)
+                    c.setFont("ArialBold", num_font)
+                    c.drawCentredString(cx + cell / 2, cy + cell * 0.3, ch)
+
+    # Mřížka
+    c.setStrokeColor(colors.black)
+    for i in range(h + 1):
+        c.setLineWidth(0.8)
+        c.line(x0, y0 - i * cell, x0 + w * cell, y0 - i * cell)
+    for i in range(w + 1):
+        c.setLineWidth(0.8)
+        c.line(x0 + i * cell, y0, x0 + i * cell, y0 - h * cell)
+
+
+def main():
+    conn = connect_mysql(database="puzzle")
+    cur = conn.cursor()
+    cur.execute(
+        "SELECT difficulty, puzzle, extra FROM puzzles "
+        "WHERE game_type='kakuro' AND puzzle_date='2026-05-08' "
+        "ORDER BY FIELD(difficulty, 'easy', 'medium', 'hard', 'expert') "
+        "LIMIT 1"
+    )
+    row = cur.fetchone()
+    cur.close()
+    conn.close()
+
+    if not row:
+        print("Žádná data.")
+        return
+
+    difficulty, data_str, extra = row
+    grid = parse_grid(data_str)
+    grid_size = len(grid[0])
+
+    page_w, page_h = A4
+    board_cm = 11
+    cell = board_cm * cm / grid_size
+    board = grid_size * cell
+
+    c = Canvas(str(OUTPUT), pagesize=A4)
+
+    # Zadání (bez čísel)
+    x0 = (page_w - board) / 2
+    y0 = page_h - 2 * cm
+    draw_kakuro(c, x0, y0, cell, data_str,
+                f"Kakuro {difficulty.capitalize()} — 2026-05-08", show_solution=False)
+
+    # Řešení (s čísly)
+    y0_sol = y0 - board - 3 * cm
+    draw_kakuro(c, x0, y0_sol, cell, data_str,
+                f"Řešení", show_solution=True)
+
+    c.save()
+    print(f"PDF uloženo: {OUTPUT}")
+
+
+if __name__ == "__main__":
+    main()