Fix alias for superclass with identical name (#1981)

If an externally imported model is renamed because it has the same name as the current model, the alias was incorrectly suffixed with the original name.
koxudaxi · Jun 5, 2024 · ee11cef · ee11cef
1 parent d875f85
commit ee11cef
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/datamodel_code_generator/parser/base.py b/datamodel_code_generator/parser/base.py
@@ -707,7 +707,7 @@ def __change_from_import(
                 if from_ and import_ and alias != name:
                     data_type.alias = (
                         alias
-                        if from_ == '.' and data_type.full_name == import_
+                        if data_type.reference.short_name == import_
                         else f'{alias}.{name}'
                     )
 

diff --git a/tests/data/expected/main/main_require_referenced_field/required.py b/tests/data/expected/main/main_require_referenced_field/required.py
@@ -9,6 +9,6 @@
 from .referenced import Model as Model_1
 
 
-class Model(Model_1.Model):
+class Model(Model_1):
     some_optional_property: str
     some_optional_typed_property: datetime
diff --git a/tests/data/expected/main/main_require_referenced_field_pydantic_v2/required.py b/tests/data/expected/main/main_require_referenced_field_pydantic_v2/required.py
@@ -9,6 +9,6 @@
 from .referenced import Model as Model_1
 
 
-class Model(Model_1.Model):
+class Model(Model_1):
     some_optional_property: str
     some_optional_typed_property: AwareDatetime