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
|
from typing import (
Any,
Dict,
List,
)
from graphql import ExecutionResult
from ...graphql_request import GraphQLRequest
from ..exceptions import (
TransportProtocolError,
)
def _raise_protocol_error(result_text: str, reason: str) -> None:
raise TransportProtocolError(
f"Server did not return a valid GraphQL result: " f"{reason}: " f"{result_text}"
)
def _validate_answer_is_a_list(results: Any) -> None:
if not isinstance(results, list):
_raise_protocol_error(
str(results),
"Answer is not a list",
)
def _validate_data_and_errors_keys_in_answers(results: List[Dict[str, Any]]) -> None:
for result in results:
if "errors" not in result and "data" not in result:
_raise_protocol_error(
str(results),
'No "data" or "errors" keys in answer',
)
def _validate_every_answer_is_a_dict(results: List[Dict[str, Any]]) -> None:
for result in results:
if not isinstance(result, dict):
_raise_protocol_error(str(results), "Not every answer is dict")
def _validate_num_of_answers_same_as_requests(
reqs: List[GraphQLRequest],
results: List[Dict[str, Any]],
) -> None:
if len(reqs) != len(results):
_raise_protocol_error(
str(results),
(
"Invalid number of answers: "
f"{len(results)} answers received for {len(reqs)} requests"
),
)
def _answer_to_execution_result(result: Dict[str, Any]) -> ExecutionResult:
return ExecutionResult(
errors=result.get("errors"),
data=result.get("data"),
extensions=result.get("extensions"),
)
def get_batch_execution_result_list(
reqs: List[GraphQLRequest],
answers: List,
) -> List[ExecutionResult]:
_validate_answer_is_a_list(answers)
_validate_num_of_answers_same_as_requests(reqs, answers)
_validate_every_answer_is_a_dict(answers)
_validate_data_and_errors_keys_in_answers(answers)
return [_answer_to_execution_result(answer) for answer in answers]
|
