ab-openapi-python-generator 2.1.4__py3-none-any.whl → 2.2.0__py3-none-any.whl

ab_openapi_python_generator/language_converters/python/templates/apiconfig.jinja2

@@ -1,38 +0,0 @@
-{% if env_token_name is not none %}import os{% endif %}
-
-from pydantic import BaseModel, Field
-from typing import Optional, Union
-
-class APIConfig(BaseModel):
-    base_path: str = {% if servers|length > 0 %} '{{ servers[0].url }}' {% else %} 'NO SERVER' {% endif %}
-
-    verify: Union[bool, str] = True
-{% if env_token_name is none %}
-    access_token : Optional[str] = None
-{% endif %}
-
-    def get_access_token(self) -> Optional[str]:
-{% if env_token_name is not none %}
-        try:
-            return os.environ['{{ env_token_name }}']
-        except KeyError:
-            return None
-{% else %}
-        return self.access_token
-{% endif %}
-
-    def set_access_token(self, value : str):
-{% if env_token_name is not none %}
-        raise Exception("This client was generated with an environment variable for the access token. Please set the environment variable '{{ env_token_name }}' to the access token.")
-{% else %}
-        self.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}"

ab_openapi_python_generator/language_converters/python/templates/apiconfig_pydantic_2.jinja2

@@ -1,42 +0,0 @@
-{% if env_token_name is not none %}import os{% endif %}
-
-from pydantic import BaseModel, Field
-from typing import Optional, Union
-
-class APIConfig(BaseModel):
-    model_config = {
-        "validate_assignment": True
-    }
-
-    base_path: str = {% if servers|length > 0 %} '{{ servers[0].url }}' {% else %} 'NO SERVER' {% endif %}
-
-    verify: Union[bool, str] = True
-{% if env_token_name is none %}
-    access_token : Optional[str] = None
-{% endif %}
-
-    def get_access_token(self) -> Optional[str]:
-{% if env_token_name is not none %}
-        try:
-            return os.environ['{{ env_token_name }}']
-        except KeyError:
-            return None
-{% else %}
-        return self.access_token
-{% endif %}
-
-    def set_access_token(self, value : str):
-{% if env_token_name is not none %}
-        raise Exception("This client was generated with an environment variable for the access token. Please set the environment variable '{{ env_token_name }}' to the access token.")
-{% else %}
-        self.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}"

ab_openapi_python_generator/language_converters/python/templates/httpx.jinja2

