import typing as t from sqlglot import expressions as exp def replace_id_value(node, replacement_mapping: t.Dict[exp.Identifier, exp.Identifier]): if isinstance(node, exp.Identifier) and node in replacement_mapping: node = node.replace(replacement_mapping[node].copy()) return node