Vincent Bourdon
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
from logimage.domain.entities.puzzle import NonogramPuzzle
|
|
from logimage.domain.ports.image_source import ImageSource
|
|
from logimage.domain.ports.image_converter import ImageConverter
|
|
from logimage.domain.ports.pdf_exporter import PdfExporter
|
|
|
|
_DIFFICULTY_SIZES: dict[str, tuple[int, int]] = {
|
|
"easy": (10, 10),
|
|
"medium": (15, 15),
|
|
"hard": (20, 20),
|
|
}
|
|
|
|
|
|
@dataclass
|
|
class GeneratePuzzlesRequest:
|
|
count: int = 1
|
|
difficulty: str = "medium"
|
|
size: tuple[int, int] | None = None
|
|
theme: str | None = None
|
|
output_path: Path = field(default_factory=lambda: Path("puzzles.pdf"))
|
|
with_solution: bool = False
|
|
|
|
|
|
class GeneratePuzzlesUseCase:
|
|
def __init__(
|
|
self,
|
|
image_source: ImageSource,
|
|
image_converter: ImageConverter,
|
|
pdf_exporter: PdfExporter,
|
|
) -> None:
|
|
self._image_source = image_source
|
|
self._image_converter = image_converter
|
|
self._pdf_exporter = pdf_exporter
|
|
|
|
def execute(self, request: GeneratePuzzlesRequest) -> None:
|
|
width, height = request.size or _DIFFICULTY_SIZES.get(request.difficulty, (15, 15))
|
|
puzzles: list[NonogramPuzzle] = []
|
|
for _ in range(request.count):
|
|
image_data = self._image_source.fetch(request.theme)
|
|
grid = self._image_converter.to_grid(image_data.content, width, height)
|
|
puzzle = NonogramPuzzle.from_grid(grid, image_data.title, image_data.content)
|
|
puzzles.append(puzzle)
|
|
self._pdf_exporter.export(puzzles, request.output_path, request.with_solution)
|