Fixed issue causing KeyError to be raised when building schema from m…

…ultiple `BaseModel` with the same names declared in separate classes (pydantic#1912)
jusexton · Dec 29, 2020 · d1b69f0 · d1b69f0
1 parent de0657e
commit d1b69f0
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 1 deletion.
diff --git a/changes/1912-JSextonn.md b/changes/1912-JSextonn.md
@@ -0,0 +1 @@
+Fixed issue causing KeyError to be raised when building schema from multiple `BaseModel` with the same names declared in separate classes.
diff --git a/pydantic/schema.py b/pydantic/schema.py
@@ -395,7 +395,7 @@ def get_flat_models_from_models(models: Sequence[Type['BaseModel']]) -> TypeMode
 
 
 def get_long_model_name(model: TypeModelOrEnum) -> str:
-    return f'{model.__module__}__{model.__name__}'.replace('.', '__')
+    return f'{model.__module__}__{model.__qualname__}'.replace('.', '__')
 
 
 def field_type_schema(

diff --git a/tests/test_schema.py b/tests/test_schema.py
@@ -2097,3 +2097,19 @@ class Model(BaseModel):
         'properties': {'a': {'title': 'A', 'type': 'string'}},
         'required': ['a'],
     }
+
+
+def test_multiple_nested_model_declarations_do_not_raise_error():
+    class ModelOne(BaseModel):
+        class NestedModel(BaseModel):
+            a: float
+
+        nested: NestedModel
+
+    class ModelTwo(BaseModel):
+        class NestedModel(BaseModel):
+            b: float
+
+        nested: NestedModel
+
+    schema([ModelOne, ModelTwo])
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Fixed issue causing KeyError to be raised when building schema from multiple `BaseModel` with the same names declared in separate classes.