Column Lineage is failing when there a column name is associated with multiple tables.

[AyushAryan]

Hi everyone,
I was trying to create lineage column lineage with the below sample query and was unable to get lineage for TABLE1 and TABLE2 related columns. Can anyone help me with this

SELECT
    ACTUAL_TABLE.SAMPLE_COLUMN,
    TABLE.INPUT_COL,
    TABLE.INPUT_COL_2,
    TABLE1.INPUT_COL_2,
    TABLE2.INPUT_COL_2,
    TABLE.INPUT_COL_3,
    TABLE1.INPUT_COL_3,
    TABLE2.INPUT_COL_3
FROM
    DATA ACTUAL_TABLE
LEFT OUTER JOIN LOOKUP.REF_COUNTRY_2 TABLE
    ON ACTUAL_TABLE.DATA_SUB = TABLE.INPUT_COL
    AND TABLE.DATE_STAMP = TO_DATE('12/12/2001', 'MM/DD/YYYY')
LEFT OUTER JOIN LOOKUP.REF_COUNTRY_2 TABLE1
    ON ACTUAL_TABLE.DATA_SUB = TABLE1.INPUT_COL
    AND TABLE1.DATE_STAMP = TO_DATE('12/12/2002', 'MM/DD/YYYY')
LEFT OUTER JOIN LOOKUP.REF_COUNTRY_2 TABLE2
    ON ACTUAL_TABLE.DATA_SUB = TABLE2.INPUT_COL
    AND TABLE2.DATE_STAMP = TO_DATE('12/12/2004', 'MM/DD/YYYY').

Here is my snippet of the code-:

ast = sqlglot.parse_one(clean, read = "oracle")
qualified_expression = sqlglot.parse_one(ast, dialect= "oracle")
select_exprs = parsed.find_all(exp.Select)

for select_expr in select_exprs:
    for column in select_expr.args["expressions"]:
        if isinstance(column, exp.Star):
            continue
        column_name = column.alias_or_name
        lineage_node = lineage(column_name, qualified_expression, dialect="oracle")
        for node in lineage_node.walk():
            print(node.name)

[tobymao]

you should add explicit alias your column names since they are duplicates