53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
|
import asyncio
|
||
|
|
from typing import Optional
|
||
|
|
from enum import Enum
|
||
|
|
|
||
|
|
|
||
|
|
class QueueState(Enum):
|
||
|
|
STOPPED = 0
|
||
|
|
IDLE = 1
|
||
|
|
POPPING = 2 # As in: Popping actions from the queue and performing them
|
||
|
|
|
||
|
|
|
||
|
|
class ActionQueue:
|
||
|
|
def __init__(self, idle_action=None) -> None:
|
||
|
|
self.queued_actions: list = []
|
||
|
|
self.idle_action: Optional[any] = None
|
||
|
|
self.state = QueueState.STOPPED
|
||
|
|
self.background_task = None
|
||
|
|
|
||
|
|
if idle_action:
|
||
|
|
self.set_idle_action(idle_action)
|
||
|
|
|
||
|
|
async def run_queue(self):
|
||
|
|
while len(self.queued_actions) > 0:
|
||
|
|
self.state = QueueState.POPPING
|
||
|
|
action = self.queued_actions.pop(0)
|
||
|
|
action[0](*(action[1]), **(action[2]))
|
||
|
|
|
||
|
|
self.state = QueueState.IDLE
|
||
|
|
self.idle_action()
|
||
|
|
|
||
|
|
async def stop_queue(self):
|
||
|
|
if self.background_task is None:
|
||
|
|
return
|
||
|
|
|
||
|
|
self.state = QueueState.STOPPED
|
||
|
|
self.background_task.cancel()
|
||
|
|
|
||
|
|
async def restart_queue(self):
|
||
|
|
await self.stop_queue()
|
||
|
|
self.background_task = asyncio.create_task(self.run_queue())
|
||
|
|
|
||
|
|
async def add_action_to_queue(self, action, *args, **kwargs):
|
||
|
|
self.queued_actions.append((action, args, kwargs))
|
||
|
|
|
||
|
|
if self.state == QueueState.IDLE:
|
||
|
|
await self.restart_queue()
|
||
|
|
|
||
|
|
async def set_idle_action(self, action):
|
||
|
|
self.idle_action = action
|
||
|
|
|
||
|
|
if self.state == QueueState.IDLE:
|
||
|
|
await self.restart_queue()
|