Added logic for proper parsing

Author: Alfonso Tankia

openapi_python_client/parser/openapi.py

@@ -94,7 +94,7 @@ class Endpoint:
     path_parameters: List[Property] = field(default_factory=list)
     header_parameters: List[Property] = field(default_factory=list)
     responses: List[Response] = field(default_factory=list)
-    yaml_body: Optional[Reference] = None
+    yaml_body: Optional[Property] = None
     form_body_reference: Optional[Reference] = None
     json_body: Optional[Property] = None
     multipart_body_reference: Optional[Reference] = None

openapi_python_client/parser/responses.py

@@ -3,7 +3,6 @@
 from typing import Tuple, Union
 
 import attr
-import yaml
 
 from .. import schema as oai
 from .errors import ParseError, PropertyError
@@ -23,7 +22,7 @@ class Response:
     "application/json": "response.json()",
     "application/octet-stream": "response.content",
     "text/html": "response.text",
-    "text/yaml": "response.text",
+    "text/yaml": "yaml.safe_load(response.text.encode('utf-8'))",
 }
 
 

openapi_python_client/templates/endpoint_macros.pyi

@@ -56,6 +56,17 @@ if {% if not property.required %}not isinstance({{ property_name }}, Unset) and
 {% endif %}
 {% endmacro %}
 
+{% macro yaml_body(endpoint) %}
+{% if endpoint.yaml_body %}
+    {% set property = endpoint.yaml_body %}
+    {% set destination = "yaml_" + property.python_name %}
+    {% if property.template %}
+        {% from "property_templates/" + property.template import transform %}
+{{ transform(property, property.python_name, destination) }}
+    {% endif %}
+{% endif %}
+{% endmacro %}
+
 {% macro return_type(endpoint) %}
 {% if endpoint.responses | length == 0 %}
 None

openapi_python_client/templates/endpoint_module.pyi

@@ -1,6 +1,7 @@
 from typing import Any, Dict, List, Optional, Union, cast
 
 import httpx
+import yaml
 from attr import asdict
 
 from ...client import AuthenticatedClient, Client
@@ -10,7 +11,7 @@ from ...types import Response
 {{ relative }}
 {% endfor %}
 
-{% from "endpoint_macros.pyi" import header_params, query_params, json_body, return_type, arguments, client, kwargs, parse_response %}
+{% from "endpoint_macros.pyi" import header_params, query_params, json_body, yaml_body, return_type, arguments, client, kwargs, parse_response %}
 
 {% set return_string = return_type(endpoint) %}
 {% set parsed_responses = (endpoint.responses | length > 0) and return_string != "None" %}
@@ -33,6 +34,8 @@ def _get_kwargs(
 
     {{ json_body(endpoint) | indent(4) }}
 
+    {{ yaml_body(endpoint) | indent(4) }}
+
     return {
         "url": url,
         "headers": headers,