refactor: set response status code type directly in response type def

Author: Ykrej

openapi_python_client/config.py

@@ -124,6 +124,6 @@ def from_sources(
             file_encoding=file_encoding,
             overwrite=overwrite,
             output_path=output_path,
-            allow_int_response_codes=config_file.allow_int_response_codes
+            allow_int_response_codes=config_file.allow_int_response_codes,
         )
         return config

openapi_python_client/parser/responses.py

@@ -14,6 +14,8 @@
 from .errors import ParseError, PropertyError
 from .properties import AnyProperty, Property, Schemas, property_from_data
 
+_STATUS_CODES = set(status.value for status in HTTPStatus)
+
 
 @define
 class Responses:
@@ -60,7 +62,7 @@ def __init__(self, *, pattern: str, code_range: tuple[int, int] | None):
         self.pattern = pattern
         self.range = code_range
         self.is_official = self.range is None or all(
-            code in HTTPStatus for code in range(self.range[0], self.range[1] + 1)
+            code in _STATUS_CODES for code in range(self.range[0], self.range[1] + 1)
         )
 
     @staticmethod

openapi_python_client/templates/endpoint_module.py.jinja

@@ -11,12 +11,11 @@ from ... import errors
 {{ relative }}
 {% endfor %}
 
-{% set all_standard_response_codes = endpoint.responses|map(attribute="status_code")|map(attribute="is_official")|all %}
-{% if config.allow_int_response_codes and not all_standard_response_codes %}
-HTTPStatus = int
+{% if config.allow_int_response_codes and not endpoint.responses|map(attribute="status_code")|map(attribute="is_official")|all %}
+{% set http_status_type = "int" %}
 {% else %}
 import http
-HTTPStatus = http.HTTPStatus
+{% set http_status_type = "http.HTTPStatus" %}
 {% endif %}
 
 {% from "endpoint_macros.py.jinja" import header_params, cookie_params, query_params,
@@ -99,9 +98,9 @@ def _parse_response(*, client: AuthenticatedClient | Client, response: httpx.Res
     {% endif %}
 
 
-def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[{{ return_string }}, HTTPStatus]:
+def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Response) -> Response[{{ return_string }}, {{ http_status_type }}]:
     return Response(
-        status_code=HTTPStatus(response.status_code),
+        status_code={{ http_status_type }}(response.status_code),
         content=response.content,
         headers=response.headers,
         parsed=_parse_response(client=client, response=response),
@@ -110,7 +109,7 @@ def _build_response(*, client: AuthenticatedClient | Client, response: httpx.Res
 
 def sync_detailed(
     {{ arguments(endpoint) | indent(4) }}
-) -> Response[{{ return_string }}, HTTPStatus]:
+) -> Response[{{ return_string }}, {{ http_status_type }}]:
     {{ docstring(endpoint, return_string, is_detailed=true) | indent(4) }}
 
     kwargs = _get_kwargs(
@@ -136,7 +135,7 @@ def sync(
 
 async def asyncio_detailed(
     {{ arguments(endpoint) | indent(4) }}
-) -> Response[{{ return_string }}, HTTPStatus]:
+) -> Response[{{ return_string }}, {{ http_status_type }}]:
     {{ docstring(endpoint, return_string, is_detailed=true) | indent(4) }}
 
     kwargs = _get_kwargs(