pangea-sdk 6.2.0b1__py3-none-any.whl → 6.2.0b2__py3-none-any.whl

pangea/asyncio/services/authz.py

@@ -88,7 +88,7 @@ class AuthZAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with empty result.
             Available response fields can be found in our
-            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/tuple/create).
+            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/tuple/create-post).
 
         Examples:
             await authz.tuple_create(
@@ -134,7 +134,7 @@ class AuthZAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with a list of tuples and the last token.
             Available response fields can be found in our
-            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/tuple/list).
+            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/tuple/list-post).
 
         Examples:
             await authz.tuple_list(TupleListFilter(subject_type="user", subject_id="user_1"))
@@ -158,7 +158,7 @@ class AuthZAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with empty result.
             Available response fields can be found in our
-            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/tuple/delete).
+            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/tuple/delete-post).
 
         Examples:
             await authz.tuple_delete(
@@ -202,7 +202,7 @@ class AuthZAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with the result of the check.
             Available response fields can be found in our
-            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/check).
+            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/check-post).
 
         Examples:
             await authz.check(
@@ -236,7 +236,7 @@ class AuthZAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with a list of resource IDs.
             Available response fields can be found in our
-            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/list-resources).
+            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/list-resources-post).
 
         Examples:
             await authz.list_resources(
@@ -270,7 +270,7 @@ class AuthZAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with a list of subjects.
             Available response fields can be found in our
-            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/list-subjects).
+            [API Documentation](https://pangea.cloud/docs/api/authz#/v1/list-subjects-post).
 
         Examples:
             await authz.list_subjects(

pangea/asyncio/services/base.py

@@ -1,5 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 from __future__ import annotations
 
 from typing import Dict, Optional, Type, Union

pangea/asyncio/services/file_scan.py

@@ -1,5 +1,11 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
+from __future__ import annotations
+
 import io
 import logging
 from typing import Dict, List, Optional, Tuple
@@ -16,7 +22,7 @@ class FileScanAsync(ServiceBaseAsync):
     """FileScan service client.
 
     Provides methods to interact with Pangea FileScan Service:
-        https://pangea.cloud/docs/api/embargo
+        https://pangea.cloud/docs/api/file-scan
 
     The following information is needed:
         PANGEA_TOKEN - service token which can be found on the Pangea User
@@ -98,7 +104,7 @@ class FileScanAsync(ServiceBaseAsync):
         files: Optional[List[Tuple]] = None
         if file or file_path:
             if file_path:
-                file = open(file_path, "rb")
+                file = open(file_path, "rb")  # noqa: SIM115
             if transfer_method == TransferMethod.POST_URL:
                 params = get_file_upload_params(file)  # type: ignore[arg-type]
                 crc = params.crc_hex

pangea/asyncio/services/intel.py

@@ -1,5 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 import hashlib
 from typing import List, Optional
 
@@ -144,8 +148,8 @@ class FileIntelAsync(ServiceBaseAsync):
             )
         """
 
-        data = open(filepath, "rb")
-        hash = hashlib.sha256(data.read()).hexdigest()
+        with open(filepath, "rb") as data:
+            hash = hashlib.sha256(data.read()).hexdigest()
 
         input = m.FileReputationRequest(hash=hash, hash_type="sha256", verbose=verbose, raw=raw, provider=provider)
         return await self.request.post(

pangea/asyncio/services/prompt_guard.py

@@ -1,10 +1,18 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, Literal
 
 from pangea.asyncio.services.base import ServiceBaseAsync
 from pangea.config import PangeaConfig
-from pangea.services.prompt_guard import GuardResult, Message
+from pangea.response import PangeaResponseResult
+from pangea.services.prompt_guard import (
+    AuditDataActivityConfig,
+    GuardResult,
+    Message,
+    ServiceConfigFilter,
+    ServiceConfigsPage,
+)
 
 if TYPE_CHECKING:
     from collections.abc import Iterable
@@ -82,3 +90,105 @@ class PromptGuardAsync(ServiceBaseAsync):
             GuardResult,
             data={"messages": messages, "analyzers": analyzers, "classify": classify},
         )
+
+    async def get_service_config(
+        self,
+        *,
+        id: str | None = None,
+        version: str | None = None,
+        analyzers: Mapping[str, bool] | None = None,
+        malicious_detection_threshold: float | None = None,
+        benign_detection_threshold: float | None = None,
+        audit_data_activity: AuditDataActivityConfig | None = None,
+    ) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config
+        """
+        return await self.request.post(
+            "v1beta/config",
+            data={
+                "id": id,
+                "version": version,
+                "analyzers": analyzers,
+                "malicious_detection_threshold": malicious_detection_threshold,
+                "benign_detection_threshold": benign_detection_threshold,
+                "audit_data_activity": audit_data_activity,
+            },
+            result_class=PangeaResponseResult,
+        )
+
+    async def create_service_config(
+        self,
+        *,
+        id: str | None = None,
+        version: str | None = None,
+        analyzers: Mapping[str, bool] | None = None,
+        malicious_detection_threshold: float | None = None,
+        benign_detection_threshold: float | None = None,
+        audit_data_activity: AuditDataActivityConfig | None = None,
+    ) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_create
+        """
+        return await self.request.post(
+            "v1beta/config/create",
+            data={
+                "id": id,
+                "version": version,
+                "analyzers": analyzers,
+                "malicious_detection_threshold": malicious_detection_threshold,
+                "benign_detection_threshold": benign_detection_threshold,
+                "audit_data_activity": audit_data_activity,
+            },
+            result_class=PangeaResponseResult,
+        )
+
+    async def update_service_config(
+        self,
+        *,
+        id: str | None = None,
+        version: str | None = None,
+        analyzers: Mapping[str, bool] | None = None,
+        malicious_detection_threshold: float | None = None,
+        benign_detection_threshold: float | None = None,
+        audit_data_activity: AuditDataActivityConfig | None = None,
+    ) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_update
+        """
+        return await self.request.post(
+            "v1beta/config/update",
+            data={
+                "id": id,
+                "version": version,
+                "analyzers": analyzers,
+                "malicious_detection_threshold": malicious_detection_threshold,
+                "benign_detection_threshold": benign_detection_threshold,
+                "audit_data_activity": audit_data_activity,
+            },
+            result_class=PangeaResponseResult,
+        )
+
+    async def delete_service_config(self, id: str) -> PangeaResponse[PangeaResponseResult]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_delete
+        """
+        return await self.request.post("v1beta/config/delete", data={"id": id}, result_class=PangeaResponseResult)
+
+    async def list_service_configs(
+        self,
+        *,
+        filter: ServiceConfigFilter | None = None,
+        last: str | None = None,
+        order: Literal["asc", "desc"] | None = None,
+        order_by: Literal["id", "created_at", "updated_at"] | None = None,
+        size: int | None = None,
+    ) -> PangeaResponse[ServiceConfigsPage]:
+        """
+        OperationId: prompt_guard_post_v1beta_config_list
+        """
+        return await self.request.post(
+            "v1beta/config/list",
+            data={"filter": filter, "last": last, "order": order, "order_by": order_by, "size": size},
+            result_class=ServiceConfigsPage,
+        )

pangea/asyncio/services/redact.py

@@ -1,5 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 from __future__ import annotations
 
 from collections.abc import Mapping, Sequence
@@ -95,7 +99,7 @@ class RedactAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with redacted text in the response.result property,
                 available response fields can be found in our
-                [API Documentation](https://pangea.cloud/docs/api/redact#redact).
+                [API Documentation](https://pangea.cloud/docs/api/redact#redact-post).
 
         Examples:
             response = redact.redact(text="Jenny Jenny... 555-867-5309")
@@ -154,7 +158,7 @@ class RedactAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with redacted data in the response.result field,
                 available response fields can be found in our
-                [API Documentation](https://pangea.cloud/docs/api/redact#redact-structured)
+                [API Documentation](https://pangea.cloud/docs/api/redact#redact-structured-post)
 
         Examples:
             data = {
@@ -199,7 +203,7 @@ class RedactAsync(ServiceBaseAsync):
         Returns:
             Pangea Response with redacted data in the response.result field,
                 available response fields can be found in our
-                [API Documentation](https://pangea.cloud/docs/api/redact#unredact)
+                [API Documentation](https://pangea.cloud/docs/api/redact#unredact-post)
         """
         input = m.UnredactRequest(redacted_data=redacted_data, fpe_context=fpe_context)
         return await self.request.post("v1/unredact", m.UnredactResult, data=input.model_dump(exclude_none=True))

pangea/asyncio/services/sanitize.py

@@ -1,5 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 from __future__ import annotations
 
 import io
@@ -116,7 +120,7 @@ class SanitizeAsync(ServiceBaseAsync):
         files: Optional[List[Tuple]] = None
         if file or file_path:
             if file_path:
-                file = open(file_path, "rb")
+                file = open(file_path, "rb")  # noqa: SIM115
             if transfer_method == TransferMethod.POST_URL and (sha256 is None or crc32c is None or size is None):
                 params = get_file_upload_params(file)  # type: ignore[arg-type]
                 crc32c = params.crc_hex if crc32c is None else crc32c

pangea/asyncio/services/share.py

@@ -1,5 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 from __future__ import annotations
 
 import io
@@ -194,7 +198,7 @@ class ShareAsync(ServiceBaseAsync):
 
     async def get_archive(
         self,
-        ids: List[str] = [],
+        ids: list[str],
         format: Optional[m.ArchiveFormat] = None,
         transfer_method: Optional[TransferMethod] = None,
         bucket_id: Optional[str] = None,

pangea/asyncio/services/vault.py

@@ -1,5 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 from __future__ import annotations
 
 from collections.abc import Mapping
@@ -164,7 +168,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the id of the deleted secret or key
                 is returned in the response.result field.
-                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault#delete).
+                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-general#/v1/delete-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -197,7 +201,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the secret or key
                 is returned in the response.result field.
-                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault#retrieve).
+                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-general#/v1/get-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -283,7 +287,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where a list of secrets or keys
                 is returned in the response.result field.
-                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault#list).
+                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-general#/v1/list-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -345,7 +349,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the item ID is returned in the
             response.result field. Available response fields can be found in our
-            [API documentation](https://pangea.cloud/docs/api/vault#update).
+            [API documentation](https://pangea.cloud/docs/api/vault/v1-general#/v1/update-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -1680,7 +1684,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the encrypted message in base64 is returned
             in the response.result field. Available response fields can be found
-            in our [API documentation](https://pangea.cloud/docs/api/vault#encrypt).
+            in our [API documentation](https://pangea.cloud/docs/api/vault/v1-keys#/v1/key/encrypt-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -1716,7 +1720,7 @@ class VaultAsync(ServiceBaseAsync):
 
         Returns:
             A PangeaResponse where the decrypted message in base64 is returned
-            in the response.result field. Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault#decrypt).
+            in the response.result field. Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-keys#/v1/key/decrypt-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -1750,7 +1754,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the signature of the message in base64 is
             returned in the response.result field. Available response fields can
-            be found in our [API documentation](https://pangea.cloud/docs/api/vault#sign).
+            be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-keys#/v1/key/sign-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -1788,7 +1792,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the signature is valid
                 is returned in the response.result field.
-                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault#verify).
+                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-keys#/v1/key/verify-post).
 
         Examples:
             response = await vault.verify(
@@ -1826,7 +1830,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the signature is valid
                 is returned in the response.result field.
-                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault#verify-jwt).
+                Available response fields can be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-jwt#/v1/key/verify/jwt-post).
 
         Examples:
             response = await vault.jwt_verify(jws="ewogICJhbGciO...")
@@ -1851,7 +1855,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the signed JSON Web Token (JWS) is returned
             in the response.result field. Available response fields can be found
-            in our [API documentation](https://pangea.cloud/docs/api/vault#sign-a-jwt).
+            in our [API documentation](https://pangea.cloud/docs/api/vault/v1-jwt#/v1/key/sign/jwt-post).
 
         Examples:
             response = await vault.jwt_sign(
@@ -1881,7 +1885,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the JSON Web Key Set (JWKS) object is
             returned in the response.result field. Available response fields can
-            be found in our [API documentation](https://pangea.cloud/docs/api/vault#retrieve-jwk).
+            be found in our [API documentation](https://pangea.cloud/docs/api/vault/v1-jwt#/v1/get/jwk-post).
 
         Examples:
             response = await vault.jwk_get("pvi_p6g5i3gtbvqvc3u6zugab6qs6r63tqf5")
@@ -1913,7 +1917,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A PangeaResponse where the state change object is returned in the
             response.result field. Available response fields can be found in our
-            [API documentation](https://pangea.cloud/docs/api/vault#change-state).
+            [API documentation](https://pangea.cloud/docs/api/vault/v1-general#/v1/state/change-post).
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -2014,7 +2018,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A `PangeaResponse` where the encrypted object is returned in the
             `response.result` field. Available response fields can be found in
-            our [API documentation](https://pangea.cloud/docs/api/vault#encrypt-structured).
+            our [API documentation](https://pangea.cloud/docs/api/vault/v1-keys#/v1/key/encrypt/structured-post).
 
         Raises:
             PangeaAPIException: If an API error happens.
@@ -2070,7 +2074,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A `PangeaResponse` where the decrypted object is returned in the
             `response.result` field. Available response fields can be found in
-            our [API documentation](https://pangea.cloud/docs/api/vault#decrypt-structured).
+            our [API documentation](https://pangea.cloud/docs/api/vault/v1-keys#/v1/key/decrypt/structured-post).
 
         Examples:
             data = {"field1": [1, 2, "kxcbC9E9IlgVaSCChPWUMgUC3ko=", "6FfI/LCzatLRLNAc8SuBK/TDnGxp"], "field2": "data2"}
@@ -2214,7 +2218,7 @@ class VaultAsync(ServiceBaseAsync):
         Returns:
             A `PangeaResponse` where the exported key is returned in the
             `response.result` field. Available response fields can be found in
-            our [API documentation](https://pangea.cloud/docs/api/vault#export).
+            our [API documentation](https://pangea.cloud/docs/api/vault/v1-general#/v1/export-post).
 
         Raises:
             PangeaAPIException: If an API error happens.

pangea/audit_logger.py

@@ -1,5 +1,7 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+from __future__ import annotations
+
 import logging
 
 import pangea.exceptions as pe
@@ -20,7 +22,7 @@ class AuditLogger(logging.Logger):
     """
 
     def __init__(self, *args, **kwargs):
-        super(AuditLogger, self).__init__(*args, **kwargs)
+        super().__init__(*args, **kwargs)
 
     def set_auditor(self, auditor: Audit):
         """Sets the internal Pangea Audit Service client instance

pangea/deep_verify.py

@@ -1,13 +1,16 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
 
+from __future__ import annotations
+
 import argparse
 import io
 import math
 import os
 import sys
+from collections.abc import Iterator
 from itertools import groupby
-from typing import Dict, Iterator, List, Optional, Set, TypedDict, Union
+from typing import Optional, TypedDict, Union
 
 import pangea.services.audit.util as audit_util
 from pangea.services import Audit
@@ -24,7 +27,7 @@ class Errors(TypedDict):
     buffer_missing: int
 
 
-root_hashes: Dict[int, str] = {}
+root_hashes: dict[int, str] = {}
 
 
 def num_lines(f: io.TextIOWrapper) -> int:
@@ -124,10 +127,7 @@ def get_root_hash(audit: Audit, tree_size: int) -> str:
 
 
 def print_error(msg: str, level: str = "error"):
-    if level == "warning":
-        dot = "🟡"
-    else:
-        dot = "🔴"
+    dot = "🟡" if level == "warning" else "🔴"
     print(f"{dot} {msg:200s}")
 
 
@@ -148,14 +148,14 @@ def deep_verify(audit: Audit, file: io.TextIOWrapper) -> Errors:
     }
 
     events = file_events(root_hashes, file)
-    events_by_idx: Union[List[Event], Iterator[Event]]
+    events_by_idx: Union[list[Event], Iterator[Event]]
     cold_indexes = SequenceFollower()
     for leaf_index, events_by_idx in groupby(events, lambda event: event.get("leaf_index")):
         events_by_idx = list(events_by_idx)
         buffer_lines = (cnt, cnt + len(events_by_idx) - 1)
         if leaf_index is None:
             print_error(
-                f"Lines {buffer_lines[0]}-{buffer_lines[1]} ({buffer_lines[1]-buffer_lines[0]+1}): Buffer was not persisted"
+                f"Lines {buffer_lines[0]}-{buffer_lines[1]} ({buffer_lines[1] - buffer_lines[0] + 1}): Buffer was not persisted"
             )
             errors["not_persisted"] += len(events_by_idx)
             cnt += len(events_by_idx)
@@ -164,8 +164,8 @@ def deep_verify(audit: Audit, file: io.TextIOWrapper) -> Errors:
         cold_indexes.add(leaf_index)
 
         cold_path_size: Optional[int] = None
-        hot_indexes: Set[int] = set()
-        for i, event in enumerate(events_by_idx):
+        hot_indexes: set[int] = set()
+        for _i, event in enumerate(events_by_idx):
             cnt += 1
             tree_size = get_tree_size(event)
             if tree_size not in root_hashes:
@@ -203,11 +203,11 @@ def deep_verify(audit: Audit, file: io.TextIOWrapper) -> Errors:
             errors["missing"] += len(hot_indexes_diff)
             print(f"missing hot indexes: {hot_indexes_diff}")
             print(f"hot_indexes: {hot_indexes} ")
-            print(f"events:")
+            print("events:")
             for e in events_by_idx:
                 print(e)
             print_error(
-                f"Lines {buffer_lines[0]}-{buffer_lines[1]} ({buffer_lines[1]-buffer_lines[0]}), Buffer #{cold_idx}: {len(hot_indexes_diff)} event(s) missing"
+                f"Lines {buffer_lines[0]}-{buffer_lines[1]} ({buffer_lines[1] - buffer_lines[0]}), Buffer #{cold_idx}: {len(hot_indexes_diff)} event(s) missing"
             )
 
     cold_holes = cold_indexes.holes()
@@ -232,7 +232,7 @@ def create_parser():
         "-f",
         required=True,
         type=argparse.FileType("r"),
-        help="Event input file. Must be a collection of " "JSON Objects separated by newlines",
+        help="Event input file. Must be a collection of JSON Objects separated by newlines",
     )
     return parser
 

pangea/deprecated.py

@@ -19,7 +19,7 @@ def pangea_deprecated(*args, **kwargs):
         def wrapper(*iargs, **ikwargs):
             return deprecated(*args, **kwargs)(f)(*iargs, **ikwargs)
 
-        setattr(wrapper, "_deprecated", kwargs)
+        wrapper._deprecated = kwargs
         return wrapper
 
     return decorator

pangea/dump_audit.py

@@ -7,7 +7,6 @@ import json
 import os
 import sys
 from datetime import datetime
-from typing import Tuple
 
 import dateutil.parser
 
@@ -103,7 +102,7 @@ def dump_after(audit: Audit, output: io.TextIOWrapper, start: datetime, last_eve
 
 def dump_page(
     audit: Audit, output: io.TextIOWrapper, start: datetime, end: datetime, first: bool = False
-) -> Tuple[datetime, int, bool, str, int]:
+) -> tuple[datetime, int, bool, str, int]:
     PAGE_SIZE = 1000
     print(start, end)
     print("Dumping...")
@@ -175,7 +174,7 @@ def parse_args(parser: argparse.ArgumentParser):
         raise ValueError("domain missing")
 
     if args.output is None:
-        args.output = open(f"dump-{datetime.now().strftime('%Y%m%d%H%M%S')}.jsonl", "w")
+        args.output = open(f"dump-{datetime.now().strftime('%Y%m%d%H%M%S')}.jsonl", "w")  # noqa: SIM115
 
     args.start = make_aware_datetime(args.start)
     args.end = make_aware_datetime(args.end)

pangea/exceptions.py

@@ -1,6 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
 
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 from typing import List, Optional
 
 from pangea.response import AcceptedResult, ErrorField, PangeaResponse
@@ -44,7 +47,7 @@ class PangeaAPIException(PangeaException):
     response: PangeaResponse
 
     def __init__(self, message: str, response: PangeaResponse):
-        super(PangeaAPIException, self).__init__(message)
+        super().__init__(message)
         self.response = response
 
     @property
@@ -92,7 +95,7 @@ class UnauthorizedException(PangeaAPIException):
 
     def __init__(self, service_name: str, response: PangeaResponse):
         message = f"User is not authorized to access service {service_name}"
-        super(UnauthorizedException, self).__init__(message, response)
+        super().__init__(message, response)
 
 
 class NotFound(PangeaAPIException):
@@ -100,20 +103,20 @@ class NotFound(PangeaAPIException):
 
     def __init__(self, url: str, response: PangeaResponse):
         message = f"Resource url:'{url}' not found"
-        super(NotFound, self).__init__(message, response)
+        super().__init__(message, response)
 
 
 class ServiceNotEnabledException(PangeaAPIException):
     def __init__(self, service_name: str, response: PangeaResponse):
         message = f"{service_name} is not enabled. Go to console.pangea.cloud/service/{service_name} to enable"
-        super(ServiceNotEnabledException, self).__init__(message, response)
+        super().__init__(message, response)
 
 
 class MissingConfigID(PangeaAPIException):
     """No config ID was provided in either token scopes or explicitly"""
 
     def __init__(self, service_name: str, response: PangeaResponse):
-        super(MissingConfigID, self).__init__(
+        super().__init__(
             f"Token did not contain a config scope for service {service_name}. Create a new token or provide a config ID explicitly in the service base",
             response,
         )

pangea/file_uploader.py

@@ -1,5 +1,9 @@
 # Copyright 2022 Pangea Cyber Corporation
 # Author: Pangea Cyber Corporation
+
+# TODO: Modernize.
+# ruff: noqa: UP006, UP035
+
 import io
 import logging
 from typing import Dict, Optional