Skip to content

Commit 3052fd4

Browse files
committed
fix(client): move payload to GCClient
1 parent 7c53cc9 commit 3052fd4

File tree

3 files changed

+12
-12
lines changed

3 files changed

+12
-12
lines changed

pygitguardian/client.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,12 @@
1111
import requests
1212
from requests import Response, Session, codes
1313

14-
from .config import DEFAULT_API_VERSION, DEFAULT_BASE_URI, DEFAULT_TIMEOUT
14+
from .config import (
15+
DEFAULT_API_VERSION,
16+
DEFAULT_BASE_URI,
17+
DEFAULT_TIMEOUT,
18+
MAXIMUM_PAYLOAD_SIZE,
19+
)
1520
from .iac_models import (
1621
IaCDiffScanResult,
1722
IaCScanParameters,
@@ -190,6 +195,7 @@ def __init__(
190195
"Authorization": f"Token {api_key}",
191196
},
192197
)
198+
self.maximum_payload_size = MAXIMUM_PAYLOAD_SIZE
193199
self.secret_scan_preferences = SecretScanPreferences()
194200

195201
def request(
@@ -584,10 +590,9 @@ def read_metadata(self) -> Optional[Detail]:
584590
result.status_code = resp.status_code
585591
return result
586592
metadata = ServerMetadata.from_dict(resp.json())
587-
if "general__maximum_payload_size" in metadata.preferences:
588-
metadata.secret_scan_preferences.general__maximum_payload_size = (
589-
metadata.preferences["general__maximum_payload_size"]
590-
)
593+
self.maximum_payload_size = metadata.preferences.get(
594+
"general__maximum_payload_size", MAXIMUM_PAYLOAD_SIZE
595+
)
591596
self.secret_scan_preferences = metadata.secret_scan_preferences
592597
return None
593598

pygitguardian/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@
44

55
MULTI_DOCUMENT_LIMIT = 20
66
DOCUMENT_SIZE_THRESHOLD_BYTES = 1048576 # 1MB
7-
GENERAL_MAXIMUM_PAYLOAD = 2621440 # 25MB
7+
MAXIMUM_PAYLOAD_SIZE = 2621440 # 25MB

pygitguardian/models.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,7 @@
1616
)
1717
from typing_extensions import Self
1818

19-
from .config import (
20-
DOCUMENT_SIZE_THRESHOLD_BYTES,
21-
GENERAL_MAXIMUM_PAYLOAD,
22-
MULTI_DOCUMENT_LIMIT,
23-
)
19+
from .config import DOCUMENT_SIZE_THRESHOLD_BYTES, MULTI_DOCUMENT_LIMIT
2420

2521

2622
class ToDictMixin:
@@ -662,7 +658,6 @@ def __repr__(self) -> str:
662658
class SecretScanPreferences:
663659
maximum_document_size: int = DOCUMENT_SIZE_THRESHOLD_BYTES
664660
maximum_documents_per_scan: int = MULTI_DOCUMENT_LIMIT
665-
general__maximum_payload_size: int = GENERAL_MAXIMUM_PAYLOAD
666661

667662

668663
@dataclass

0 commit comments

Comments
 (0)