sql_metadata_parser_extract_tables_columns_from_query.py

1from sql_metadata import Parser
2
3sql = "SELECT a, b FROM table_1 JOIN table_2 ON table_1.id = table_2.id WHERE c > 10"
4parser = Parser(sql)
5
6# Get table names
7print(parser.tables)
8# ['table_1', 'table_2']
9
10# Get column names
11print(parser.columns)
12# ['a', 'b', 'table_1.id', 'table_2.id', 'c']
13
14# Get columns to tables mapping
15print(parser.columns_dict)
16# {'select': ['a', 'b'], 'join': ['table_1.id', 'table_2.id'], 'where': ['c']}