fixed mypy issues

Author: eadwinCode

ellar/core/modules/ref/base.py

@@ -1,6 +1,5 @@
 import typing as t
 from abc import ABC, abstractmethod
-from functools import cached_property
 from typing import Type
 
 from ellar.auth.constants import POLICY_KEYS
@@ -66,16 +65,21 @@ def __init__(
         self._routers: t.List[t.Union[BaseRoute]] = []
         self._exports: t.List[t.Type] = []
         self._providers: t.Dict[t.Type, ProviderConfig] = {}
+        self._module_execution_context: t.Optional[ModuleExecutionContext] = None
 
     def __repr__(self) -> str:
         return f"<{self.__class__.__name__} name={self.name} module={self.module}>"
 
     def __hash__(self) -> int:
         return hash(self.module)
 
-    @cached_property
+    @property
     def module_context(self) -> ModuleExecutionContext:
-        return ModuleExecutionContext(container=self.container, module=self.module)
+        if self._module_execution_context is None:
+            self._module_execution_context = ModuleExecutionContext(
+                container=self.container, module=self.module
+            )
+        return self._module_execution_context
 
     @property
     def tree_manager(self) -> ModuleTreeManager:

ellar/testing/dependency_analyzer.py

@@ -68,7 +68,7 @@ def _build_module_tree(self) -> "ModuleTreeManager":
         module_setup = ModuleSetup(self.application_module)
         return AppFactory.read_all_module(module_setup)
 
-    def get_type_from_tag(self, tag: t.NewType) -> t.Optional[t.Type[t.Any]]:
+    def get_type_from_tag(self, tag: t.Any) -> t.Optional[t.Type[t.Any]]:
         """
         Get the type/interface associated with a given tag from ApplicationModule tree
 
@@ -86,7 +86,7 @@ def get_type_from_tag(self, tag: t.NewType) -> t.Optional[t.Type[t.Any]]:
         if not hasattr(tag, "__name__"):
             return None
 
-        tag_instance = tag.__name__
+        tag_instance = getattr(tag, "__name__", None)
         if not isinstance(tag_instance, Tag):
             return None
 

ellar/testing/module.py

@@ -271,7 +271,10 @@ def _analyze_and_resolve_controller_dependencies(
             for dep_type in dependencies:
                 # Handle tagged dependencies (InjectByTag)
                 if isinstance(dep_type, t.NewType):
-                    dep_type = app_analyzer.get_type_from_tag(dep_type)
+                    resolved_type = app_analyzer.get_type_from_tag(dep_type)
+                    if not resolved_type:
+                        continue
+                    dep_type = resolved_type
 
                 providing_module = app_analyzer.find_module_providing_service(dep_type)