Merge pull request #1710 from dbcli/RW/suppress-sqlglotrs-warnings

Suppress warnings when `sqlglotrs` is installed

Author: rolandwalker

changelog.md

@@ -6,6 +6,12 @@ Features
 * Add prompt format string for literal backslash.
 
 
+Bug Fixes
+---------
+* Suppress warnings when `sqlglotrs` is installed.
+
+
+
 1.64.0 (2026/03/13)
 ==============
 

mycli/main.py

@@ -26,6 +26,7 @@
 from textwrap import dedent
 from time import sleep, time
 from urllib.parse import parse_qs, unquote, urlparse
+import warnings
 
 from cli_helpers.tabular_output import TabularOutputFormatter, preprocessors
 from cli_helpers.tabular_output.output_formatter import MISSING_VALUE as DEFAULT_MISSING_VALUE
@@ -58,9 +59,18 @@
 from pymysql.constants.CR import CR_SERVER_LOST
 from pymysql.constants.ER import ACCESS_DENIED_ERROR, HANDSHAKE_ERROR
 from pymysql.cursors import Cursor
-import sqlglot
 import sqlparse
 
+with warnings.catch_warnings():
+    # for sqlglot v29.0.1
+    warnings.filterwarnings(
+        'ignore',
+        message=r'sqlglot\[rs\] is deprecated',
+        category=UserWarning,
+        module='sqlglot',
+    )
+    import sqlglot
+
 from mycli import __version__
 from mycli.clibuffer import cli_is_multiline
 from mycli.clistyle import style_factory_helpers, style_factory_toolkit

mycli/packages/hybrid_redirection.py

@@ -1,7 +1,16 @@
 import functools
 import logging
-
-import sqlglot
+import warnings
+
+with warnings.catch_warnings():
+    # for sqlglot v29.0.1
+    warnings.filterwarnings(
+        'ignore',
+        message=r'sqlglot\[rs\] is deprecated',
+        category=UserWarning,
+        module='sqlglot',
+    )
+    import sqlglot
 
 from mycli.compat import WIN
 from mycli.packages.special.delimitercommand import DelimiterCommand

mycli/packages/parseutils.py

@@ -2,12 +2,22 @@
 
 import re
 from typing import Any, Generator, Literal
+import warnings
 
-import sqlglot
 import sqlparse
 from sqlparse.sql import Function, Identifier, IdentifierList, Token, TokenList
 from sqlparse.tokens import DML, Keyword, Punctuation
 
+with warnings.catch_warnings():
+    # for sqlglot v29.0.1
+    warnings.filterwarnings(
+        'ignore',
+        message=r'sqlglot\[rs\] is deprecated',
+        category=UserWarning,
+        module='sqlglot',
+    )
+    import sqlglot
+
 sqlparse.engine.grouping.MAX_GROUPING_DEPTH = None  # type: ignore[assignment]
 sqlparse.engine.grouping.MAX_GROUPING_TOKENS = None  # type: ignore[assignment]