:feat: Implemented aiohttp

Author: MarcoMuellner

poetry.lock

@@ -51,10 +51,12 @@ pytest-cov = "^3.0.0"
 fastapi = "^0.78.0"
 uvicorn = "^0.18.1"
 respx = "^0.20.1"
+aiohttp = "^3.8.3"
 
 [tool.poetry.scripts]
 openapi-python-generator = "openapi_python_generator.__main__:main"
 
+
 [tool.coverage.paths]
 source = ["src", "*/site-packages"]
 tests = ["tests", "*/tests"]

pyproject.toml

@@ -0,0 +1,35 @@
+async def {{ operation_id }}({{ params }}) -> {% if return_type.type is none or return_type.type.converted_type is none %}None{% else %}{{ return_type.type.converted_type}}{% endif %}:
+    base_path = APIConfig.base_path
+    path = f'{{ path_name }}'
+    headers = {
+    'Content-Type': 'application/json',
+    'Accept': 'application/json',
+    'Authorization': f'Bearer { APIConfig.get_access_token() }',
+    }
+    query_params : Dict[str,Any] = {
+    {% if query_params|length > 0 %}
+        {{ query_params | join(',\n') | safe }}
+    {% endif %}
+    }
+
+    query_params = {key:value for (key,value) in query_params.items() if value is not None}
+
+    async with aiohttp.ClientSession(headers=headers) as session:
+        async with session.request(
+            '{{ method }}',
+            base_path + path,
+            params=query_params,
+        ) as inital_response:
+            if inital_response.status != {{ return_type.status_code }}:
+                raise HTTPException(inital_response.status, f'{{ operationId }} failed with status code: {inital_response.status}')
+            response = await inital_response.json()
+
+{% if return_type.complex_type %}
+{% if return_type.list_type is none %}
+            return {{ return_type.type.converted_type }}(**response.json()) if response.json() is not None else {{ return_type.type.converted_type }}()
+{% else %}
+            return [{{ return_type.list_type }}(**item) for item in response.json()]
+{% endif %}
+{% else %}
+            return response.json()
+{% endif %}

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

@@ -15,10 +15,7 @@ def runner() -> CliRunner:
 
 @pytest.mark.parametrize(
     "library",
-    [
-        HTTPLibrary.httpx,
-        HTTPLibrary.requests,
-    ],
+    [HTTPLibrary.httpx, HTTPLibrary.requests, HTTPLibrary.aiohttp],
 )
 def test_issue_7(runner: CliRunner, model_data_with_cleanup, library) -> None:
     """

tests/regression/test_issue_7.py

@@ -35,6 +35,8 @@ def test_get_auth_token_without_env(model_data_with_cleanup):
         (HTTPLibrary.requests, False),
         (HTTPLibrary.httpx, True),
         (HTTPLibrary.requests, True),
+        (HTTPLibrary.aiohttp, True),
+        (HTTPLibrary.aiohttp, False),
     ],
 )
 @respx.mock

tests/test_generated_code.py

@@ -16,10 +16,7 @@ def runner() -> CliRunner:
 
 @pytest.mark.parametrize(
     "library",
-    [
-        HTTPLibrary.httpx,
-        HTTPLibrary.requests,
-    ],
+    [HTTPLibrary.httpx, HTTPLibrary.requests, HTTPLibrary.aiohttp],
 )
 def test_main_succeeds(runner: CliRunner, model_data_with_cleanup, library) -> None:
     """It exits with a status code of zero."""