Source code for soulfire.actions

from .bot_live_pb2 import BOT_ACTION_STATUS_COMPLETED, BotActionResult
from .errors import SoulFireActionError



[docs]
def require_action(result: BotActionResult) -> BotActionResult:
    if result.status != BOT_ACTION_STATUS_COMPLETED:
        raise SoulFireActionError(result)
    return result




[docs]
def action_headers(headers: dict[str, str] | None, token: str | None) -> dict[str, str] | None:
    if token is None:
        return headers
    result = dict(headers or {})
    result["X-SoulFire-Control-Token"] = token
    return result