Added scheduler flag. (#143)

Closes #142.

Author: s3rius

taskiq/abc/broker.py

@@ -68,7 +68,10 @@ class AsyncBroker(ABC):
     """
 
     available_tasks: Dict[str, AsyncTaskiqDecoratedTask[Any, Any]] = {}
+    # True only if broker runs in worker process.
     is_worker_process: bool = False
+    # True only if broker runs in scheduler process.
+    is_scheduler_process: bool = False
 
     def __init__(
         self,

taskiq/cli/scheduler/run.py

@@ -5,6 +5,7 @@
 
 from pycron import is_now
 
+from taskiq.abc.broker import AsyncBroker
 from taskiq.cli.scheduler.args import SchedulerArgs
 from taskiq.cli.utils import import_object, import_tasks
 from taskiq.kicker import AsyncKicker
@@ -70,6 +71,7 @@ async def run_scheduler(args: SchedulerArgs) -> None:  # noqa: C901, WPS210, WPS
 
     :param args: parsed CLI args.
     """
+    AsyncBroker.is_scheduler_process = True
     scheduler = import_object(args.scheduler)
     if not isinstance(scheduler, TaskiqScheduler):
         print(  # noqa: WPS421