63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import secrets
|
||
|
|
import threading
|
||
|
|
import time
|
||
|
|
from collections.abc import Callable
|
||
|
|
|
||
|
|
from .errors import Conflict, Missing
|
||
|
|
|
||
|
|
UNDO = 3.0
|
||
|
|
|
||
|
|
|
||
|
|
class Pending:
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self.lock = threading.Lock()
|
||
|
|
self.batches: dict[str, dict] = {}
|
||
|
|
self.applied: set[str] = set()
|
||
|
|
|
||
|
|
def add(self, items: list[dict], source: str, proof: dict | None, apply: Callable[[dict], None]) -> dict:
|
||
|
|
"""Args:
|
||
|
|
items: the checked inbox items.
|
||
|
|
source: operator, or rule:<id> for an automatic approval.
|
||
|
|
proof: the signature relayed to the executors once the undo
|
||
|
|
window passed; None when no item needs one.
|
||
|
|
apply: settles the batch after the undo window.
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
The batch without its proof and timer.
|
||
|
|
"""
|
||
|
|
batch_id = secrets.token_hex(8)
|
||
|
|
timer = threading.Timer(UNDO, self.fire, args=(batch_id, apply))
|
||
|
|
timer.daemon = True
|
||
|
|
batch = {"id": batch_id, "items": items, "source": source, "due": time.time() + UNDO}
|
||
|
|
with self.lock:
|
||
|
|
self.batches[batch_id] = {**batch, "proof": proof, "timer": timer}
|
||
|
|
timer.start()
|
||
|
|
return dict(batch)
|
||
|
|
|
||
|
|
def fire(self, batch_id: str, apply: Callable[[dict], None]) -> None:
|
||
|
|
with self.lock:
|
||
|
|
batch = self.batches.pop(batch_id, None)
|
||
|
|
if batch is not None:
|
||
|
|
self.applied.add(batch_id)
|
||
|
|
if batch is not None:
|
||
|
|
apply(batch)
|
||
|
|
|
||
|
|
def undo(self, batch_id: str) -> dict:
|
||
|
|
with self.lock:
|
||
|
|
batch = self.batches.pop(batch_id, None)
|
||
|
|
if batch is None and batch_id in self.applied:
|
||
|
|
raise Conflict({"id": batch_id, "status": "applied"})
|
||
|
|
if batch is None:
|
||
|
|
raise Missing(batch_id)
|
||
|
|
batch["timer"].cancel()
|
||
|
|
return batch
|
||
|
|
|
||
|
|
def holds(self, kind_name: str, item_id: str) -> bool:
|
||
|
|
with self.lock:
|
||
|
|
return any(item["kind"] == kind_name and item["id"] == item_id for batch in self.batches.values() for item in batch["items"])
|
||
|
|
|
||
|
|
def listed(self) -> list[dict]:
|
||
|
|
with self.lock:
|
||
|
|
return [{key: batch[key] for key in ("id", "items", "source", "due")} for batch in self.batches.values()]
|