pangea-sdk 6.2.0b1__py3-none-any.whl → 6.3.0__py3-none-any.whl

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
@@ -55,7 +59,6 @@ from pangea.services.vault.models.common import (
     JWTVerifyResult,
     ListRequest,
     ListResult,
-    Metadata,
     PangeaToken,
     PangeaTokenRotateRequest,
     RequestManualRotationState,
@@ -164,7 +167,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 +200,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 +286,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
@@ -315,7 +318,7 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: str | None = None,
         enabled: bool | None = None,
@@ -345,7 +348,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
@@ -391,7 +394,7 @@ class VaultAsync(ServiceBaseAsync):
         result_class: type[TResult] = SecretStoreResult,  # type: ignore[assignment]
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
         **kwargs: Any,
@@ -416,22 +419,22 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
     ) -> PangeaResponse[Secret]:
         """
         Store secret
 
-        Store a secret.
+        Store a secret as a Vault item.
 
         Args:
-            secret: The secret value.
-            name: The name of this item.
-            folder: The folder where this item is stored.
-            metadata: User-provided metadata.
-            tags: A list of user-defined tags.
-            disabled_at: Timestamp indicating when the item will be disabled.
+            secret: Secret value
+            name: Name of the item
+            folder: Folder where the item is stored
+            metadata: Metadata provided by the user
+            tags: List of user-defined tags
+            disabled_at: Timestamp indicating when the item will be disabled
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -457,25 +460,38 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
         rotation_frequency: str | None = None,
-        rotation_state: RotationState | None = None,
+        rotation_state: Literal["deactivated", "destroyed", "inherited"] | None = None,
         rotation_grace_period: str | None = None,
     ) -> PangeaResponse[PangeaToken]:
         """
         Store secret
 
-        Store a Pangea token.
+        Store a Pangea token as a Vault item.
 
         Args:
-            token: The Pangea token value.
-            name: The name of this item.
-            folder: The folder where this item is stored.
-            metadata: User-provided metadata.
-            tags: A list of user-defined tags.
-            disabled_at: Timestamp indicating when the item will be disabled.
+            token: Pangea token value
+            name: Name of the item
+            folder: Folder where the item is stored
+            metadata: Metadata provided by the user
+            tags: List of user-defined tags
+            disabled_at: Timestamp indicating when the item will be disabled
+            rotation_frequency: Time interval between item rotations, provided
+              as a positive number followed by a time unit: `secs`, `mins`, `hrs`,
+              `days`, `weeks`, `months`, or `years`. You can use abbreviations
+              like `1d`. Omit to inherit from the parent folder or default
+              settings. Set to `never` to disable rotation.
+            rotation_state: Target state for the previous version after
+                rotation. Set to `inherited` to inherit from the parent folder
+                or default settings.
+            rotation_grace_period: Grace period for the previous version,
+              provided as a positive number followed by a time unit: `secs`,
+              `mins`, `hrs`, `days`, `weeks`, `months`, or `years`. You can use
+              abbreviations like `1d`. Omit to inherit from the parent folder or
+              default settings.
 
         Raises:
             PangeaAPIException: If an API Error happens
@@ -506,7 +522,7 @@ class VaultAsync(ServiceBaseAsync):
         *,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         disabled_at: datetime.datetime | None = None,
         rotation_frequency: str | None = None,
@@ -668,7 +684,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeySigningAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -714,7 +730,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyEncryptionAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -760,7 +776,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyJwtAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -806,7 +822,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyPkiAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -852,7 +868,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -898,7 +914,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyEncryptionAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -944,7 +960,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyJwtAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -990,7 +1006,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyFpeAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1036,7 +1052,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: SymmetricKeyAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1081,7 +1097,7 @@ class VaultAsync(ServiceBaseAsync):
         algorithm: AsymmetricKeyAlgorithm | SymmetricKeyAlgorithm,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1147,7 +1163,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1197,7 +1213,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1247,7 +1263,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1297,7 +1313,7 @@ class VaultAsync(ServiceBaseAsync):
         private_key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1346,7 +1362,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1394,7 +1410,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1442,7 +1458,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1499,7 +1515,7 @@ class VaultAsync(ServiceBaseAsync):
         key: str | None = None,
         name: str | None = None,
         folder: str | None = None,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState | None = RequestRotationState.INHERITED,
@@ -1680,7 +1696,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 +1732,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 +1766,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 +1804,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 +1842,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 +1867,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 +1897,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 +1929,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
@@ -1937,7 +1953,7 @@ class VaultAsync(ServiceBaseAsync):
         name: str,
         folder: str,
         *,
-        metadata: Metadata | None = None,
+        metadata: Mapping[str, str] | None = None,
         tags: Tags | None = None,
         rotation_frequency: str | None = None,
         rotation_state: RequestRotationState = RequestRotationState.INHERITED,
@@ -2014,7 +2030,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 +2086,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 +2230,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