temporary-indexing/temp_index.py

"""Generates PDF cards from a calibre metadata.db."""

from textwrap import shorten

from calibrestekje import Book, Comment, Publisher, init_session
from reportlab.lib.pagesizes import *
from reportlab.lib.styles import ParagraphStyle, getSampleStyleSheet
from reportlab.pdfgen import canvas
from reportlab.platypus import PageBreak, Paragraph, SimpleDocTemplate, Spacer


def make_cards(filename, db_path, fields):
    """The main entrypoint for card generation."""
    doc = create_doc(filename)
    content = get_fields(db_path, fields)
    doc.build(content)


def get_fields(db_path, fields):
    """Retrieve fields from the metadata."""
    content = []
    styles = getSampleStyleSheet()
    session = init_session(db_path)

    for book in session.query(Book).all():
        if "title" in fields:
            tag = "<font size=12>{}</font>".format(book.title)
            ptitle = Paragraph(tag, styles["Italic"])
            content.append(ptitle)
            content.append(Spacer(1, 12))

        if "uploaded_at" in fields:
            tag = "<font size=10>Uploaded: {}</font>".format(book.timestamp)
            ptime = Paragraph(tag, styles["Normal"])
            content.append(ptime)
            content.append(Spacer(1, 12))

        if "comments" in fields:
            comments = ", ".join(
                [
                    shorten(comment.text, width=50, placeholder="...")
                    for comment in book.comments
                ]
            )
            tag = "<font size=10>{}</font>".format(comments)
            pcomments = Paragraph(tag)
            content.append(pcomments)

        if "authors" in fields:
            format_string = "<font size=12>{}</font>"
            all_authors = [author.name for author in book.authors]
            glued_together = format_string.format(", ".join(all_authors))

            p = Paragraph(glued_together, styles["Normal"])
            content.append(p)
            content.append(PageBreak())
            content.append(Spacer(6, 12))

        if "tags" in fields:
            format_string = "<font size=10>{}</font>"
            all_tags = [tag.name for tag in book.tags]
            tags_glued_together = format_string.format(", ".join(all_tags))

            p = Paragraph(tags_glued_together, styles["Normal"])
            content.append(p)
            content.append(PageBreak())
            content.append(Spacer(6, 12))

    return content


def create_doc(filename):
    """Build the Report Lab document template."""
    return SimpleDocTemplate(
        filename,
        pagesize=landscape(A6),
        rightMargin=18,
        leftMargin=18,
        topMargin=0,
        bottomMargin=18,
    )