pydocket 0.0.1__py3-none-any.whl → 0.1.0__py3-none-any.whl

docket/__init__.py

@@ -8,7 +8,16 @@ from importlib.metadata import version
 
 __version__ = version("pydocket")
 
-from .dependencies import CurrentDocket, CurrentWorker, Retry
+from .annotations import Logged
+from .dependencies import (
+    CurrentDocket,
+    CurrentExecution,
+    CurrentWorker,
+    ExponentialRetry,
+    Retry,
+    TaskKey,
+    TaskLogger,
+)
 from .docket import Docket
 from .execution import Execution
 from .worker import Worker
@@ -19,6 +28,11 @@ __all__ = [
     "Execution",
     "CurrentDocket",
     "CurrentWorker",
+    "CurrentExecution",
+    "TaskKey",
+    "TaskLogger",
     "Retry",
+    "ExponentialRetry",
+    "Logged",
     "__version__",
 ]

docket/annotations.py

@@ -0,0 +1,30 @@
+import abc
+import inspect
+from typing import Any, Iterable, Mapping, Self
+
+
+class Annotation(abc.ABC):
+    @classmethod
+    def annotated_parameters(cls, signature: inspect.Signature) -> Mapping[str, Self]:
+        annotated: dict[str, Self] = {}
+
+        for param_name, param in signature.parameters.items():
+            if param.annotation == inspect.Parameter.empty:
+                continue
+
+            try:
+                metadata: Iterable[Any] = param.annotation.__metadata__
+            except AttributeError:
+                continue
+
+            for arg_type in metadata:
+                if isinstance(arg_type, cls):
+                    annotated[param_name] = arg_type
+                elif isinstance(arg_type, type) and issubclass(arg_type, cls):
+                    annotated[param_name] = arg_type()
+
+        return annotated
+
+
+class Logged(Annotation):
+    """Instructs docket to include arguments to this parameter in the log."""