registry: unregister the manager and class

fastapi · Jan 17, 2025 · 80f2935 · 80f2935
1 parent 1c5263e
commit 80f2935
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/sqlmodel/main.py b/sqlmodel/main.py
@@ -591,6 +591,14 @@ def __init__(
         return cls.__do_init__(bases, dict_, kw)
 
     def sqlmodel_rebuild(cls) -> None:
+        reg = cls._sa_registry
+        # clear any exisiting mappers for the cls
+        manager = [m for m in reg._managers if m.class_ == cls]
+        if len(manager) > 0:
+            for m in manager:
+                reg._dispose_manager_and_mapper(m)
+            del reg._managers[m]
+
         return cls.__do_init__(cls.__bases__, cls.__dict__, {})
 
     # Override SQLAlchemy, allow both SQLAlchemy and plain Pydantic models