@@ -1,126 +0,0 @@
-{# NOTE: this template renders ONE function. It must be valid standalone python. #}
-
-{% if async_client %}async {% endif %}def {{ operation_id }}(
-    api_config_override: Optional[APIConfig] = None{% if params.strip() %}, *, {{ params.rstrip(', ') }}{% endif %}
-) -> {% if is_sse %}{% if async_client %}AsyncGenerator[Any, None]{% else %}Iterator[Any]{% endif %}{% else %}{% if return_type.type is none or return_type.type.converted_type is none %}None{% else %}{{ return_type.type.converted_type }}{% endif %}{% endif %}:
-    api_config = api_config_override if api_config_override else APIConfig()
-
-    base_path = api_config.base_path
-    path = f'{{ path_name }}'
-
-    headers = {
-        "Content-Type": "application/json",
-{% if is_sse %}
-        "Accept": "text/event-stream",
-{% else %}
-        "Accept": "application/json",
-{% endif %}
-        "Authorization": f"Bearer { api_config.get_access_token() }",
-        {{ header_params | join(',\n') | safe }}
-    }
-
-    query_params: Dict[str, Any] = {}
-    query_params = {key: value for (key, value) in query_params.items() if value is not None}
-
-{% if is_sse %}
-    import json
-
-{% if async_client %}
-    async with httpx.AsyncClient(base_url=base_path, verify=api_config.verify) as client:
-        async with client.stream(
-            "{{ method }}",
-            httpx.URL(path),
-            headers=headers,
-            params=query_params,
-{% if body_param %}
-            json={{ body_param }},
-{% endif %}
-        ) as response:
-            if response.status_code != {{ return_type.status_code }}:
-                raise HTTPException(
-                    response.status_code,
-                    f"{{ operation_id }} failed with status code: {response.status_code}",
-                )
-
-            async for line in response.aiter_lines():
-                if not line:
-                    continue
-                if line.startswith("data:"):
-                    payload = line[len("data:"):].strip()
-                    if not payload:
-                        continue
-                    if payload == "[DONE]":
-                        break
-                    try:
-                        yield json.loads(payload)
-                    except Exception:
-                        yield payload
-{% else %}
-    with httpx.Client(base_url=base_path, verify=api_config.verify) as client:
-        with client.stream(
-            "{{ method }}",
-            httpx.URL(path),
-            headers=headers,
-            params=query_params,
-{% if body_param %}
-            json={{ body_param }},
-{% endif %}
-        ) as response:
-            if response.status_code != {{ return_type.status_code }}:
-                raise HTTPException(
-                    response.status_code,
-                    f"{{ operation_id }} failed with status code: {response.status_code}",
-                )
-
-            for line in response.iter_lines():
-                if not line:
-                    continue
-                if line.startswith("data:"):
-                    payload = line[len("data:"):].strip()
-                    if not payload:
-                        continue
-                    if payload == "[DONE]":
-                        break
-                    try:
-                        yield json.loads(payload)
-                    except Exception:
-                        yield payload
-{% endif %}
-
-{% else %}
-{% if async_client %}
-    async with httpx.AsyncClient(base_url=base_path, verify=api_config.verify) as client:
-        response = await client.request(
-{% else %}
-    with httpx.Client(base_url=base_path, verify=api_config.verify) as client:
-        response = client.request(
-{% endif %}
-            "{{ method }}",
-            httpx.URL(path),
-            headers=headers,
-            params=query_params,
-{% if body_param %}
-            json={{ body_param }},
-{% endif %}
-        )
-
-    if response.status_code != {{ return_type.status_code }}:
-        raise HTTPException(
-            response.status_code,
-            f"{{ operation_id }} failed with status code: {response.status_code}",
-        )
-
-    body = None if {{ return_type.status_code }} == 204 else response.json()
-
-{% if return_type.type is none or return_type.type.converted_type is none %}
-    return None
-{% elif return_type.complex_type %}
-{% if return_type.list_type is none %}
-    return {{ return_type.type.converted_type }}(**body) if body is not None else {{ return_type.type.converted_type }}()
-{% else %}
-    return [{{ return_type.list_type }}(**item) for item in body]
-{% endif %}
-{% else %}
-    return body
-{% endif %}
-{% endif %}

ab_openapi_python_generator/language_converters/python/templates/requests.jinja2

@@ -1,50 +0,0 @@
-def {{ operation_id }}(api_config_override : Optional[APIConfig] = None{% if params.strip() %}, *, {{ params.rstrip(', ') }}{% endif %}) -> {% if return_type.type is none or return_type.type.converted_type is none %}None{% else %}{{ return_type.type.converted_type}}{% endif %}:
-    api_config = api_config_override if api_config_override else APIConfig()
-
-    base_path = api_config.base_path
-    path = f'{{ path_name }}'
-    headers = {
-        'Content-Type': 'application/json',
-        'Accept': 'application/json',
-        'Authorization': f'Bearer { api_config.get_access_token() }',
-        {{ header_params | join(',\n') | safe }}
-        }
-    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}
-
-    response = requests.request(
-        '{{ method }}',
-        f'{base_path}{path}',
-        headers=headers,
-        params=query_params,
-        verify=api_config.verify,
-        {% if body_param %}
-        {% if use_orjson %}
-        content=orjson.dumps({{ body_param }})
-        {% else %}
-        json = {{ body_param }}
-        {% endif %}
-        {% endif %}
-    )
-    if response.status_code != {{ return_type.status_code }}:
-        raise HTTPException(response.status_code, f'{{ operation_id }} failed with status code: {response.status_code}')
-    else:
-        {# Conditional body parsing: avoid calling .json() for 204 #}
-        body = None if {{ return_type.status_code }} == 204 else response.json()
-
-{% if return_type.type is none or return_type.type.converted_type is none %}
-    return None
-{% elif return_type.complex_type %}
-    {%- if return_type.list_type is none %}
-    return {{ return_type.type.converted_type }}(**body) if body is not None else {{ return_type.type.converted_type }}()
-    {%- else %}
-    return [{{ return_type.list_type }}(**item) for item in body]
-    {%- endif %}
-{% else %}
-    return body
-{% endif %}

ab_openapi_python_generator/language_converters/python/templates/service.jinja2

@@ -1,12 +0,0 @@
-from typing import *
-import {{ library_import }}
-
-{% if use_orjson %}
-import orjson
-from uuid import UUID
-{% endif %}
-
-from ..models import *
-from ..api_config import APIConfig, HTTPException
-
-{{ content | safe}}