from datetime import datetime _store: dict[int, tuple[datetime, int]] = {} def is_on_cooldown(action_id: int) -> bool: if action_id not in _store: return False triggered_at, duration = _store[action_id] return (datetime.utcnow() - triggered_at).total_seconds() < duration def set_cooldown(action_id: int, duration_seconds: int) -> None: if duration_seconds > 0: _store[action_id] = (datetime.utcnow(), duration_seconds)