publish.py

import typing
from itertools import zip_longest
from pathlib import Path

from reportlab.lib import pagesizes
from reportlab.lib.units import inch
from reportlab.pdfbase.pdfdoc import PDFInfo
from reportlab.platypus import SimpleDocTemplate

from blender_block import BlenderBlock
from block_pair import BlockPair
from footer import FooterCanvas
from publisher import Publisher
from shuffle_pages import ShuffledPagesPublisher
from svg_diagram import SvgDiagram


def write_blocks(blocks: typing.Sequence[BlenderBlock], out_path: Path) -> None:
    subtitle = blocks[0].subtitle
    subject = 'A word puzzle by Don Kirkby'
    author = 'Don Kirkby'
    if subtitle:
        subtitle_parts = subtitle.split()
        if subtitle_parts[0].lower() == 'by':
            subtitle_parts.pop(0)
            author = ' '.join(subtitle_parts)
        else:
            subject = subtitle

    doc = SimpleDocTemplate(str(out_path),
                            leftMargin=inch * 1.1,
                            rightMargin=inch * 1.1,
                            pagesize=pagesizes.letter,
                            author=author,
                            title=blocks[0].title,
                            subject=subject,
                            keywords=['puzzles',
                                      'word-puzzles',
                                      'games',
                                      'word-games'],
                            creator=PDFInfo.creator)

    blocks_iter = iter(blocks)

    doc.build([SvgDiagram(BlockPair(left, right).as_svg()).to_reportlab()
               for left, right in zip_longest(blocks_iter, blocks_iter)],
              canvasmaker=FooterCanvas)


class SortedWordsPublisher(Publisher):
    def write(self, source: str, out_path: Path) -> None:
        blocks = BlenderBlock.read(source, width=30, height=6, scale=0.5398)
        write_blocks(blocks, out_path)


def main():
    SortedWordsPublisher().publish()
    ShuffledPagesPublisher().publish()


if __name__ == '__main__':
    main()