fix: Please the MyPy gods

Lookyloo · Jun 13, 2024 · 7c9f0a9 · 7c9f0a9
1 parent 49f8088
commit 7c9f0a9
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/pylookyloo/api.py b/pylookyloo/api.py
@@ -486,6 +486,26 @@ def send_mail(self, tree_uuid: str, email: str = '', comment: str | None = None)
         r = self.session.post(urljoin(self.root_url, str(PurePosixPath('json', tree_uuid, 'report'))), json=to_send)
         return r.json()
 
+    @overload
+    def upload_capture(self, *, quiet: Literal[True],
+                       listing: bool = False,
+                       full_capture: Path | BytesIO | str | None = None,
+                       har: Path | BytesIO | str | None = None,
+                       html: Path | BytesIO | str | None = None,
+                       last_redirected_url: str | None = None,
+                       screenshot: Path | BytesIO | str | None = None) -> str:
+        ...
+
+    @overload
+    def upload_capture(self, *, quiet: Literal[False]=False,
+                       listing: bool = False,
+                       full_capture: Path | BytesIO | str | None = None,
+                       har: Path | BytesIO | str | None = None,
+                       html: Path | BytesIO | str | None = None,
+                       last_redirected_url: str | None = None,
+                       screenshot: Path | BytesIO | str | None = None) -> tuple[str, dict[str, str]]:
+        ...
+
     def upload_capture(self, *, quiet: bool = False,
                        listing: bool = False,
                        full_capture: Path | BytesIO | str | None = None,