path: root/sqlglot/dataframe/sql/transforms.py

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