Vladimir Buzalka
162 lines
4.5 KiB
Python
162 lines
4.5 KiB
Python
"""
|
|
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()
|