feat: init type hints

vvanglro · Nov 4, 2022 · ac233db · ac233db
1 parent 91a0c34
commit ac233db
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/pwhtmltopdf/core.py b/pwhtmltopdf/core.py
@@ -1,10 +1,27 @@
 import pathlib
+import typing
 
 from pwhtmltopdf.abc import BaseHTP
 from pwhtmltopdf.types import StrPath, StrPathLike
 
 
 class HtmlToPdf(BaseHTP):
+    def __init__(
+        self,
+        static_root: StrPath = None,
+        wait_until: typing.Optional[
+            typing.Literal["commit", "domcontentloaded", "load", "networkidle"]
+        ] = None,
+        print_background: typing.Optional[bool] = None,
+        prefer_css_page_size: typing.Optional[bool] = None,
+    ):
+        super().__init__(
+            static_root=static_root,
+            wait_until=wait_until,
+            print_background=print_background,
+            prefer_css_page_size=prefer_css_page_size,
+        )
+
     async def from_url(
         self,
         url: str,

diff --git a/pwhtmltopdf/py.typed b/pwhtmltopdf/py.typed