- Calling custom HTTP Exception on request failure

Author: MarcoMuellner

src/openapi_python_generator/language_converters/python/templates/apiconfig.jinja2

@@ -26,3 +26,12 @@ class APIConfig():
 {% else %}
         _access_token = value
 {% endif %}
+
+class HTTPException(Exception):
+    def __init__(self, status_code: int, message: str):
+        self.status_code = status_code
+        self.message = message
+        super().__init__(f"{status_code} {message}")
+
+    def __str__(self):
+        return f"{self.status_code} {self.message}"

src/openapi_python_generator/language_converters/python/templates/httpx.jinja2

@@ -29,7 +29,7 @@ with httpx.Client(base_url=base_path) as client:
     )
 
     if response.status_code != {{ return_type.status_code }}:
-        raise Exception(f'{{ operationId }} failed with status code: {response.status_code}')
+        raise HTTPException(response.status_code, f'{{ operationId }} failed with status code: {response.status_code}')
 {% if return_type.complex_type %}
 {% if return_type.list_type is none %}
     return {{ return_type.type.converted_type }}(**response.json())

src/openapi_python_generator/language_converters/python/templates/requests.jinja2

@@ -22,7 +22,7 @@ def {{ operation_id }}({{ params }}) -> {% if return_type.type is none or return
         {% endif %}
     )
     if response.status_code != {{ return_type.status_code }}:
-        raise Exception(f'{{ operationId }} failed with status code: {response.status_code}')
+        raise HTTPException(response.status_code, f'{{ operationId }} failed with status code: {response.status_code}')
 {% if return_type.complex_type %}
 {% if return_type.list_type is none %}
     return {{ return_type.type.converted_type }}(**response.json())

src/openapi_python_generator/language_converters/python/templates/service.jinja2

@@ -3,6 +3,6 @@ import {{ library_import }}
 import json
 
 from ..models import *
-from ..api_config import APIConfig
+from ..api_config import APIConfig, HTTPException
 
 {{ content | safe}}