Allow Unset values to avoid being send during serialization

Author: bowenwr

openapi_python_client/parser/properties.py

@@ -64,23 +64,38 @@ def get_imports(self, *, prefix: str) -> Set[str]:
             back to the root of the generated client.
         """
         if self.nullable or not self.required:
-            return {"from typing import Optional"}
+            return {"from typing import Optional",
+                    "from typing import cast",
+                    f"from {prefix}types import UNSET"}
         return set()
 
     def to_string(self) -> str:
         """ How this should be declared in a dataclass """
         if self.default:
             default = self.default
         elif not self.required:
-            default = "None"
+            default = "cast(None, UNSET)"
         else:
             default = None
 
         if default is not None:
-            return f"{self.python_name}: {self.get_type_string()} = {self.default}"
+            return f"{self.python_name}: {self.get_type_string()} = {default}"
         else:
             return f"{self.python_name}: {self.get_type_string()}"
 
+    def to_query_method_arg(self) -> str:
+        """ How this should be declared in a query string """
+        if self.default:
+            default = self.default
+        elif not self.required:
+            default = "None"
+        else:
+            default = None
+
+        if default is not None:
+            return f"{self.python_name}: {self.get_type_string()} = {default}"
+        else:
+            return f"{self.python_name}: {self.get_type_string()}"
 
 @dataclass
 class StringProperty(Property):

openapi_python_client/templates/endpoint_macros.pyi

@@ -80,7 +80,7 @@ client: Client,
 {% endif %}
 {# path parameters #}
 {% for parameter in endpoint.path_parameters %}
-{{ parameter.to_string() }},
+{{parameter.to_query_method_arg()}},
 {% endfor %}
 {# Form data if any #}
 {% if endpoint.form_body_reference %}
@@ -96,10 +96,10 @@ json_body: {{ endpoint.json_body.get_type_string() }},
 {% endif %}
 {# query parameters #}
 {% for parameter in endpoint.query_parameters %}
-{{ parameter.to_string() }},
+{{parameter.to_query_method_arg()}},
 {% endfor %}
 {% for parameter in endpoint.header_parameters %}
-{{ parameter.to_string() }},
+{{ parameter.to_query_method_arg() }},
 {% endfor %}
 {% endmacro %}
 

openapi_python_client/templates/model.pyi

@@ -24,11 +24,17 @@ class {{ model.reference.class_name }}:
         {% endif %}
         {% endfor %}
 
-        return {
-            {% for property in model.required_properties + model.optional_properties %}
-            "{{ property.name }}": {{ property.python_name }},
-            {% endfor %}
-        }
+        properties: Dict[str, Any] = dict()
+
+        {% for property in model.required_properties + model.optional_properties %}
+        {% if not property.required %}
+        if {{property.python_name}} is not UNSET:
+            properties["{{ property.name }}"] = {{ property.python_name }}
+        {% else %}
+        properties["{{ property.name }}"] = {{ property.python_name }}
+        {% endif %}
+        {% endfor %}
+        return properties
 
     @staticmethod
     def from_dict(d: Dict[str, Any]) -> "{{ model.reference.class_name }}":

openapi_python_client/templates/property_templates/date_property.pyi

@@ -12,6 +12,9 @@ if {{ source }} is not None:
 {% if property.required %}
 {{ destination }} = {{ source }}.isoformat()
 {% else %}
-{{ destination }} = {{ source }}.isoformat() if {{ source }} else None
+if {{ source }} is UNSET:
+    {{ destination }} = UNSET
+else:
+    {{ destination }} = {{ source }}.isoformat() if {{ source }} else None
 {% endif %}
 {% endmacro %}

openapi_python_client/templates/property_templates/datetime_property.pyi

@@ -12,6 +12,9 @@ if {{ source }} is not None:
 {% if property.required %}
 {{ destination }} = {{ source }}.isoformat()
 {% else %}
-{{ destination }} = {{ source }}.isoformat() if {{ source }} else None
+if {{ source }} is UNSET:
+    {{ destination }} = UNSET
+else:
+    {{ destination }} = {{ source }}.isoformat() if {{ source }} else None
 {% endif %}
 {% endmacro %}

openapi_python_client/templates/property_templates/dict_property.pyi

@@ -12,6 +12,9 @@ if {{ source }} is not None:
 {% if property.required %}
 {{ destination }} = {{ source }}
 {% else %}
-{{ destination }} = {{ source }} if {{ source }} else None
+if {{ source }} is UNSET:
+    {{ destination }} = UNSET
+else:
+    {{ destination }} = {{ source }} if {{ source }} else None
 {% endif %}
 {% endmacro %}

openapi_python_client/templates/property_templates/enum_property.pyi

@@ -1,3 +1,5 @@
+from openapi_python_client.templates.types import UNSET
+
 {% macro construct(property, source) %}
 {% if property.required %}
 {{ property.python_name }} = {{ property.reference.class_name }}({{ source }})
@@ -12,6 +14,9 @@ if {{ source }} is not None:
 {% if property.required %}
 {{ destination }} = {{ source }}.value
 {% else %}
-{{ destination }} = {{ source }}.value if {{ source }} else None
+if {{ source }} is UNSET:
+    {{ destination }} = UNSET
+else:
+    {{ destination }} = {{ source }}.value if {{ source }} else None
 {% endif %}
 {% endmacro %}

openapi_python_client/templates/property_templates/file_property.pyi

@@ -7,6 +7,9 @@
 {% if property.required %}
 {{ destination }} = {{ source }}.to_tuple()
 {% else %}
-{{ destination }} = {{ source }}.to_tuple() if {{ source }} else None
+if {{ source }} is UNSET:
+    {{ destination }} = UNSET
+else:
+    {{ destination }} = {{ source }}.to_tuple() if {{ source }} else None
 {% endif %}
 {% endmacro %}

openapi_python_client/templates/property_templates/list_property.pyi

@@ -1,3 +1,5 @@
+from openapi_python_client.templates.types import UNSET
+
 {% macro construct(property, source) %}
 {% set inner_property = property.inner_property %}
 {% if inner_property.template %}
@@ -36,6 +38,8 @@ for {{ inner_source }} in {{ source }}:
 {% if not property.required %}
 if {{ source }} is None:
     {{ destination }} = None
+elif {{ source }} is UNSET:
+    {{ destination }} = UNSET
 else:
     {{ _transform(property, source, destination) | indent(4) }}
 {% else %}

openapi_python_client/templates/property_templates/ref_property.pyi

@@ -12,6 +12,9 @@ if {{ source }} is not None:
 {% if property.required %}
 {{ destination }} = {{ source }}.to_dict()
 {% else %}
-{{ destination }} = {{ source }}.to_dict() if {{ source }} else None
+if {{ source }} is UNSET:
+    {{ destination }} = UNSET
+else:
+    {{ destination }} = {{ source }}.to_dict() if {{ source }} else None
 {% endif %}
 {% endmacro %}