snowflake-cli 3.7.2__py3-none-any.whl → 3.8.0__py3-none-any.whl

snowflake/cli/_plugins/sql/commands.py

@@ -14,6 +14,8 @@
 
 from __future__ import annotations
 
+import sys
+from logging import getLogger
 from pathlib import Path
 from typing import List, Optional
 
@@ -26,8 +28,14 @@ from snowflake.cli.api.commands.flags import (
 from snowflake.cli.api.commands.overrideable_parameter import OverrideableOption
 from snowflake.cli.api.commands.snow_typer import SnowTyperFactory
 from snowflake.cli.api.commands.utils import parse_key_value_variables
-from snowflake.cli.api.output.types import CommandResult, MultipleResults, QueryResult
+from snowflake.cli.api.exceptions import CliArgumentError
+from snowflake.cli.api.output.types import (
+    CommandResult,
+    MultipleResults,
+    QueryResult,
+)
 
+logger = getLogger(__name__)
 # simple Typer with defaults because it won't become a command group as it contains only one command
 app = SnowTyperFactory()
 
@@ -38,7 +46,7 @@ SourceOption = OverrideableOption(
 )
 
 
-@app.command(name="sql", requires_connection=True, no_args_is_help=True)
+@app.command(name="sql", requires_connection=True, no_args_is_help=False)
 @with_project_definition(is_optional=True)
 def execute_sql(
     query: Optional[str] = SourceOption(
@@ -69,6 +77,12 @@ def execute_sql(
         "--retain-comments",
         help="Retains comments in queries passed to Snowflake",
     ),
+    single_transaction: Optional[bool] = typer.Option(
+        False,
+        help="Connects with autocommit disabled. Wraps BEGIN/COMMIT around statements to execute them as a single transaction, ensuring all commands complete successfully or no change is applied.",
+        flag_value=False,
+        is_flag=True,
+    ),
     **options,
 ) -> CommandResult:
     """
@@ -86,9 +100,46 @@ def execute_sql(
     if data_override:
         data = {v.key: v.value for v in parse_key_value_variables(data_override)}
 
-    single_statement, cursors = SqlManager().execute(
-        query, files, std_in, data=data, retain_comments=retain_comments
+    retain_comments = bool(retain_comments)
+    single_transaction = bool(single_transaction)
+    std_in = bool(std_in)
+
+    no_source_provided = not any([query, files, std_in])
+    if no_source_provided and not sys.stdin.isatty():
+        maybe_pipe = sys.stdin.read().strip()
+        if maybe_pipe:
+            query = maybe_pipe
+            std_in = True
+
+    if no_source_provided:
+        if single_transaction:
+            raise CliArgumentError("single transaction cannot be used with REPL")
+        from snowflake.cli._plugins.sql.repl import Repl
+
+        Repl(SqlManager(), data=data, retain_comments=retain_comments).run()
+        sys.exit(0)
+
+    manager = SqlManager()
+
+    expected_results_cnt, cursors = manager.execute(
+        query,
+        files,
+        std_in,
+        data=data,
+        retain_comments=retain_comments,
+        single_transaction=single_transaction,
     )
-    if single_statement:
-        return QueryResult(next(cursors))
+    if expected_results_cnt == 0:
+        # case expected if input only scheduled async queries
+        list(cursors)  # evaluate the result to schedule potential async queries
+        # ends gracefully with no message for consistency with snowsql.
+        sys.exit(0)
+
+    if expected_results_cnt == 1:
+        # evaluate the result to schedule async queries
+        results = list(cursors)
+        if not results:
+            return sys.exit(0)
+        return QueryResult(results[0])
+
     return MultipleResults((QueryResult(c) for c in cursors))

snowflake/cli/_plugins/sql/lexer/__init__.py

@@ -0,0 +1,7 @@
+from snowflake.cli._plugins.sql.lexer.completer import cli_completer
+from snowflake.cli._plugins.sql.lexer.lexer import CliLexer
+
+__all__ = (
+    "CliLexer",
+    "cli_completer",
+)

snowflake/cli/_plugins/sql/lexer/completer.py

@@ -0,0 +1,12 @@
+from prompt_toolkit.completion import WordCompleter, merge_completers
+from snowflake.cli._plugins.sql.lexer.functions import FUNCTIONS
+from snowflake.cli._plugins.sql.lexer.keywords import KEYWORDS
+from snowflake.cli._plugins.sql.lexer.types import TYPES
+
+functions_completer = WordCompleter(FUNCTIONS, ignore_case=True)
+keywords_completer = WordCompleter(KEYWORDS, ignore_case=True)
+types_completer = WordCompleter(TYPES, ignore_case=True)
+
+cli_completer = merge_completers(
+    [functions_completer, keywords_completer, types_completer]
+)

snowflake/cli/_plugins/sql/lexer/functions.py

@@ -0,0 +1,421 @@
+FUNCTIONS = (
+    "ABS",
+    "ACOS",
+    "ACOSH",
+    "ADD_MONTHS",
+    "AND",
+    "ANY_VALUE",
+    "APPROXIMATE_COUNT_DISTINCT",
+    "APPROXIMATE_JACCARD_INDEX",
+    "APPROXIMATE_SIMILARITY",
+    "APPROX_COUNT_DISTINCT",
+    "APPROX_PERCENTILE",
+    "APPROX_PERCENTILE_ACCUMULATE",
+    "APPROX_PERCENTILE_COMBINE",
+    "APPROX_PERCENTILE_ESTIMATE",
+    "APPROX_TOP_K",
+    "APPROX_TOP_K_ACCUMULATE",
+    "APPROX_TOP_K_COMBINE",
+    "APPROX_TOP_K_ESTIMATE",
+    "ARRAYAGG",
+    "ARRAYS_OVERLAP",
+    "ARRAY_AGG",
+    "ARRAY_APPEND",
+    "ARRAY_CAT",
+    "ARRAY_COMPACT",
+    "ARRAY_CONSTRUCT",
+    "ARRAY_CONSTRUCT_COMPACT",
+    "ARRAY_CONTAINS",
+    "ARRAY_INSERT",
+    "ARRAY_POSITION",
+    "ARRAY_PREPEND",
+    "ARRAY_SIZE",
+    "ARRAY_SLICE",
+    "ARRAY_TO_STRING",
+    "ASCII",
+    "ASIN",
+    "ASINH",
+    "AS_ARRAY",
+    "AS_BINARY",
+    "AS_BOOLEAN",
+    "AS_CHAR",
+    "AS_DATE",
+    "AS_DECIMAL",
+    "AS_DOUBLE",
+    "AS_INTEGER",
+    "AS_NUMBER",
+    "AS_OBJECT",
+    "AS_REAL",
+    "AS_TIME",
+    "AS_TIMESTAMP_LTZ",
+    "AS_TIMESTAMP_NTZ",
+    "AS_TIMESTAMP_TZ",
+    "AS_VARCHAR",
+    "ATAN",
+    "ATAN2",
+    "ATANH",
+    "AVG",
+    "BASE64_DECODE_BINARY",
+    "BASE64_DECODE_STRING",
+    "BASE64_ENCODE",
+    "BETWEEN",
+    "BINARY_AS_STRING",
+    "BITAND",
+    "BITANDAGG",
+    "BITAND_AGG",
+    "BITNOT",
+    "BITOR",
+    "BITORAGG",
+    "BITOR_AGG",
+    "BITSHIFTLEFT",
+    "BITSHIFTRIGHT",
+    "BITXOR",
+    "BITXORAGG",
+    "BITXOR_AGG",
+    "BIT_AND",
+    "BIT_ANDAGG",
+    "BIT_AND_AGG",
+    "BIT_LENGTH",
+    "BIT_NOT",
+    "BIT_OR",
+    "BIT_ORAGG",
+    "BIT_OR_AGG",
+    "BIT_SHIFTLEFT",
+    "BIT_SHIFTRIGHT",
+    "BIT_XOR",
+    "BIT_XORAGG",
+    "BIT_XOR_AGG",
+    "BOOLAND",
+    "BOOLNOT",
+    "BOOLOR",
+    "BOOLXOR",
+    "CASE",
+    "CAST",
+    "CBRT",
+    "CEIL",
+    "CHAR",
+    "CHARINDEX",
+    "CHECK_JSON",
+    "CHECK_XML",
+    "CHR",
+    "COALESCE",
+    "COMPRESS",
+    "CONCAT",
+    "CONTAINS",
+    "CONVERT_TIMEZONE",
+    "CORR",
+    "COS",
+    "COSH",
+    "COT",
+    "COUNT",
+    "COVAR_POP",
+    "COVAR_SAMP",
+    "CUME_DIST",
+    "CURRENT_CLIENT",
+    "CURRENT_DATABASE",
+    "CURRENT_DATE",
+    "CURRENT_ROLE",
+    "CURRENT_SCHEMA",
+    "CURRENT_SCHEMAS",
+    "CURRENT_SESSION",
+    "CURRENT_STATEMENT",
+    "CURRENT_TIME",
+    "CURRENT_TIMESTAMP",
+    "CURRENT_TRANSACTION",
+    "CURRENT_USER",
+    "CURRENT_VERSION",
+    "CURRENT_WAREHOUSE",
+    "DATEADD",
+    "DATEDIFF",
+    "DATEFROMPARTS",
+    "DATE_FROM_PARTS",
+    "DATE_PART",
+    "DATE_TRUNC",
+    "DAY",
+    "DAYNAME",
+    "DAYOFMONTH",
+    "DAYOFWEEK",
+    "DAYOFWEEKISO",
+    "DAYOFYEAR",
+    "DECODE",
+    "DECOMPRESS_BINARY",
+    "DECOMPRESS_STRING",
+    "DEFAULT_FLAGS",
+    "DEGREES",
+    "DENSE_RANK",
+    "DIV",
+    "EDITDISTANCE",
+    "ENDSWITH",
+    "EQUAL_NULL",
+    "EXP",
+    "EXTRACT",
+    "FACTORIAL",
+    "FIRST_VALUE",
+    "FLOOR",
+    "GET",
+    "GET_DDL",
+    "GET_PATH",
+    "GET_USAGE_FOR_QUERY",
+    "GREATEST",
+    "GROUPING_ID",
+    "HASH",
+    "HASH_AGG",
+    "HAVERSINE",
+    "HEX_DECODE_BINARY",
+    "HEX_DECODE_STRING",
+    "HEX_ENCODE",
+    "HLL",
+    "HLL_ACCUMULATE",
+    "HLL_COMBINE",
+    "HLL_ESTIMATE",
+    "HLL_EXPORT",
+    "HLL_IMPORT",
+    "HOUR",
+    "IFF",
+    "IFNULL",
+    "ILIKE",
+    "IN",
+    "INITCAP",
+    "INSERT",
+    "IS_ARRAY",
+    "IS_BINARY",
+    "IS_BOOLEAN",
+    "IS_CHAR",
+    "IS_DATE",
+    "IS_DATE_VALUE",
+    "IS_DECIMAL",
+    "IS_DOUBLE",
+    "IS_INTEGER",
+    "IS_NULL_VALUE",
+    "IS_OBJECT",
+    "IS_REAL",
+    "IS_TIME",
+    "IS_TIMESTAMP_LTZ",
+    "IS_TIMESTAMP_NTZ",
+    "IS_TIMESTAMP_TZ",
+    "IS_VARCHAR",
+    "LAG",
+    "LAST_DAY",
+    "LAST_QUERY_ID",
+    "LAST_TRANSACTION",
+    "LAST_VALUE",
+    "LEAD",
+    "LEAST",
+    "LEFT",
+    "LENGTH",
+    "LIKE",
+    "LISTAGG",
+    "LN",
+    "LOCALTIME",
+    "LOCALTIMESTAMP",
+    "LOG",
+    "LOWER",
+    "LPAD",
+    "LQ",
+    "LQ",
+    "LTRIM",
+    "MASK_TIMESTAMP",
+    "MAX",
+    "MD5",
+    "MD5_BINARY",
+    "MD5_HEX",
+    "MD5_NUMBER",
+    "MEDIAN",
+    "MIN",
+    "MINHASH",
+    "MINHASH_COMBINE",
+    "MINUTE",
+    "MOD",
+    "MONTH",
+    "MONTHNAME",
+    "NEGATE",
+    "NEXT_DAY",
+    "NORMAL",
+    "NORMALIZE",
+    "NOT",
+    "NTH_VALUE",
+    "NTILE",
+    "NULLIF",
+    "NVL",
+    "NVL2",
+    "OBJECTAGG",
+    "OBJECT_AGG",
+    "OBJECT_CONSTRUCT",
+    "OBJECT_DELETE",
+    "OBJECT_INSERT",
+    "OCTET_LENGTH",
+    "OR",
+    "PARSE_IP",
+    "PARSE_JSON",
+    "PARSE_URL",
+    "PARSE_XML",
+    "PERCENTILE_CONT",
+    "PERCENTILE_DISC",
+    "PERCENT_RANK",
+    "PI",
+    "POSITION",
+    "POW",
+    "POWER",
+    "PREVIOUS_DAY",
+    "QPDPP",
+    "QUARTER",
+    "RADIANS",
+    "RANDOM",
+    "RANDSTR",
+    "RANK",
+    "REGEXP",
+    "REGEXP_COUNT",
+    "REGEXP_INSTR",
+    "REGEXP_LIKE",
+    "REGEXP_REPLACE",
+    "REGEXP_SUBSTR",
+    "REGR_AVGX",
+    "REGR_AVGY",
+    "REGR_COUNT",
+    "REGR_INTERCEPT",
+    "REGR_R2",
+    "REGR_SLOPE",
+    "REGR_SXX",
+    "REGR_SXY",
+    "REGR_SYY",
+    "REGR_VALX",
+    "REGR_VALY",
+    "REPEAT",
+    "REPLACE",
+    "REVERSE",
+    "RIGHT",
+    "RLIKE",
+    "ROUND",
+    "ROW_NUMBER",
+    "RPAD",
+    "RTRIM",
+    "RTRIMMED_LENGTH",
+    "SECOND",
+    "SEQ1",
+    "SEQ2",
+    "SEQ4",
+    "SEQ8",
+    "SHA1",
+    "SHA1_BINARY",
+    "SHA1_HEX",
+    "SHA2",
+    "SHA2_BINARY",
+    "SHA2_HEX",
+    "SHOW_COLUMNS_IN_TABLE",
+    "SHOW_DATABASES",
+    "SHOW_DATABASES_LIKE",
+    "SHOW_FILE_FORMATS",
+    "SHOW_FILE_FORMATS_LIKE",
+    "SHOW_GRANTS_ON_DATABASE",
+    "SHOW_GRANTS_ON_OBJECT",
+    "SHOW_GRANTS_ON_SCHEMA",
+    "SHOW_GRANTS_ON_TABLE",
+    "SHOW_GRANTS_TO_ROLE",
+    "SHOW_SCHEMAS",
+    "SHOW_SCHEMAS_LIKE",
+    "SHOW_SEQUENCES",
+    "SHOW_SEQUENCES_LIKE",
+    "SHOW_STAGES",
+    "SHOW_STAGES_IN_DB",
+    "SHOW_STAGES_LIKE",
+    "SHOW_TABLES",
+    "SHOW_TABLES_LIKE",
+    "SHOW_USER_FUNCTIONS",
+    "SHOW_USER_FUNCTIONS_LIKE",
+    "SHOW_VIEWS",
+    "SHOW_VIEWS_LIKE",
+    "SIGN",
+    "SIN",
+    "SINH",
+    "SPACE",
+    "SPLIT_PART",
+    "SQL_UDF_DESC",
+    "SQRT",
+    "SQUARE",
+    "STARTSWITH",
+    "STDDEV",
+    "STDDEV_POP",
+    "STDDEV_SAMP",
+    "STRING_AS_BINARY",
+    "STRIP_NULL_VALUE",
+    "SUBSTR",
+    "SUBSTRING",
+    "SUM",
+    "TAN",
+    "TANH",
+    "TIMEADD",
+    "TIMEDIFF",
+    "TIMEFROMPARTS",
+    "TIMESTAMPADD",
+    "TIMESTAMPDIFF",
+    "TIMESTAMPFROMPARTS",
+    "TIMESTAMPLTZFROMPARTS",
+    "TIMESTAMPNTZFROMPARTS",
+    "TIMESTAMPTZFROMPARTS",
+    "TIMESTAMP_FROM_PARTS",
+    "TIMESTAMP_LTZ_FROM_PARTS",
+    "TIMESTAMP_NTZ_FROM_PARTS",
+    "TIMESTAMP_TZ_FROM_PARTS",
+    "TIME_FROM_PARTS",
+    "TIME_SLICE",
+    "TO_ARRAY",
+    "TO_BINARY",
+    "TO_BOOLEAN",
+    "TO_CHAR",
+    "TO_DATE",
+    "TO_DECIMAL",
+    "TO_DOUBLE",
+    "TO_JSON",
+    "TO_NUMBER",
+    "TO_NUMERIC",
+    "TO_OBJECT",
+    "TO_TIME",
+    "TO_TIMESTAMP",
+    "TO_TIMESTAMP_LTZ",
+    "TO_TIMESTAMP_NTZ",
+    "TO_TIMESTAMP_TZ",
+    "TO_VARCHAR",
+    "TO_VARIANT",
+    "TO_XML",
+    "TRANSLATE",
+    "TRIM",
+    "TRUNC",
+    "TRUNCATE",
+    "TRY_BASE64_DECODE_BINARY",
+    "TRY_BASE64_DECODE_STRING",
+    "TRY_CAST",
+    "TRY_HEX_DECODE_BINARY",
+    "TRY_HEX_DECODE_STRING",
+    "TRY_TO_BINARY",
+    "TRY_TO_BOOLEAN",
+    "TRY_TO_DATE",
+    "TRY_TO_DECIMAL",
+    "TRY_TO_DOUBLE",
+    "TRY_TO_NUMBER",
+    "TRY_TO_NUMERIC",
+    "TRY_TO_TIME",
+    "TRY_TO_TIMESTAMP",
+    "TRY_TO_TIMESTAMP_LTZ",
+    "TRY_TO_TIMESTAMP_NTZ",
+    "TRY_TO_TIMESTAMP_TZ",
+    "TYPEOF",
+    "UNICODE",
+    "UNIFORM",
+    "UPPER",
+    "UUID_STRING",
+    "VARIANCE",
+    "VARIANCE_POP",
+    "VARIANCE_SAMP",
+    "VAR_POP",
+    "VAR_SAMP",
+    "WEEK",
+    "WEEKISO",
+    "WEEKOFYEAR",
+    "WIDTH_BUCKET",
+    "XMLGET",
+    "YEAR",
+    "YEAROFWEEK",
+    "YEAROFWEEKISO",
+    "ZEROIFNULL",
+    "ZIPF",
+)