pylookyloo 1.36.1__tar.gz → 1.36.3__tar.gz

{pylookyloo-1.36.1 → pylookyloo-1.36.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pylookyloo
-Version: 1.36.1
+Version: 1.36.3
 Summary: Python CLI and module for Lookyloo
 License-Expression: GPL-2.0-or-later
 License-File: LICENSE
@@ -24,7 +24,7 @@ Classifier: Topic :: Security
 Provides-Extra: docs
 Provides-Extra: examples
 Requires-Dist: Sphinx (>=9.1.0) ; (python_version >= "3.12") and (extra == "docs")
-Requires-Dist: pylacus (>=1.21.0) ; extra == "examples"
+Requires-Dist: pylacus (>=1.21.2) ; extra == "examples"
 Requires-Dist: requests (>=2.32.5)
 Project-URL: Documentation, https://pylookyloo.readthedocs.io/en/latest/
 Project-URL: Repository, https://github.com/lookyloo/PyLookyloo

{pylookyloo-1.36.1 → pylookyloo-1.36.3}/pylookyloo/api.py

@@ -678,7 +678,8 @@ class Lookyloo():
                        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:
+                       screenshot: Path | BytesIO | str | None = None,
+                       categories: list[str] | None=None) -> str:
         ...
 
     @overload
@@ -688,7 +689,8 @@ class Lookyloo():
                        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]]:
+                       screenshot: Path | BytesIO | str | None = None,
+                       categories: list[str] | None=None) -> tuple[str, dict[str, str]]:
         ...
 
     def upload_capture(self, *, quiet: bool = False,
@@ -697,7 +699,8 @@ class Lookyloo():
                        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 | tuple[str, dict[str, str]]:
+                       screenshot: Path | BytesIO | str | None = None,
+                       categories: list[str] | None=None) -> str | tuple[str, dict[str, str]]:
         '''Upload a capture via har-file and others
 
         :param quiet: Returns the UUID only, instead of the the UUID and the potential error / warning messages
@@ -707,6 +710,7 @@ class Lookyloo():
         :param html: rendered HTML of the capture
         :param last_redirected_url: The landing page of the capture
         :param screenshot: Screenshot of the capture
+        :param categories: The categories assigned to the capture
         '''
         def encode_document(document: Path | BytesIO | str) -> str:
             if isinstance(document, str):
@@ -719,6 +723,8 @@ class Lookyloo():
             return base64.b64encode(document.getvalue()).decode()
 
         to_send: dict[str, Any] = {'listing': listing}
+        if categories:
+            to_send['categories'] = categories
 
         if full_capture:
             b64_full_capture = encode_document(full_capture)
@@ -740,9 +746,16 @@ class Lookyloo():
         else:
             raise PyLookylooError("Full capture or at least har-file are required")
 
-        r = self.session.post(urljoin(self.root_url, str(PurePosixPath('json', 'upload'))), json=to_send)
-        r.raise_for_status()
-        json_response = r.json()
+        try:
+            r = self.session.post(urljoin(self.root_url, str(PurePosixPath('json', 'upload'))), json=to_send)
+            json_response = r.json()
+            r.raise_for_status()
+        except requests.exceptions.HTTPError:
+            # We should have a response with details
+            raise PyLookylooError(f'Unable to upload capture: {json_response}.')
+        except Exception as e:
+            raise PyLookylooError(f'Unable to upload capture: {e}.')
+
         uuid = json_response['uuid']
         messages = json_response['messages']
 

{pylookyloo-1.36.1 → pylookyloo-1.36.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pylookyloo"
-version = "1.36.1"
+version = "1.36.3"
 description = "Python CLI and module for Lookyloo"
 authors = [
     {name="Raphaël Vinot", email="raphael.vinot@circl.lu"}
@@ -37,7 +37,7 @@ lookyloo = 'pylookyloo:main'
 
 [project.optional-dependencies]
 docs = ["Sphinx (>=9.1.0) ; python_version >= \"3.12\""]
-examples = ["pylacus (>=1.21.0)"]
+examples = ["pylacus (>=1.21.2)"]
 
 [tool.poetry.group.dev.dependencies]
 mypy = "^1.19.1"