Source code for pi_portal.modules.tasks.queue.persist_queue_router

"""A persist-queue backed task router."""

import logging

from pi_portal.modules.tasks.enums import RoutingLabel
from .bases.router_base import TaskRouterBase
from .persist_queue import Queue


class TaskRouter(TaskRouterBase):
  """A persist-queue back task router.

  :param log:  A logger instance.
  """

  def __init__(self, log: logging.Logger) -> None:
    self.queues = {}
    for routing_label in RoutingLabel:
      self.queues[routing_label] = Queue(log, routing_label=routing_label)