1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
from __future__ import annotations
import logging
import time
from typing import Any
import httpx
logger = logging.getLogger(__name__)
class SourceHutClientError(RuntimeError):
"""Raised when a SourceHut GraphQL request fails."""
def _graphql_error_summary(errors: Any) -> str:
if not isinstance(errors, list):
return "unexpected error payload"
return f"{len(errors)} GraphQL error(s)"
class SourceHutGraphQLClient:
def __init__(
self,
endpoint: str,
token: str,
*,
timeout: float = 15.0,
max_retries: int = 2,
request_delay: float = 0.5,
transport: httpx.BaseTransport | None = None,
) -> None:
self.endpoint = endpoint
self.timeout = timeout
self.max_retries = max_retries
self.request_delay = request_delay
headers = {
"Authorization": f"Bearer {token}",
"Content-Type": "application/json",
}
self._client = httpx.Client(headers=headers, timeout=timeout, transport=transport)
def execute(self, query: str, variables: dict[str, Any] | None = None) -> dict[str, Any]:
payload = {"query": query, "variables": variables or {}}
attempts = self.max_retries + 1
for attempt in range(attempts):
if attempt > 0:
time.sleep(2 ** (attempt - 1))
elif self.request_delay > 0:
time.sleep(self.request_delay)
try:
response = self._client.post(self.endpoint, json=payload)
response.raise_for_status()
body = response.json()
except httpx.HTTPStatusError as exc:
logger.warning(
"SourceHut HTTP failure from %s on attempt %s/%s: status=%s",
self.endpoint,
attempt + 1,
attempts,
exc.response.status_code,
)
if exc.response.status_code >= 500 and attempt < attempts - 1:
continue
raise SourceHutClientError(f"HTTP error from SourceHut: {exc.response.status_code}") from exc
except httpx.HTTPError as exc:
logger.warning(
"SourceHut network failure from %s on attempt %s/%s",
self.endpoint,
attempt + 1,
attempts,
)
if attempt < attempts - 1:
continue
raise SourceHutClientError("Network error while contacting SourceHut") from exc
if "errors" in body:
logger.warning(
"SourceHut GraphQL failure from %s: %s",
self.endpoint,
_graphql_error_summary(body["errors"]),
)
raise SourceHutClientError(
f"GraphQL errors returned by SourceHut: {_graphql_error_summary(body['errors'])}"
)
return body.get("data", {})
raise SourceHutClientError("SourceHut request exhausted retries")
def close(self) -> None:
self._client.close()
|
