Replaced dependency graph with external library. (#57)

Signed-off-by: Pavel Kirilin <[email protected]>

Author: s3rius

poetry.lock

@@ -54,6 +54,7 @@ mock = "^4.0.3"
 anyio = "^3.6.1"
 pytest-xdist = { version = "^2.5.0", extras = ["psutil"] }
 types-mock = "^4.0.15"
+taskiq_dependencies =  "~1.0.0"
 
 [tool.poetry.extras]
 zmq = ["pyzmq"]

pyproject.toml

@@ -1,4 +1,6 @@
 """Distributed task manager."""
+from taskiq_dependencies import Depends as TaskiqDepends
+
 from taskiq.abc.broker import AsyncBroker, AsyncTaskiqDecoratedTask
 from taskiq.abc.formatter import TaskiqFormatter
 from taskiq.abc.middleware import TaskiqMiddleware
@@ -8,7 +10,6 @@
 from taskiq.brokers.shared_broker import async_shared_broker
 from taskiq.brokers.zmq_broker import ZeroMQBroker
 from taskiq.context import Context
-from taskiq.dependencies import TaskiqDepends
 from taskiq.events import TaskiqEvents
 from taskiq.exceptions import TaskiqError
 from taskiq.funcs import gather

taskiq/__init__.py

@@ -8,17 +8,17 @@
 class ScheduleSource(ABC):
     """Abstract class for source of scheduled tasks."""
 
-    async def startup(self) -> None:
+    async def startup(self) -> None:  # noqa: B027
         """Action to execute during startup."""
 
-    async def shutdown(self) -> None:
+    async def shutdown(self) -> None:  # noqa: B027
         """Actions to execute during shutdown."""
 
     @abstractmethod
     async def get_schedules(self) -> List["ScheduledTask"]:
         """Get list of taskiq schedules."""
 
-    async def add_schedule(self, schedule: "ScheduledTask") -> None:
+    async def add_schedule(self, schedule: "ScheduledTask") -> None:  # noqa: B027
         """
         Add a new schedule.
 

taskiq/abc/schedule_source.py

@@ -2,11 +2,12 @@
 from collections import OrderedDict
 from typing import Any, Callable, Coroutine, Optional, TypeVar, get_type_hints
 
+from taskiq_dependencies import DependencyGraph
+
 from taskiq.abc.broker import AsyncBroker
 from taskiq.abc.result_backend import AsyncResultBackend, TaskiqResult
 from taskiq.cli.worker.args import WorkerArgs
 from taskiq.cli.worker.receiver import Receiver
-from taskiq.dependencies import DependencyGraph
 from taskiq.events import TaskiqEvents
 from taskiq.exceptions import TaskiqError
 from taskiq.message import BrokerMessage

taskiq/brokers/inmemory_broker.py

@@ -6,13 +6,14 @@
 from time import time
 from typing import Any, Callable, Dict, get_type_hints
 
+from taskiq_dependencies import DependencyGraph
+
 from taskiq.abc.broker import AsyncBroker
 from taskiq.abc.middleware import TaskiqMiddleware
 from taskiq.cli.worker.args import WorkerArgs
 from taskiq.cli.worker.log_collector import log_collector
 from taskiq.cli.worker.params_parser import parse_params
 from taskiq.context import Context
-from taskiq.dependencies import DependencyGraph
 from taskiq.message import BrokerMessage, TaskiqMessage
 from taskiq.result import TaskiqResult
 from taskiq.state import TaskiqState
@@ -164,7 +165,7 @@ async def run_task(  # noqa: C901, WPS210
             dep_ctx = None
             if dependency_graph:
                 # Create a context for dependency resolving.
-                dep_ctx = dependency_graph.ctx(
+                dep_ctx = dependency_graph.async_ctx(
                     {
                         Context: Context(message, self.broker),
                         TaskiqState: self.broker.state,