Function update_environment

misc/python/materialize/mzexplore/sql.py:128–140

(
    db: Database, env: dict[str, str]
)

@contextlib.contextmanager
def update_environment(
    db: Database, env: dict[str, str]
) -> Generator[Database, None, None]:
    original = dict()
    for e in db.query_all("SHOW ALL"):
        key, old_value = e["name"], e["setting"]
        if key in env:
            original[key] = old_value
            new_value = env[key]
            db.execute(f"SET {identifier(key)} = {literal(new_value)}")
    yield db
    for key, old_value in original.items():
        db.execute(f"SET {identifier(key)} = {literal(old_value)}")


# Utility functions