Update keywords_lists.py (#428)

* Update keywords_lists.py Add HAVING keyword for column sections * testing for HAVING keyword * Update test/test_getting_columns.py --------- Co-authored-by: Andy Tang <[email protected]> Co-authored-by: Maciej Brencz <[email protected]>
macbre · Sep 23, 2023 · c76aacb · c76aacb
1 parent 67d8c36
commit c76aacb
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/sql_metadata/keywords_lists.py b/sql_metadata/keywords_lists.py
@@ -9,6 +9,7 @@
 KEYWORDS_BEFORE_COLUMNS = {
     "SELECT",
     "WHERE",
+    "HAVING",
     "ORDERBY",
     "GROUPBY",
     "ON",
@@ -55,6 +56,7 @@
 COLUMNS_SECTIONS = {
     "SELECT": "select",
     "WHERE": "where",
+    "HAVING": "having",
     "ORDERBY": "order_by",
     "ON": "join",
     "USING": "join",

diff --git a/test/test_getting_columns.py b/test/test_getting_columns.py
@@ -461,3 +461,19 @@ def test_aliases_switching_column_names():
     parsed = Parser(query)
     assert parsed.columns == ["a", "b"]
     assert parsed.columns_dict == {"select": ["a", "b"]}
+
+
+def test_having_columns():
+    query = """
+    SELECT Country
+    FROM Customers
+    GROUP BY Country
+    HAVING COUNT(CustomerID) > 5;
+    """
+    parsed = Parser(query)
+    assert parsed.columns == ["Country", "CustomerID"]
+    assert parsed.columns_dict == {
+        "select": ["Country"],
+        "group_by": ["Country"],
+        "having": ["CustomerID"],
+    }