Skip to content

Commit 9bf6281

Browse files
authored
Fix columns reading for queries with double INNER JOIN (#583)
Resolves #580
1 parent 3d0c043 commit 9bf6281

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

sql_metadata/keywords_lists.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@
6565
"INTO": "insert",
6666
"SET": "update",
6767
"GROUPBY": "group_by",
68+
"INNERJOIN": "inner_join",
6869
}
6970

7071

test/test_getting_columns.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -528,3 +528,16 @@ def test_nested_queries():
528528
parser = Parser(query)
529529
assert parser.columns == ["dt"]
530530
assert parser.columns_dict == {"select": ["dt"]}
531+
532+
533+
def test_double_inner_join():
534+
query = """
535+
SELECT loan.loan_id, district.a3 AS district, district.a11 AS average_salary
536+
FROM loan
537+
INNER JOIN account ON loan.account_id = account.account_id
538+
INNER JOIN district ON account.district_id = district.district_id WHERE loan.duration = 60
539+
"""
540+
541+
parser = Parser(query)
542+
assert "loan.account_id" in parser.columns
543+
assert parser.tables == ["loan", "account"]

0 commit comments

Comments
 (0)