11from http import HTTPStatus
2- from typing import Any, Dict, List, Optional, Union, cast
2+ from typing import Any, Dict, List, Union, cast
33from time import sleep
44import random
55
66import httpx
77
88from ...client import AuthenticatedClient, Client
99from ...types import Response, UNSET
10+ from ... import errors
1011
1112{% for relative in endpoint .relative_imports %}
1213{{ relative }}
1314{% endfor %}
1415
16+ SLEEP_TIME = 0.05
17+ NUM_RETRIES = 3
18+
1519{% from "endpoint_macros.py.jinja" import header_params , cookie_params , query_params , json_body , multipart_body ,
1620 arguments , client , kwargs , parse_response , docstring %}
1721
18- {% set return_string = endpoint .response_type () %}
19- {% set return_type = endpoint .responses [0].prop .get_type_string () if endpoint .responses [0].prop .get_type_string () != "Any" else None %}
20- {% set parsed_responses = (endpoint .responses | length > 0) and return_string != "Any" %}
22+ {% set return_string = endpoint .responses [0].prop .get_type_string () if endpoint .responses [0].prop .get_type_string () != "Any" else "None" %}
23+ {% set parsed_responses = (endpoint .responses | length > 0) %}
2124
2225def _get_kwargs(
2326 {{ arguments(endpoint) | indent(4) }}
@@ -48,7 +51,8 @@ def _get_kwargs(
4851 "headers": headers,
4952 "cookies": cookies,
5053 "timeout": client.get_timeout(),
51- {% if endpoint .form_body_class %}
54+ "follow_redirects": client.follow_redirects,
55+ {% if endpoint .form_body %}
5256 "data": form_data.to_dict(),
5357 {% elif endpoint .multipart_body %}
5458 "files": {{ "multipart_" + endpoint.multipart_body.python_name }},
@@ -61,70 +65,72 @@ def _get_kwargs(
6165 }
6266
6367
64- {% if parsed_responses %}
65- def _parse_response(*, response: httpx.Response) -> {{ return_type }}:
66- {% for response in endpoint .responses if response .status_code >= 299 %}
67- if response.status_code == {{ response.status_code }}:
68- raise {{ response.prop.class_info.name }}({{ response.source }}, response.status_code)
68+ def _parse_response(*, client: Client, response: httpx.Response) -> {% if return_string == "None" %} Any{% else %} {{ return_string }}{% endif %} :
69+ {% for response in endpoint .responses %}
70+ if response.status_code == HTTPStatus.{{ response.status_code.name }}:
71+ {% import "property_templates/" + response .prop .template as prop_template %}
72+ {% if 200 <= response .status_code <= 299 %}
73+ {% if prop_template .construct %}
74+ {{ prop_template.construct(response.prop, response.source) | indent(8) }}
75+ {% else %}
76+ {{ response.prop.python_name }} = cast({{ response.prop.get_type_string() }}, {{ response.source }})
77+ {% endif %}
78+ return {{ response.prop.python_name }}
79+ {% else %}
80+ raise {{ response.prop.get_type_string() }}.init_exception({{ response.source }}, response.status_code)
81+ {% endif %}
6982 {% endfor %}
70- {% set ok_response = endpoint .responses [0]%}
71- {% import "property_templates/" + ok_response .prop .template as ok_response_prop_template %}
72- {% if ok_response .prop and ok_response .prop .required_properties == [] and ok_response .prop .optional_properties == [] %}
73- {{ ok_response.prop.python_name }} = {{ ok_response.prop.class_info.name }}({{ ok_response.source }})
74- {% elif ok_response_prop_template .construct %}
75- {{ ok_response_prop_template.construct(ok_response.prop, ok_response.source) | indent(8) }}
76- {% else %}
77- {{ ok_response.prop.python_name }} = None
78- {% endif %}
79- return {{ok_response.prop.python_name}}
80- {% endif %}
83+ raise errors.UnexpectedStatus(response.status_code, response.content)
84+
85+
86+ def _build_response(*, client: Client, response: httpx.Response) -> Response[{{ return_string }}]:
87+ return Response(
88+ status_code=HTTPStatus(response.status_code),
89+ content=response.content,
90+ headers=response.headers,
91+ parsed=_parse_response(client=client, response=response),
92+ )
8193
82- sleep_time = 0.05
83- num_retries = 3
8494
8595def sync_detailed(
8696 {{ arguments(endpoint) | indent(4) }}
87- ) -> {{return_type}} :
88- {{ docstring(endpoint, return_type ) | indent(4) }}
97+ ) -> Response[{{ return_string }}] :
98+ {{ docstring(endpoint, return_string, is_detailed=true ) | indent(4) }}
8999
90100 kwargs = _get_kwargs(
91101 {{ kwargs(endpoint) }}
92102 )
93103
94104 kwargs['headers'] = {'svix-req-id':f"{random.getrandbits(32)}", **kwargs['headers']}
95105
96- retry_count = 0
97- for retry in range(num_retries):
106+ for retry_count in range(NUM_RETRIES):
98107 response = httpx.request(
99108 verify=client.verify_ssl,
100109 **kwargs,
101110 )
102- if response.status_code >= 500 and retry < num_retries:
103- retry_count +=1
104- kwargs['headers']['svix-retry-count']= str(retry_count)
105- sleep(sleep_time)
106- sleep_time = sleep_time * 2
111+ if response.status_code >= 500:
112+ kwargs['headers']['svix-retry-count'] = str(retry_count)
113+ sleep(SLEEP_TIME * (2 ** retry_count))
107114 else:
108115 break
109116
110-
111- return _parse_response(response=response)
117+ return _build_response(client=client, response=response)
112118
113119{% if parsed_responses %}
114120def sync(
115121 {{ arguments(endpoint) | indent(4) }}
116- ) -> {{ return_type }}:
117- {{ docstring(endpoint, return_type ) | indent(4) }}
122+ ) -> {{ return_string }}:
123+ {{ docstring(endpoint, return_string, is_detailed=false ) | indent(4) }}
118124
119125 return sync_detailed(
120126 {{ kwargs(endpoint) }}
121- )
127+ ).parsed
122128{% endif %}
123129
124130async def asyncio_detailed(
125131 {{ arguments(endpoint) | indent(4) }}
126- ) -> {{ return_type }}:
127- {{ docstring(endpoint, return_type ) | indent(4) }}
132+ ) -> Response[ {{ return_string }}] :
133+ {{ docstring(endpoint, return_string, is_detailed=true ) | indent(4) }}
128134
129135 kwargs = _get_kwargs(
130136 {{ kwargs(endpoint) }}
@@ -133,28 +139,25 @@ async def asyncio_detailed(
133139 kwargs['headers'] = {'svix-req-id':f"{random.getrandbits(32)}", **kwargs['headers']}
134140
135141 async with httpx.AsyncClient(verify=client.verify_ssl) as _client:
136- retry_count = 0
137- for retry in range(num_retries):
142+ for retry_count in range(NUM_RETRIES):
138143 response = await _client.request(
139144 **kwargs
140145 )
141- if response.status_code >= 500 and retry < num_retries:
142- retry_count +=1
143- kwargs['headers']['svix-retry-count']= str(retry_count)
144- sleep(sleep_time)
145- sleep_time = sleep_time * 2
146+ if response.status_code >= 500:
147+ kwargs['headers']['svix-retry-count'] = str(retry_count)
148+ sleep(SLEEP_TIME * (2 ** retry_count))
146149 else:
147150 break
148151
149- return _parse_response( response=response)
152+ return _build_response(client=client, response=response)
150153
151154{% if parsed_responses %}
152155async def asyncio(
153156 {{ arguments(endpoint) | indent(4) }}
154- ) -> {{ return_type }}:
155- {{ docstring(endpoint, return_type ) | indent(4) }}
157+ ) -> {{ return_string }}:
158+ {{ docstring(endpoint, return_string, is_detailed=false ) | indent(4) }}
156159
157160 return (await asyncio_detailed(
158161 {{ kwargs(endpoint) }}
159- ))
160- {% endif %}
162+ )).parsed
163+ {% endif %}
0 commit comments