fix(client): move payload to GCClient

GitGuardian · Jan 9, 2024 · 3052fd4 · 3052fd4
1 parent 7c53cc9
commit 3052fd4
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/pygitguardian/client.py b/pygitguardian/client.py
@@ -11,7 +11,12 @@
 import requests
 from requests import Response, Session, codes
 
-from .config import DEFAULT_API_VERSION, DEFAULT_BASE_URI, DEFAULT_TIMEOUT
+from .config import (
+    DEFAULT_API_VERSION,
+    DEFAULT_BASE_URI,
+    DEFAULT_TIMEOUT,
+    MAXIMUM_PAYLOAD_SIZE,
+)
 from .iac_models import (
     IaCDiffScanResult,
     IaCScanParameters,
@@ -190,6 +195,7 @@ def __init__(
                 "Authorization": f"Token {api_key}",
             },
         )
+        self.maximum_payload_size = MAXIMUM_PAYLOAD_SIZE
         self.secret_scan_preferences = SecretScanPreferences()
 
     def request(
@@ -584,10 +590,9 @@ def read_metadata(self) -> Optional[Detail]:
             result.status_code = resp.status_code
             return result
         metadata = ServerMetadata.from_dict(resp.json())
-        if "general__maximum_payload_size" in metadata.preferences:
-            metadata.secret_scan_preferences.general__maximum_payload_size = (
-                metadata.preferences["general__maximum_payload_size"]
-            )
+        self.maximum_payload_size = metadata.preferences.get(
+            "general__maximum_payload_size", MAXIMUM_PAYLOAD_SIZE
+        )
         self.secret_scan_preferences = metadata.secret_scan_preferences
         return None
 

diff --git a/pygitguardian/config.py b/pygitguardian/config.py
@@ -4,4 +4,4 @@
 
 MULTI_DOCUMENT_LIMIT = 20
 DOCUMENT_SIZE_THRESHOLD_BYTES = 1048576  # 1MB
-GENERAL_MAXIMUM_PAYLOAD = 2621440  # 25MB
+MAXIMUM_PAYLOAD_SIZE = 2621440  # 25MB
diff --git a/pygitguardian/models.py b/pygitguardian/models.py
@@ -16,11 +16,7 @@
 )
 from typing_extensions import Self
 
-from .config import (
-    DOCUMENT_SIZE_THRESHOLD_BYTES,
-    GENERAL_MAXIMUM_PAYLOAD,
-    MULTI_DOCUMENT_LIMIT,
-)
+from .config import DOCUMENT_SIZE_THRESHOLD_BYTES, MULTI_DOCUMENT_LIMIT
 
 
 class ToDictMixin:
@@ -662,7 +658,6 @@ def __repr__(self) -> str:
 class SecretScanPreferences:
     maximum_document_size: int = DOCUMENT_SIZE_THRESHOLD_BYTES
     maximum_documents_per_scan: int = MULTI_DOCUMENT_LIMIT
-    general__maximum_payload_size: int = GENERAL_MAXIMUM_PAYLOAD
 
 
 @dataclass