#965

mabel-dev · Apr 2, 2023 · a8be8da · a8be8da
1 parent 1926543
commit a8be8da
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/opteryx/components/logical_planner/custom_builders.py b/opteryx/components/logical_planner/custom_builders.py
@@ -42,13 +42,13 @@ class RelationDescription:
     cache: Any = None
 
 
-def extract_show_filter(ast):
+def extract_show_filter(ast, name_column="name"):
     """filters are used in SHOW queries"""
     filters = ast["filter"]
     if filters is None:
         return None
     if "Like" in filters:
-        left = ExpressionTreeNode(NodeType.IDENTIFIER, value="name")
+        left = ExpressionTreeNode(NodeType.IDENTIFIER, value=name_column)
         right = ExpressionTreeNode(NodeType.LITERAL_VARCHAR, value=filters["Like"])
         root = ExpressionTreeNode(
             NodeType.COMPARISON_OPERATOR,

diff --git a/opteryx/components/logical_planner/queries.py b/opteryx/components/logical_planner/queries.py
@@ -457,7 +457,7 @@ def show_variables_query(ast, properties):
     plan.add_node("show", show)
     last_node = "show"
 
-    filters = custom_builders.extract_show_filter(ast["ShowVariables"])
+    filters = custom_builders.extract_show_filter(ast["ShowVariables"], "variable_name")
     if filters:
         plan.add_node(
             "filter",

diff --git a/tests/sql_battery/test_shapes_and_errors_battery.py b/tests/sql_battery/test_shapes_and_errors_battery.py
@@ -922,6 +922,7 @@ def test_sql_battery(statement, rows, columns, exception):
         print(
             f"\033[0;36m{(index + 1):04}\033[0m {printable[0:width - 1].ljust(width)}",
             end="",
+            flush=True,
         )
         test_sql_battery(statement, rows, cols, err)
         print(f"\033[0;32m{str(int((time.monotonic_ns() - start)/1e6)).rjust(4)}ms\033[0m ✅")