Fix enum default values to use short names in nested package paths (#2644)

Author: koxudaxi

src/datamodel_code_generator/model/enum.py

@@ -156,4 +156,4 @@ def __init__(self, enum: Enum, field: DataModelFieldBase) -> None:
 
     def __repr__(self) -> str:
         """Return string representation of enum member."""
-        return f"{self.alias or self.enum.name}.{self.field.name}"
+        return f"{self.alias or self.enum.class_name}.{self.field.name}"

tests/data/expected/main/openapi/nested_package_enum_default/io/example/api/__init__.py

@@ -0,0 +1,3 @@
+# generated by datamodel-codegen:
+#   filename:  nested_package_enum_default.json
+#   timestamp: 1985-10-26T08:21:00+00:00

tests/data/expected/main/openapi/nested_package_enum_default/io/example/api/v1.py

@@ -0,0 +1,26 @@
+# generated by datamodel-codegen:
+#   filename:  nested_package_enum_default.json
+#   timestamp: 1985-10-26T08:21:00+00:00
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from enum import Enum
+from typing import Optional
+
+
+class Resolution(Enum):
+    Required = 'Required'
+    Optional = 'Optional'
+
+
+class Policy(Enum):
+    Allow = 'Allow'
+    Deny = 'Deny'
+
+
+@dataclass
+class BucketSpec:
+    resolution: Optional[Resolution] = Resolution.Required
+    policy: Optional[Policy] = Policy.Allow
+    name: Optional[str] = None

tests/data/openapi/nested_package_enum_default.json

@@ -0,0 +1,42 @@
+{
+  "openapi": "3.0.0",
+  "info": {
+    "title": "Test API",
+    "version": "1.0.0"
+  },
+  "paths": {},
+  "components": {
+    "schemas": {
+      "io.example.api.v1.Resolution": {
+        "type": "string",
+        "enum": ["Required", "Optional"],
+        "default": "Required"
+      },
+      "io.example.api.v1.Policy": {
+        "type": "string",
+        "enum": ["Allow", "Deny"],
+        "default": "Allow"
+      },
+      "io.example.api.v1.BucketSpec": {
+        "type": "object",
+        "properties": {
+          "resolution": {
+            "allOf": [
+              {"$ref": "#/components/schemas/io.example.api.v1.Resolution"}
+            ],
+            "default": "Required"
+          },
+          "policy": {
+            "allOf": [
+              {"$ref": "#/components/schemas/io.example.api.v1.Policy"}
+            ],
+            "default": "Allow"
+          },
+          "name": {
+            "type": "string"
+          }
+        }
+      }
+    }
+  }
+}

tests/main/openapi/test_main_openapi.py

@@ -3318,3 +3318,18 @@ def test_main_allof_enum_ref(output_file: Path) -> None:
         input_file_type=None,
         assert_func=assert_file_content,
     )
+
+
+def test_main_nested_package_enum_default(output_dir: Path) -> None:
+    """Test enum default values use short names in same module with nested package paths."""
+    with freeze_time(TIMESTAMP):
+        run_main_and_assert(
+            input_path=OPEN_API_DATA_PATH / "nested_package_enum_default.json",
+            output_path=output_dir,
+            expected_directory=EXPECTED_OPENAPI_PATH / "nested_package_enum_default",
+            extra_args=[
+                "--output-model-type",
+                "dataclasses.dataclass",
+                "--set-default-enum-member",
+            ],
+        )