satcfdi.printer

Complexity

Total Complexity

12

Size/Duplication

Total Lines	63
Duplicated Lines	0 %

Test Coverage

Coverage

64.85%

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	Representable.html_str_all()	0	3	1
A	Representable.pdf_bytes_all()	0	7	2
A	Representable.html_write()	0	2	1
A	Representable.html_str()	0	2	1
A	Representable.pdf_bytes()	0	6	2
A	Representable.pdf_write()	0	7	2
A	Representable.html_write_all()	0	3	1
A	Representable.pdf_write_all()	0	8	2

from lxml.etree import QName

try:
    import weasyprint
    PDF_CSS = weasyprint.CSS(string="@page {margin: 1.0cm 1.27cm 1.1cm 0.85cm;}")
except OSError as ex:
    weasyprint = None

from .render import PDF_INIT_TEMPLATE


class Representable:
    tag = None

    def html_write(self, target, init_template=PDF_INIT_TEMPLATE):
        init_template.stream({"c": self, "k": QName(self.tag).localname}).dump(target)

    def html_str(self, init_template=PDF_INIT_TEMPLATE) -> str:
        return init_template.render({"c": self, "k": QName(self.tag).localname})

    def pdf_write(self, target, init_template=PDF_INIT_TEMPLATE):
        if weasyprint is None:
            raise ImportError("weasyprint is not installed")

        weasyprint.HTML(string=self.html_str(init_template=init_template)).write_pdf(
            target=target,
            stylesheets=[PDF_CSS]
        )

    def pdf_bytes(self, init_template=PDF_INIT_TEMPLATE) -> bytes:
        if weasyprint is None:
            raise ImportError("weasyprint is not installed")

        return weasyprint.HTML(string=self.html_str(init_template=init_template)).write_pdf(
            stylesheets=[PDF_CSS]
        )

    @staticmethod
    def html_write_all(objs, target, init_template=PDF_INIT_TEMPLATE):
        init_template.stream({"c": [(QName(a.tag).localname, a) for a in objs], "k": '_multiple'}).dump(target)

    @staticmethod
    def html_str_all(objs, init_template=PDF_INIT_TEMPLATE) -> str:
        return init_template.render({"c": [(QName(a.tag).localname, a) for a in objs], "k": '_multiple'})

    @staticmethod
    def pdf_write_all(objs, target, init_template=PDF_INIT_TEMPLATE):
        if weasyprint is None:
            raise ImportError("weasyprint is not installed")

        weasyprint.HTML(string=Representable.html_str_all(objs, init_template=init_template)).write_pdf(
            target=target,
            stylesheets=[PDF_CSS]
        )

    @staticmethod
    def pdf_bytes_all(objs, init_template=PDF_INIT_TEMPLATE) -> bytes:
        if weasyprint is None:
            raise ImportError("weasyprint is not installed")

        return weasyprint.HTML(string=Representable.html_str_all(objs, init_template=init_template)).write_pdf(
            stylesheets=[PDF_CSS]
        )