fix: Work around Enum breaking changes in 3.11.0

Author: dbanty

openapi_python_client/schema/parameter_location.py

@@ -1,10 +1,25 @@
-from enum import Enum
+# Python 3.11 has StrEnum but breaks the old `str, Enum` hack.
+# Unless this gets fixed, we need to have two implementations :(
+import sys
 
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
 
-class ParameterLocation(str, Enum):
-    """The places Parameters can be put when calling an Endpoint"""
+    class ParameterLocation(StrEnum):
+        """The places Parameters can be put when calling an Endpoint"""
 
-    QUERY = "query"
-    PATH = "path"
-    HEADER = "header"
-    COOKIE = "cookie"
+        QUERY = "query"
+        PATH = "path"
+        HEADER = "header"
+        COOKIE = "cookie"
+
+else:
+    from enum import Enum
+
+    class ParameterLocation(str, Enum):
+        """The places Parameters can be put when calling an Endpoint"""
+
+        QUERY = "query"
+        PATH = "path"
+        HEADER = "header"
+        COOKIE = "cookie"

openapi_python_client/templates/endpoint_module.py.jinja

@@ -61,7 +61,7 @@ def _get_kwargs(
 {% if parsed_responses %}
 def _parse_response(*, response: httpx.Response) -> Optional[{{ return_string }}]:
     {% for response in endpoint.responses %}
-    if response.status_code == {{ response.status_code }}:
+    if response.status_code == HTTPStatus.{{ response.status_code.name }}:
         {% import "property_templates/" + response.prop.template as prop_template %}
         {% if prop_template.construct %}
         {{ prop_template.construct(response.prop, response.source) | indent(8) }}

tests/test_parser/test_openapi.py

@@ -8,6 +8,7 @@
 from openapi_python_client.parser.errors import ParseError
 from openapi_python_client.parser.openapi import Endpoint, EndpointCollection
 from openapi_python_client.parser.properties import IntProperty, Parameters, Schemas
+from openapi_python_client.schema import ParameterLocation
 
 MODULE_NAME = "openapi_python_client.parser.openapi"