fix: source table name and select statement both in parentheses

reata · Mar 13, 2022 · 3a8db5d · 3a8db5d
1 parent 57a3b64
commit 3a8db5d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/sqllineage/utils/sqlparse.py b/sqllineage/utils/sqlparse.py
@@ -22,8 +22,8 @@ def is_subquery(token: TokenList) -> bool:
     if isinstance(token, Parenthesis):
         # in case of subquery in nested parenthesis, find the innermost one first
         while True:
-            _, sub_paren = token.token_next_by(i=Parenthesis)
-            if sub_paren is not None:
+            idx, sub_paren = token.token_next_by(i=Parenthesis)
+            if sub_paren is not None and idx == 1:
                 token = sub_paren
             else:
                 break

diff --git a/tests/test_others.py b/tests/test_others.py
@@ -53,6 +53,12 @@ def test_create_as_with_parenthesis_around_table_name():
     )
 
 
+def test_create_as_with_parenthesis_around_both():
+    assert_table_lineage_equal(
+        "CREATE TABLE tab1 AS (SELECT * FROM (tab2))", {"tab2"}, {"tab1"}
+    )
+
+
 def test_create_like():
     assert_table_lineage_equal("CREATE TABLE tab1 LIKE tab2", {"tab2"}, {"tab1"})