aspyx 1.2.0__py3-none-any.whl → 1.4.0__py3-none-any.whl

aspyx/di/__init__.py

@@ -1,14 +1,15 @@
 """
 This module provides dependency injection and aop capabilities for Python applications.
 """
-from .di import DIException, AbstractCallableProcessor, LifecycleCallable, Lifecycle, Providers, Environment, ClassInstanceProvider, injectable, factory, environment, inject, order, create, on_init, on_running, on_destroy, inject_environment, Factory, PostProcessor
+from .di import conditional, requires_class, requires_feature, DIException, AbstractCallableProcessor, LifecycleCallable, Lifecycle, Providers, Environment, ClassInstanceProvider, injectable, factory, environment, inject, order, create, on_init, on_running, on_destroy, inject_environment, Factory, PostProcessor
 
-# import something from the subpackages, so that teh decorators are executed
+# import something from the subpackages, so that the decorators are executed
 
 from .configuration import ConfigurationManager
 from .aop import before
+from .threading import SynchronizeAdvice
 
-imports = [ConfigurationManager, before]
+imports = [ConfigurationManager, before, SynchronizeAdvice]
 
 __all__ = [
     "ClassInstanceProvider",
@@ -30,5 +31,8 @@ __all__ = [
     "AbstractCallableProcessor",
     "LifecycleCallable",
     "DIException",
-    "Lifecycle"
+    "Lifecycle",
+    "conditional",
+    "requires_class",
+    "requires_feature"
 ]

aspyx/di/aop/aop.py

@@ -12,10 +12,8 @@ from dataclasses import dataclass
 from enum import auto, Enum
 from typing import Optional, Dict, Type, Callable
 
-from aspyx.di.di import order
 from aspyx.reflection import Decorators, TypeDescriptor
-from aspyx.di import injectable, Providers, ClassInstanceProvider, Environment, PostProcessor
-
+from aspyx.di import injectable, order, Providers, ClassInstanceProvider, Environment, PostProcessor
 
 class AOPException(Exception):
     """
@@ -50,7 +48,7 @@ class AspectTarget(ABC):
     __slots__ = [
         "_function",
         "_type",
-
+        "_async",
         "_clazz",
         "_instance",
         "names",
@@ -65,6 +63,7 @@ class AspectTarget(ABC):
     def __init__(self):
         self._clazz = None
         self._instance = None
+        self._async = False
         self._function = None
         self._type = None
 
@@ -108,11 +107,28 @@ class AspectTarget(ABC):
 
         return self
 
+    def that_are_async(self):
+        """
+        matches methods that are async
+        :return: self
+        """
+        self._async = True
+        return self
+
     def of_type(self, type: Type):
+        """
+        matches methods belonging to a class or classes that are subclasses of the specified type
+        :return: self
+        """
         self.types.append(type)
         return self
 
     def decorated_with(self, decorator):
+        """
+        matches methods or classes that are decorated with the specified decorator
+        :param decorator:  the decorator callable
+        :return:
+        """
         self.decorators.append(decorator)
         return self
 
@@ -128,6 +144,9 @@ class AspectTarget(ABC):
         return self
 
 class ClassAspectTarget(AspectTarget):
+    """
+    An AspectTarget matching classes
+    """
     # properties
 
     __slots__ = [
@@ -167,6 +186,10 @@ class ClassAspectTarget(AspectTarget):
     # fluent
 
 class MethodAspectTarget(AspectTarget):
+    """
+       An AspectTarget matching methods
+       """
+
     # properties
 
     __slots__ = [ ]
@@ -178,6 +201,11 @@ class MethodAspectTarget(AspectTarget):
 
         method_descriptor = descriptor.get_method(func.__name__)
 
+        # async
+
+        if self._async is not method_descriptor.is_async():
+            return False
+
         # type
 
         if len(self.types) > 0:
@@ -219,14 +247,14 @@ def classes():
     return ClassAspectTarget()
 
 
-class JoinPoint:
+class Aspect:
     __slots__ = [
         "next",
     ]
 
     # constructor
 
-    def __init__(self, next: 'JoinPoint'):
+    def __init__(self, next: 'Aspect'):
         self.next = next
 
     # public
@@ -234,45 +262,61 @@ class JoinPoint:
     def call(self, invocation: 'Invocation'):
         pass
 
-class FunctionJoinPoint(JoinPoint):
+    async def call_async(self, invocation: 'Invocation'):
+        pass
+
+class FunctionAspect(Aspect):
     __slots__ = [
         "instance",
         "func",
+        "order",
     ]
 
-    def __init__(self, instance, func, next: Optional['JoinPoint']):
-        super().__init__(next)
+    def __init__(self, instance, func, next_aspect: Optional['Aspect']):
+        super().__init__(next_aspect)
 
         self.instance = instance
         self.func = func
 
+        self.order = next((decorator.args[0] for decorator in Decorators.get(func) if decorator.decorator is order), 0)
+
     def call(self, invocation: 'Invocation'):
-        invocation.current_join_point = self
+        invocation.current_aspect = self
 
         return self.func(self.instance, invocation)
 
-class MethodJoinPoint(FunctionJoinPoint):
+    async def call_async(self, invocation: 'Invocation'):
+        invocation.current_aspect = self
+
+        return await self.func(self.instance, invocation)
+
+class MethodAspect(FunctionAspect):
     __slots__ = []
 
     def __init__(self, instance, func):
         super().__init__(instance, func, None)
 
     def call(self, invocation: 'Invocation'):
-        invocation.current_join_point = self
+        invocation.current_aspect = self
 
         return self.func(*invocation.args, **invocation.kwargs)
 
+    async def call_async(self, invocation: 'Invocation'):
+        invocation.current_aspect = self
+
+        return await self.func(*invocation.args, **invocation.kwargs)
+
 @dataclass
-class JoinPoints:
-    before: list[JoinPoint]
-    around: list[JoinPoint]
-    error: list[JoinPoint]
-    after: list[JoinPoint]
+class Aspects:
+    before: list[Aspect]
+    around: list[Aspect]
+    error: list[Aspect]
+    after: list[Aspect]
 
 class Invocation:
     """
     Invocation stores the relevant data of a single method invocation.
-    It holds the arguments, keyword arguments, result, error, and the join points that define the aspect behavior.
+    It holds the arguments, keyword arguments, result, error, and the aspects that define the aspect behavior.
     """
     # properties
 
@@ -282,20 +326,20 @@ class Invocation:
         "kwargs",
         "result",
         "exception",
-        "join_points",
-        "current_join_point",
+        "aspects",
+        "current_aspect",
     ]
 
     # constructor
 
-    def __init__(self, func, join_points: JoinPoints):
+    def __init__(self, func, aspects: Aspects):
         self.func = func
         self.args : list[object] = []
         self.kwargs = None
         self.result = None
         self.exception = None
-        self.join_points = join_points
-        self.current_join_point = None
+        self.aspects = aspects
+        self.current_aspect = None
 
     def call(self, *args, **kwargs):
         # remember args
@@ -305,23 +349,54 @@ class Invocation:
 
         # run all before
 
-        for join_point in self.join_points.before:
-            join_point.call(self)
+        for aspect in self.aspects.before:
+            aspect.call(self)
 
         # run around's with the method being the last aspect!
 
         try:
-            self.result = self.join_points.around[0].call(self) # will follow the proceed chain
+            self.result = self.aspects.around[0].call(self) # will follow the proceed chain
 
         except Exception as e:
             self.exception = e
-            for join_point in self.join_points.error:
-                join_point.call(self)
+            for aspect in self.aspects.error:
+                aspect.call(self)
 
         # run all before
 
-        for join_point in self.join_points.after:
-            join_point.call(self)
+        for aspect in self.aspects.after:
+            aspect.call(self)
+
+        if self.exception is not None:
+            raise self.exception # rethrow the error
+
+        return self.result
+
+    async def call_async(self, *args, **kwargs):
+        # remember args
+
+        self.args = args
+        self.kwargs = kwargs
+
+        # run all before
+
+        for aspect in self.aspects.before:
+            aspect.call(self)
+
+        # run around's with the method being the last aspect!
+
+        try:
+            self.result = await self.aspects.around[0].call_async(self) # will follow the proceed chain
+
+        except Exception as e:
+            self.exception = e
+            for aspect in self.aspects.error:
+                aspect.call(self)
+
+        # run all before
+
+        for aspect in self.aspects.after:
+            aspect.call(self)
 
         if self.exception is not None:
             raise self.exception # rethrow the error
@@ -338,29 +413,41 @@ class Invocation:
 
         # next one please...
 
-        return self.current_join_point.next.call(self)
+        return self.current_aspect.next.call(self)
+
+    async def proceed_async(self, *args, **kwargs):
+        """
+        Proceed to the next join point in the around chain up to the original method.
+        """
+        if len(args) > 0 or len(kwargs) > 0:  # as soon as we have args, we replace the current ones
+            self.args = args
+            self.kwargs = kwargs
+
+        # next one please...
+
+        return await self.current_aspect.next.call_async(self)
 
-@injectable()
-class Advice:
+class Advices:
     # static data
 
     targets: list[AspectTarget] = []
 
-    __slots__ = [
-        "cache",
-        "lock"
-    ]
+    __slots__ = []
 
     # constructor
 
     def __init__(self):
-        self.cache : Dict[Type, Dict[Callable,JoinPoints]] = {}
-        self.lock = threading.RLock()
+        pass
 
     # methods
 
-    def collect(self, clazz, member, type: AspectType, environment: Environment):
-        aspects = [FunctionJoinPoint(environment.get(target._clazz), target._function, None) for target in Advice.targets if target._type == type and target._matches(clazz, member)]
+    @classmethod
+    def collect(cls, clazz, member, type: AspectType, environment: Environment):
+        aspects = [FunctionAspect(environment.get(target._clazz), target._function, None) for target in Advices.targets if target._type == type and target._clazz is not clazz and environment.providers.get(target._clazz) is not None and target._matches(clazz, member)]
+
+        # sort according to order
+
+        aspects = sorted(aspects, key=lambda aspect: aspect.order)
 
         # link
 
@@ -371,30 +458,23 @@ class Advice:
 
         return aspects
 
-    def join_points4(self, instance, environment: Environment) -> Dict[Callable,JoinPoints]:
+    @classmethod
+    def aspects_for(cls, instance, environment: Environment) -> Dict[Callable,Aspects]:
         clazz = type(instance)
 
-        result = self.cache.get(clazz, None)
-        if result is None:
-            with self.lock:
-                result = self.cache.get(clazz, None)
-
-                if result is None:
-                    result = {}
-
-                    for _, member in inspect.getmembers(clazz, predicate=inspect.isfunction):
-                        join_points = self.compute_join_points(clazz, member, environment)
-                        if join_points is not None:
-                            result[member] = join_points
+        result = {}
 
-                    self.cache[clazz] = result
+        for _, member in inspect.getmembers(clazz, predicate=inspect.isfunction):
+            aspects = cls.compute_aspects(clazz, member, environment)
+            if aspects is not None:
+                result[member] = aspects
 
         # add around methods
 
         value = {}
 
         for key, cjp in result.items():
-            jp = JoinPoints(
+            jp = Aspects(
                 before=cjp.before,
                 around=cjp.around,
                 error=cjp.error,
@@ -402,7 +482,7 @@ class Advice:
 
             # add method to around
 
-            jp.around.append(MethodJoinPoint(instance, key))
+            jp.around.append(MethodAspect(instance, key))
             if len(jp.around) > 1:
                 jp.around[len(jp.around) - 2].next = jp.around[len(jp.around) - 1]
 
@@ -412,14 +492,15 @@ class Advice:
 
         return value
 
-    def compute_join_points(self, clazz, member, environment: Environment) -> Optional[JoinPoints]:
-        befores = self.collect(clazz, member, AspectType.BEFORE, environment)
-        arounds = self.collect(clazz, member, AspectType.AROUND, environment)
-        afters = self.collect(clazz, member, AspectType.AFTER, environment)
-        errors = self.collect(clazz, member, AspectType.ERROR, environment)
+    @classmethod
+    def compute_aspects(cls, clazz, member, environment: Environment) -> Optional[Aspects]:
+        befores = cls.collect(clazz, member, AspectType.BEFORE, environment)
+        arounds = cls.collect(clazz, member, AspectType.AROUND, environment)
+        afters = cls.collect(clazz, member, AspectType.AFTER, environment)
+        errors = cls.collect(clazz, member, AspectType.ERROR, environment)
 
         if len(befores) > 0 or len(arounds) > 0 or len(afters) > 0  or len(errors) > 0:
-            return JoinPoints(
+            return Aspects(
                 before=befores,
                 around=arounds,
                 error=errors,
@@ -437,7 +518,7 @@ def sanity_check(clazz: Type, name: str):
 
 def advice(cls):
     """
-    Classes decorated with @advice are treated as aspect classes.
+    Classes decorated with @advice are treated as advice classes.
     They can contain methods decorated with @before, @after, @around, or @error to define aspects.
     """
     Providers.register(ClassInstanceProvider(cls, True))
@@ -447,10 +528,10 @@ def advice(cls):
     for name, member in TypeDescriptor.for_type(cls).methods.items():
         decorator = next((decorator for decorator in member.decorators if decorator.decorator in [before, after, around, error]), None)
         if decorator is not None:
-            target = decorator.args[0]
+            target = decorator.args[0] # ? ...?? TODO, can be multiple
             target._clazz = cls
             sanity_check(cls, name)
-            Advice.targets.append(target)
+            Advices.targets.append(target) #??
 
     return cls
 
@@ -512,31 +593,57 @@ def around(*targets: AspectTarget):
 
     return decorator
 
-@injectable()
+@injectable(scope="environment")
 @order(0)
 class AdviceProcessor(PostProcessor):
     # properties
 
     __slots__ = [
-        "advice",
+        "lock",
+        "cache"
     ]
 
     # constructor
 
-    def __init__(self, advice: Advice):
+    def __init__(self):
         super().__init__()
 
-        self.advice = advice
+        self.cache : Dict[Type, Dict[Callable,Aspects]] = {}
+        self.lock = threading.RLock()
+
+    # local
+
+    def aspects_for(self, instance, environment: Environment) -> Dict[Callable,Aspects]:
+        clazz = type(instance)
+        result = self.cache.get(clazz, None)
+        if result is None:
+            with self.lock:
+                result = Advices.aspects_for(instance, environment)
+                self.cache[clazz] = result # TOID der cache ist zu dick?????
+
+        return result
 
     # implement
 
     def process(self, instance: object, environment: Environment):
-        join_point_dict = self.advice.join_points4(instance, environment)
+        aspect_dict = self.aspects_for(instance, environment)
 
-        for member, joinPoints in join_point_dict.items():
+        for member, aspects in aspect_dict.items():
             Environment.logger.debug("add aspects for %s:%s", type(instance), member.__name__)
 
             def wrap(jp):
-                return lambda *args, **kwargs: Invocation(member, jp).call(*args, **kwargs)
+                def sync_wrapper(*args, **kwargs):
+                    return Invocation(member, jp).call(*args, **kwargs)
+
+                return sync_wrapper
+
+            def wrap_async(jp):
+                async def async_wrapper(*args, **kwargs):
+                    return await Invocation(member, jp).call_async(*args, **kwargs)
+
+                return async_wrapper
 
-            setattr(instance, member.__name__, types.MethodType(wrap(joinPoints), instance))
+            if inspect.iscoroutinefunction(member):
+                setattr(instance, member.__name__, types.MethodType(wrap_async(aspects), instance))
+            else:
+                setattr(instance, member.__name__, types.MethodType(wrap(aspects), instance))

aspyx/di/configuration/env_configuration_source.py

@@ -52,4 +52,4 @@ class EnvConfigurationSource(ConfigurationSource):
             else:
                 exploded[key] = value
 
-        return exploded
+        return exploded

aspyx/di/configuration/yaml_configuration_source.py

@@ -23,4 +23,4 @@ class YamlConfigurationSource(ConfigurationSource):
 
     def load(self) -> dict:
         with open(self.file, "r") as file:
-            return yaml.safe_load(file)
+            return yaml.safe_load(file)