Add support for pathlib.Path paths in type hints for save method. (#36)

markdown_pdf/__init__.py

@@ -1,6 +1,7 @@
 """Markdown to pdf converter based on markdown_it and fitz."""
 import io
 import typing
+import pathlib
 from markdown_it import MarkdownIt
 import fitz
 
@@ -82,7 +83,7 @@ def add_section(self, section: Section, user_css: typing.Optional[str] = None) -
             story.draw(device)
             self.writer.end_page()
 
-    def save(self, file_name: str) -> None:
+    def save(self, file_name: typing.Union[str, pathlib.Path]) -> None:
         """Save pdf to file."""
         self.writer.close()
         doc = fitz.open("pdf", self.out_file)

tests/test/test_converter.py

@@ -51,3 +51,14 @@ def test_table(self):
         pdf = MarkdownPdf(toc_level=0)
         pdf.add_section(Section(TABLE_TEXT))
         pdf.save(self.build("table.pdf"))
+
+    def test_pathlib(self):
+        """Check pathlib.Path for save method."""
+        from pathlib import Path
+        from markdown_pdf import MarkdownPdf, Section
+
+        pdf = MarkdownPdf()
+        pdf.add_section(Section('# Hello World'))
+
+        file_path = Path(self.build('output.pdf'))
+        pdf.save(file_path)