Added optimize option for MarkdownPdf class, that generate compressed pdf-file. (#61)

* Add optimize option for compressed output. (#60)

* Fix coverage.

* Fix pdf examples.

* Fix history.txt

---------

Co-authored-by: Malte S. Stretz <mss@msquadrat.de>

Author: vb64

README.md

@@ -32,12 +32,12 @@ pip install markdown-pdf
 
 ## Usage
 
-Create a pdf with TOC (bookmarks) from headings up to level 2.
+Create a compressed pdf with TOC (bookmarks) from headings up to level 2.
 
 ```python
 from markdown_pdf import MarkdownPdf
 
-pdf = MarkdownPdf(toc_level=2)
+pdf = MarkdownPdf(toc_level=2, optimize=True)
 ```
 
 Add the first section to the pdf. The title is not included in the table of contents.

examples/markdown_pdf.pdf

@@ -1,3 +1,5 @@
++ Added `optimize` option for MarkdownPdf class, that generate compressed pdf-file.
+
 + Added file py.typed.
 
 * Package completely managed by pyproject.toml.

examples/markdown_pdf_ru.pdf

@@ -1,7 +1,7 @@
 import sys
 from markdown_pdf import MarkdownPdf, Section
 
-pdf = MarkdownPdf(toc_level=4)
+pdf = MarkdownPdf(toc_level=4, optimize=True)
 pdf.add_section(Section(open(sys.argv[1], encoding='utf-8').read()))
 pdf.meta["title"] = "MarkdownPdf module"
 pdf.save(sys.argv[2])

history.txt

@@ -42,7 +42,7 @@ class MarkdownPdf:
       "keywords": None,
     }
 
-    def __init__(self, toc_level: int = 6, mode: str = 'commonmark'):
+    def __init__(self, toc_level: int = 6, mode: str = 'commonmark', optimize: bool = False):
         """Create md -> pdf converter with given TOC level and mode of md parsing."""
         self.toc_level = toc_level
         self.toc = []
@@ -51,6 +51,8 @@ def __init__(self, toc_level: int = 6, mode: str = 'commonmark'):
         # https://markdown-it-py.readthedocs.io/en/latest/using.html#quick-start
         self.m_d = (MarkdownIt(mode).enable('table'))  # Enable support for tables
 
+        self.optimize = optimize
+
         self.out_file = io.BytesIO()
         self.writer = fitz.DocumentWriter(self.out_file)
         self.page_num = 0
@@ -100,5 +102,8 @@ def save(self, file_name: typing.Union[str, pathlib.Path]) -> None:
         doc.set_metadata(self.meta)
         if self.toc_level > 0:
             doc.set_toc(self.toc)
-        doc.save(file_name)
+        if self.optimize:
+            doc.ez_save(file_name)
+        else:
+            doc.save(file_name)
         doc.close()

makepdf.py

@@ -12,7 +12,7 @@ def test_en(self):
         """Test README.md example code."""
         from markdown_pdf import Section, MarkdownPdf
 
-        pdf = MarkdownPdf(toc_level=2)
+        pdf = MarkdownPdf(toc_level=2, optimize=True)
         pdf.add_section(Section("# Title\n", toc=False))
 
         text = """# Section with links