WIP: exporting PDF with weasyprint

Author: clstaudt

.gitignore

@@ -137,3 +137,4 @@ dmypy.json
 
 # temporary
 notebooks/tmp
+tuttle_tests/tmp

tuttle/rendering.py

@@ -11,11 +11,6 @@
 import pandas
 from loguru import logger
 
-# pdfkit needs wkhtmltopdf to be installed
-if getattr(sys, "frozen", False):
-    os.environ["PATH"] = sys._MEIPASS + os.pathsep + os.environ["PATH"]
-import pdfkit
-
 
 from .model import User, Invoice, Timesheet, Project
 from .view import Timeline
@@ -41,6 +36,27 @@ def convert_html_to_pdf(
         dest_dir (_type_): _description_
     """
     logger.info(f"converting html to pdf: {in_path} -> {out_path}")
+    _convert_html_to_pdf_with_weasyprint(
+        in_path=in_path,
+        out_path=out_path,
+        css_paths=css_paths,
+    )
+
+
+def _convert_html_to_pdf_with_pdfkit(
+    in_path,
+    out_path,
+    css_paths=[],
+):
+    """Implementation of convert_html_to_pdf using pdfkit."""
+    # pdfkit needs wkhtmltopdf to be installed
+    if getattr(sys, "frozen", False):
+        os.environ["PATH"] = sys._MEIPASS + os.pathsep + os.environ["PATH"]
+    try:
+        import pdfkit
+    except ImportError:
+        logger.error("Please install pdfkit and wkhtmltopdf")
+        raise
     try:
         pdfkit.from_file(input=in_path, output_path=out_path, css=css_paths)
     except OSError as ex:
@@ -49,6 +65,53 @@ def convert_html_to_pdf(
         pass
 
 
+def _convert_html_to_pdf_with_weasyprint(
+    in_path,
+    out_path,
+    css_paths=[],
+):
+    """Implementation of convert_html_to_pdf using weasyprint."""
+    try:
+        import weasyprint
+    except ImportError:
+        logger.error("Please install weasyprint")
+        raise
+    css_paths = [Path(css_path).resolve() for css_path in css_paths]
+    logger.debug(f"css_paths: {css_paths}")
+    (
+        weasyprint.HTML(in_path).write_pdf(
+            out_path,
+            stylesheets=css_paths,
+        )
+    )
+
+
+def _convert_html_to_pdf_with_QT(
+    in_path,
+    out_path,
+    css_paths=[],
+):
+    """Implementation of convert_html_to_pdf using QT."""
+    try:
+        from PyQt5 import QtCore, QtWidgets, QtWebEngineWidgets
+    except ImportError:
+        logger.error("Please install PyQt5")
+        raise
+    app = QtWidgets.QApplication(sys.argv)
+    loader = QtWebEngineWidgets.QWebEngineView()
+    loader.setZoomFactor(1)
+    loader.page().pdfPrintingFinished.connect(lambda *args: print("finished:", args))
+    loader.load(QtCore.QUrl(in_path))
+
+    def emit_pdf(finished):
+        loader.show()
+        loader.page().printToPdf(out_path)
+
+    loader.loadFinished.connect(emit_pdf)
+
+    app.exec()
+
+
 def render_invoice(
     user: User,
     invoice: Invoice,

tuttle_tests/conftest.py

@@ -54,8 +54,7 @@ def demo_user():
 @pytest.fixture
 def demo_clients():
     central_services = Client(
-        first_name="Central",
-        last_name="Services",
+        name="Central Services",
         invoicing_contact=Contact(
             first_name="Central",
             last_name="Services",

tuttle_tests/test_invoice.py

@@ -1,8 +1,9 @@
 """Tests for the invoice module."""
 
 import datetime
+from pathlib import Path
 
-from tuttle import invoicing, timetracking
+from tuttle import invoicing, timetracking, rendering
 from tuttle.model import Invoice, InvoiceItem
 
 
@@ -65,9 +66,63 @@ def test_generate_invoice(
         assert invoice.total > 0
 
 
-def test_render_invoice_to_html():
-    pass
+def test_render_invoice_to_html(
+    demo_user,
+    demo_projects,
+    demo_calendar_timetracking,
+):
+    for project in demo_projects:
+        timesheets = []
+        for period in ["January 2022", "February 2022"]:
+            timesheet = timetracking.generate_timesheet(
+                source=demo_calendar_timetracking,
+                project=project,
+                period_start=period,
+                item_description=project.title,
+            )
+            if not timesheet.empty:
+                timesheets.append(timesheet)
+        invoice = invoicing.generate_invoice(
+            timesheets=timesheets,
+            contract=project.contract,
+            date=datetime.date.today(),
+        )
+        # RENDERING
+        rendering.render_invoice(
+            user=demo_user,
+            invoice=invoice,
+            style="anvil",
+            document_format="html",
+            out_dir=Path("tuttle_tests/tmp"),
+        )
 
 
-def test_render_invoice_to_pdf():
-    pass
+def test_render_invoice_to_pdf(
+    demo_user,
+    demo_projects,
+    demo_calendar_timetracking,
+):
+    for project in demo_projects:
+        timesheets = []
+        for period in ["January 2022", "February 2022"]:
+            timesheet = timetracking.generate_timesheet(
+                source=demo_calendar_timetracking,
+                project=project,
+                period_start=period,
+                item_description=project.title,
+            )
+            if not timesheet.empty:
+                timesheets.append(timesheet)
+        invoice = invoicing.generate_invoice(
+            timesheets=timesheets,
+            contract=project.contract,
+            date=datetime.date.today(),
+        )
+        # RENDERING
+        rendering.render_invoice(
+            user=demo_user,
+            invoice=invoice,
+            style="anvil",
+            document_format="pdf",
+            out_dir=Path("tuttle_tests/tmp"),
+        )