added multi parents inheritance

KiraPC · Feb 22, 2022 · 7a7d99e · 7a7d99e
1 parent e105701
commit 7a7d99e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/fastapi_router_controller/lib/controller.py b/fastapi_router_controller/lib/controller.py
@@ -89,8 +89,9 @@ def add_resource(self, cls):
             raise MultipleResourceException()
 
         # check if cls (Resource) was exteded from another
-        if hasattr(cls, Controller.RC_KEY):
-            self.__get_parent_routes(cls.__router__)
+        for base in cls.__bases__:
+            if hasattr(base, Controller.RC_KEY):
+                self.__get_parent_routes(base.__router__)
 
         setattr(cls, Controller.RC_KEY, self.router)
         setattr(self, Controller.RESOURCE_CLASS_KEY, cls)