108 lines
2.9 KiB
Python
108 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Self, Any, Callable, Optional, Sequence
|
|
from Interfaces.Application.AnPInterface import AnPInterface
|
|
from Models.QueuesModel import QueuesModel
|
|
from Utils.Common import Common
|
|
|
|
class QueuesManager:
|
|
|
|
def __init__(self:Self, anp:AnPInterface) -> None:
|
|
self.anp:AnPInterface = anp
|
|
self.__queues:dict[str, QueuesModel] = {}
|
|
|
|
self.update()
|
|
|
|
def update(self:Self) -> None:
|
|
key:str
|
|
|
|
for key in ("default_queues_files", "queues_files", "default_queues", "queues"):
|
|
self.create(self.anp.settings.get(key), True)
|
|
|
|
def reset(self:Self) -> None:
|
|
self.__queues = {}
|
|
self.update()
|
|
|
|
def close(self:Self) -> None:
|
|
self.remove()
|
|
|
|
def create(self:Self, inputs:Any|None, overwrite:bool = False) -> None:
|
|
|
|
subinputs:dict[str, Any|None]
|
|
|
|
for subinputs in Common.load_json(inputs, True):
|
|
|
|
key:str
|
|
queue:Any|None
|
|
|
|
for key, queue in subinputs.items():
|
|
if overwrite or key not in self.__queues:
|
|
self.__queues[key] = QueuesModel(key, queue)
|
|
|
|
def add(self:Self, key:str, callback:Callable[[Callable[[], None]], None], *arguments:list[Any|None]) -> int|None:
|
|
if key in self.__queues:
|
|
return self.__queues[key].add(callback, *arguments)
|
|
return None
|
|
|
|
def next(self:Self) -> None:
|
|
|
|
queue:QueuesModel
|
|
|
|
for queue in self.__queues.values():
|
|
queue.next()
|
|
|
|
def cancel(self:Self, keys:Optional[str|Sequence[str]] = False, _is:Optional[int|Sequence[int]] = None) -> bool:
|
|
if keys is None:
|
|
|
|
queue:QueuesModel
|
|
|
|
for queue in self.__queues.values():
|
|
if _is is None:
|
|
queue.cancel_all()
|
|
else:
|
|
|
|
i:int
|
|
|
|
for i in Common.get_keys(_is):
|
|
queue.cancel(i)
|
|
|
|
return True
|
|
|
|
done:bool = False
|
|
key:str
|
|
|
|
for key in Common.get_keys(keys):
|
|
if key in self.__queues:
|
|
if _is is None:
|
|
self.__queues[key].cancel_all()
|
|
done = True
|
|
else:
|
|
|
|
i:int
|
|
|
|
for i in Common.get_keys(_is):
|
|
self.__queues[key].cancel(i)
|
|
done = True
|
|
|
|
return done
|
|
|
|
def remove(self:Self, keys:Optional[str|Sequence[str]] = None) -> bool:
|
|
if keys is None:
|
|
|
|
for queue in self.__queues.values():
|
|
queue.cancel_all()
|
|
self.__queues = {}
|
|
|
|
return True
|
|
|
|
done:bool = False
|
|
key:str
|
|
|
|
for key in Common.get_keys(keys):
|
|
if key in self.__queues:
|
|
del self.__queues[key]
|
|
done = True
|
|
|
|
return done
|