avoid conflict between uppercase/lowercase enum values

Author: eli-bl

openapi_python_client/parser/properties/enum_property.py

@@ -187,6 +187,23 @@ def values_from_list(values: list[str] | list[int]) -> dict[str, ValueType]:
         """Convert a list of values into dict of {name: value}, where value can sometimes be None"""
         output: dict[str, ValueType] = {}
 
+        # We normally would like to make nice-looking Python constant names for enum values, so that "myValue"
+        # becomes MY_VALUE, etc. However, that won't work if an enum has two values that differ only by case
+        # (which is allowed in OpenAPI).
+        use_case_sensitive_names = False
+        for i, value1 in enumerate(values):
+            if use_case_sensitive_names:
+                break
+            for j, value2 in enumerate(values):
+                if (
+                    i != j
+                    and isinstance(value1, str)
+                    and isinstance(value2, str)
+                    and value1.upper() == value2.upper()
+                ):
+                    use_case_sensitive_names = True
+                    break
+
         for i, value in enumerate(values):
             value = cast(Union[str, int], value)
             if isinstance(value, int):
@@ -196,11 +213,14 @@ def values_from_list(values: list[str] | list[int]) -> dict[str, ValueType]:
                     output[f"VALUE_{value}"] = value
                 continue
             if value and value[0].isalpha():
-                key = value.upper()
+                key = value
             else:
                 key = f"VALUE_{i}"
             if key in output:
                 raise ValueError(f"Duplicate key {key} in Enum")
-            sanitized_key = utils.snake_case(key).upper()
+            if use_case_sensitive_names:
+                sanitized_key = utils.sanitize(key.replace(" ", "_"))
+            else:
+                sanitized_key = utils.snake_case(key.upper()).upper()
             output[sanitized_key] = utils.remove_string_escapes(value)
         return output

tests/test_parser/test_properties/test_init.py

@@ -372,6 +372,19 @@ def test_values_from_list_duplicate(self):
         with pytest.raises(ValueError):
             EnumProperty.values_from_list(data)
 
+    def test_values_from_list_with_case_sensitive_names(self):
+        from openapi_python_client.parser.properties import EnumProperty
+
+        data = ["abc", "123", "ABC", "thing with spaces"]
+
+        result = EnumProperty.values_from_list(data)
+
+        assert result == {
+            "abc": "abc",
+            "VALUE_1": "123",
+            "ABC": "ABC",
+            "thing_with_spaces": "thing with spaces",
+        }
 
 class TestPropertyFromData:
     def test_property_from_data_str_enum(self, enum_property_factory, config):