from __future__ import annotations


class SQLStatementOperation:
    def __init__(self, payload: dict[str, str]):
        self.payload = payload

    def undo(self) -> None:
        rollback_sql = self.payload.get("rollback_sql")
        if rollback_sql is None:
            raise ValueError("Missing 'rollback_sql' in payload")
