fix: Fixed parsing logic for yaml files to use PyYaml instead of ruamel.yaml BNCH-37249

openapi_python_client/templates/endpoint_module.pyi

@@ -2,7 +2,7 @@ from typing import Any, Dict, List, Optional, Union, cast
 
 import httpx
 from attr import asdict
-from ruamel.yaml import YAML
+import yaml
 
 from ...client import AuthenticatedClient, Client
 from ...types import Response

openapi_python_client/templates/property_templates/model_property.pyi

@@ -1,8 +1,7 @@
 {% macro construct(property, source, initial_value=None) %}
 {% if property.required and not property.nullable %}
 {% if source == "response.yaml" %}
-yaml = YAML(typ="base")
-yaml_dict = yaml.load(response.text.encode('utf-8'))
+yaml_dict = yaml.safe_load(response.text.encode("utf-8"))
 {{ property.python_name }} = {{ property.reference.class_name }}.from_dict(yaml_dict)
 {% else %}
 {{ property.python_name }} = {{ property.reference.class_name }}.from_dict({{ source }})

pyproject.toml

@@ -34,7 +34,6 @@ attrs = ">=20.1.0, <22.0"
 python-dateutil = "^2.8.1"
 httpx = ">=0.15.4,<=0.22.0"
 autoflake = "^1.4"
-"ruamel.yaml" = "^0.17.21"
 
 [tool.poetry.scripts]
 openapi-python-client = "openapi_python_client.cli:app"

tests/test_templates/endpoint_module.py

@@ -2,7 +2,7 @@
 
 import httpx
 from attr import asdict
-from ruamel.yaml import YAML
+import yaml
 
 from ...client import AuthenticatedClient, Client
 from ...types import Response