fakesnow 0.9.38__py3-none-any.whl → 0.9.40__py3-none-any.whl

fakesnow/transforms/show.py

@@ -11,11 +11,52 @@ def fs_global_creation_sql() -> str:
         {SQL_CREATE_VIEW_SHOW_OBJECTS};
         {SQL_CREATE_VIEW_SHOW_TABLES};
         {SQL_CREATE_VIEW_SHOW_VIEWS};
+        {SQL_CREATE_VIEW_SHOW_COLUMNS};
+        {SQL_CREATE_VIEW_SHOW_DATABASES};
+        {SQL_CREATE_VIEW_SHOW_FUNCTIONS};
+        {SQL_CREATE_VIEW_SHOW_SCHEMAS};
+        {SQL_CREATE_VIEW_SHOW_PROCEDURES};
     """
 
 
+# see https://docs.snowflake.com/en/sql-reference/sql/show-columns
+SQL_CREATE_VIEW_SHOW_COLUMNS = """
+create view if not exists _fs_global._fs_information_schema._fs_show_columns as
+SELECT
+    table_name,
+    table_schema as "schema_name",
+    column_name,
+    CASE
+        WHEN data_type = 'NUMBER' THEN
+            '{"type":"FIXED","precision":' || numeric_precision || ',"scale":' || numeric_scale || ',"nullable":true}'
+        WHEN data_type = 'TEXT' THEN
+            '{"type":"TEXT","length":' || coalesce(character_maximum_length,16777216)  || ',"byteLength":' ||
+            CASE
+                WHEN character_maximum_length = 16777216 THEN 16777216
+                ELSE coalesce(character_maximum_length*4,16777216)
+            END  || ',"nullable":true,"fixed":false}'
+        WHEN data_type in ('TIMESTAMP_NTZ','TIMESTAMP_TZ','TIME') THEN
+            '{"type":"' || data_type || '","precision":0,"scale":9,"nullable":true}'
+        WHEN data_type = 'FLOAT' THEN '{"type":"REAL","nullable":true}'
+        WHEN data_type = 'BINARY' THEN
+            '{"type":"BINARY","length":8388608,"byteLength":8388608,"nullable":true,"fixed":true}'
+        ELSE '{"type":"' || data_type || '","nullable":true}'
+    END as "data_type",
+    CASE WHEN is_nullable = 'YES' THEN 'true' ELSE 'false' END as "null?",
+    COALESCE(column_default, '') as "default",
+    'COLUMN' as "kind",
+    '' as "expression",
+    COALESCE(comment, '') as "comment",
+    table_catalog as "database_name",
+    '' as "autoincrement",
+    NULL as "schema_evolution_record"
+FROM _fs_global._fs_information_schema._fs_columns
+ORDER BY table_catalog, table_schema, table_name, ordinal_position
+"""
+
+
 def show_columns(
-    expression: exp.Expression, current_database: str | None = None, current_schema: str | None = None
+    expression: exp.Expression, current_database: str | None, current_schema: str | None
 ) -> exp.Expression:
     """Transform SHOW COLUMNS to a query against the fs global information_schema columns table.
 
@@ -29,12 +70,7 @@ def show_columns(
     scope_kind = expression.args.get("scope_kind")
     table = expression.find(exp.Table)
 
-    if scope_kind == "ACCOUNT" or not scope_kind:
-        # all columns
-        catalog = None
-        schema = None
-        table = None
-    elif scope_kind == "DATABASE" and table:
+    if scope_kind == "DATABASE" and table:
         catalog = table.name
         schema = None
         table = None
@@ -46,42 +82,39 @@ def show_columns(
         catalog = table.catalog or current_database
         schema = table.db or current_schema
         table = table.name
+    elif scope_kind == "ACCOUNT":
+        # all columns
+        catalog = None
+        schema = None
+        table = None
+    elif not scope_kind:
+        # no explicit scope - show current database and schema only
+        catalog = current_database
+        schema = current_schema
+        table = None
     else:
         raise NotImplementedError(f"show_object_columns: {expression.sql(dialect='snowflake')}")
 
+    where = ["1=1"]
+    if catalog:
+        where.append(f"database_name = '{catalog}'")
+    if schema:
+        where.append(f"schema_name = '{schema}'")
+    if table:
+        where.append(f"table_name = '{table}'")
+    where_clause = " AND ".join(where)
+
     query = f"""
-    SELECT
-        table_name,
-        table_schema as "schema_name",
-        column_name,
-        CASE
-            WHEN data_type = 'NUMBER' THEN '{{"type":"FIXED","precision":'|| numeric_precision || ',"scale":' || numeric_scale || ',"nullable":true}}'
-            WHEN data_type = 'TEXT' THEN '{{"type":"TEXT","length":' || coalesce(character_maximum_length,16777216)  || ',"byteLength":' || CASE WHEN character_maximum_length = 16777216 THEN 16777216 ELSE coalesce(character_maximum_length*4,16777216) END  || ',"nullable":true,"fixed":false}}'
-            WHEN data_type in ('TIMESTAMP_NTZ','TIMESTAMP_TZ','TIME') THEN '{{"type":"' || data_type || '","precision":0,"scale":9,"nullable":true}}'
-            WHEN data_type = 'FLOAT' THEN '{{"type":"REAL","nullable":true}}'
-            WHEN data_type = 'BINARY' THEN '{{"type":"BINARY","length":8388608,"byteLength":8388608,"nullable":true,"fixed":true}}'
-            ELSE '{{"type":"' || data_type || '","nullable":true}}'
-        END as "data_type",
-        CASE WHEN is_nullable = 'YES' THEN 'true' ELSE 'false' END as "null?",
-        COALESCE(column_default, '') as "default",
-        'COLUMN' as "kind",
-        '' as "expression",
-        COALESCE(comment, '') as "comment",
-        table_catalog as "database_name",
-        '' as "autoincrement",
-        NULL as "schema_evolution_record"
-    FROM _fs_global._fs_information_schema._fs_columns c
-    WHERE 1=1
-    {f"AND table_catalog = '{catalog}'" if catalog else ""}
-    {f"AND table_schema = '{schema}'" if schema else ""}
-    {f"AND table_name = '{table}'" if table else ""}
-    ORDER BY table_name, ordinal_position
-    """  # noqa: E501
+    SELECT *
+    FROM _fs_global._fs_information_schema._fs_show_columns
+    WHERE {where_clause}
+    """
 
     return sqlglot.parse_one(query, read="duckdb")
 
 
-SQL_SHOW_DATABASES = """
+SQL_CREATE_VIEW_SHOW_DATABASES = """
+create view if not exists _fs_global._fs_information_schema._fs_show_databases as
 SELECT
     to_timestamp(0)::timestamptz as 'created_on',
     database_name as 'name',
@@ -102,18 +135,18 @@ WHERE database_name NOT IN ('memory', '_fs_global')
 
 
 def show_databases(expression: exp.Expression) -> exp.Expression:
-    """Transform SHOW DATABASES to a query against the information_schema.schemata table.
+    """Transform SHOW DATABASES to a query against _fs_show_databases.
 
     See https://docs.snowflake.com/en/sql-reference/sql/show-databases
     """
     if isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "DATABASES":
-        return sqlglot.parse_one(SQL_SHOW_DATABASES, read="duckdb")
+        return sqlglot.parse_one("SELECT * FROM _fs_global._fs_information_schema._fs_show_databases", read="duckdb")
 
     return expression
 
 
-# returns zero rows
-SQL_SHOW_FUNCTIONS = """
+SQL_CREATE_VIEW_SHOW_FUNCTIONS = """
+create view if not exists _fs_global._fs_information_schema._fs_show_functions as
 SELECT
     '1970-01-01 00:00:00 UTC'::timestamptz as created_on,
     'SYSTIMESTAMP' as name,
@@ -145,73 +178,14 @@ def show_functions(expression: exp.Expression) -> exp.Expression:
     See https://docs.snowflake.com/en/sql-reference/sql/show-functions
     """
     if isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "FUNCTIONS":
-        return sqlglot.parse_one(SQL_SHOW_FUNCTIONS, read="duckdb")
-
-    return expression
-
-
-# returns zero rows
-SQL_SHOW_WAREHOUSES = """
-SELECT
-    'FAKESNOW_WAREHOUSE' as name,
-    'STARTED' as state,
-    'STANDARD' as type,
-    'X-Small' as size,
-    1 as min_cluster_count,
-    1 as max_cluster_count,
-    1 as started_clusters,
-    0 as running,
-    0 as queued,
-    'N' as is_default,
-    'N' as is_current,
-    600 as auto_suspend,
-    'true' as auto_resume,
-    -- nb: deliberate space before '100' to match Snowflake's output
-    ' 100' as available,
-    '0' as provisioning,
-    '0' as quiescing,
-    '0' as other,
-    '1970-01-01 00:00:00.000000 UTC'::timestamptz as created_on,
-    '1970-01-01 00:00:00.000000 UTC'::timestamptz as resumed_on,
-    '1970-01-01 00:00:00.000000 UTC'::timestamptz as updated_on,
-    'SYSADMIN' as owner,
-    '' as comment,
-    'false' as enable_query_acceleration,
-    8 as query_acceleration_max_scale_factor,
-    'null' as resource_monitor,
-
-    -- deprecated - these 5 cols are for internal use
-    0 as actives,
-    0 as pendings,
-    0 as failed,
-    0 as suspended,
-    '123456789012' as uuid,
-
-    'STANDARD' as scaling_policy,
-    NULL as budget,
-    'ROLE' as owner_role_type,
-    NULL as resource_constraint;
-"""
-
-
-def show_warehouses(expression: exp.Expression) -> exp.Expression:
-    """Transform SHOW WAREHOUSES.
-
-    See https://docs.snowflake.com/en/sql-reference/sql/show-warehouses
-    """
-    if (
-        isinstance(expression, exp.Show)
-        and isinstance(expression.this, str)
-        and expression.this.upper() == "WAREHOUSES"
-    ):
-        return sqlglot.parse_one(SQL_SHOW_WAREHOUSES, read="duckdb")
+        return sqlglot.parse_one("SELECT * FROM _fs_global._fs_information_schema._fs_show_functions", read="duckdb")
 
     return expression
 
 
 def show_keys(
     expression: exp.Expression,
-    current_database: str | None = None,
+    current_database: str | None,
     *,
     kind: Literal["PRIMARY", "UNIQUE", "FOREIGN"],
 ) -> exp.Expression:
@@ -296,15 +270,138 @@ def show_keys(
     return expression
 
 
+SQL_CREATE_VIEW_SHOW_PROCEDURES = """
+create view if not exists _fs_global._fs_information_schema._fs_show_procedures as
+SELECT
+    '2012-08-01 07:00:00 UTC'::timestamptz as 'created_on',
+    'SYSTEM$CLASSIFY' as 'name',
+    '' as 'schema_name',
+    'Y' as 'is_builtin',
+    'N' as 'is_aggregate',
+    'N' as 'is_ansi',
+    2 as 'min_num_arguments',
+    2 as 'max_num_arguments',
+    'SYSTEM$CLASSIFY(VARCHAR, OBJECT) RETURN OBJECT' as 'arguments',
+    'classify stored proc' as 'description',
+    '' as 'catalog_name',
+    'N' as 'is_table_function',
+    'N' as 'valid_for_clustering',
+    NULL as 'is_secure',
+    '' as 'secrets',
+    '' as 'external_access_integrations',
+WHERE 0 = 1;
+"""
+
+
+def show_procedures(expression: exp.Expression) -> exp.Expression:
+    """Transform SHOW PROCEDURES.
+
+    See https://docs.snowflake.com/en/sql-reference/sql/show-procedures
+    """
+    if (
+        isinstance(expression, exp.Show)
+        and isinstance(expression.this, str)
+        and expression.this.upper() == "PROCEDURES"
+    ):
+        return sqlglot.parse_one(
+            "SELECT * FROM _fs_global._fs_information_schema._fs_show_procedures",
+            read="duckdb",
+        )
+
+    return expression
+
+
+SQL_CREATE_VIEW_SHOW_SCHEMAS = """
+create view if not exists _fs_global._fs_information_schema._fs_show_schemas as
+select
+    to_timestamp(0)::timestamptz as 'created_on',
+    case
+        when schema_name = '_fs_information_schema' then 'information_schema'
+        else schema_name
+    end as 'name',
+    NULL as 'kind',
+    catalog_name as 'database_name',
+    NULL as 'schema_name'
+from information_schema.schemata
+where not catalog_name in ('memory', 'system', 'temp', '_fs_global')
+  and not schema_name in ('main', 'pg_catalog')
+"""
+
+
+def show_schemas(expression: exp.Expression, current_database: str | None) -> exp.Expression:
+    """Transform SHOW SCHEMAS to a query against the _fs_show_schemas view.
+
+    See https://docs.snowflake.com/en/sql-reference/sql/show-schemas
+    """
+    if isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "SCHEMAS":
+        if (ident := expression.find(exp.Identifier)) and isinstance(ident.this, str):
+            database = ident.this
+        else:
+            database = current_database
+
+        query = "SELECT * FROM _fs_global._fs_information_schema._fs_show_schemas"
+
+        if database:
+            query += f" WHERE database_name = '{database}'"
+        return sqlglot.parse_one(query, read="duckdb")
+
+    return expression
+
+
+def show_stages(expression: exp.Expression, current_database: str | None, current_schema: str | None) -> exp.Expression:
+    """Transform SHOW STAGES to a select from the fake _fs_stages table."""
+    if not (
+        isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "STAGES"
+    ):
+        return expression
+
+    scope_kind = expression.args.get("scope_kind")
+    table = expression.find(exp.Table)
+
+    if scope_kind == "DATABASE":
+        catalog = (table and table.name) or current_database
+        schema = None
+    elif scope_kind == "SCHEMA" and table:
+        catalog = table.db or current_database
+        schema = table.name
+    elif scope_kind == "TABLE" and table:
+        catalog = table.db or current_database
+        assert isinstance(table.this, exp.Identifier)
+        schema = table.this.this
+    elif scope_kind == "ACCOUNT":
+        # show all objects / tables in the account
+        catalog = None
+        schema = None
+    else:
+        # no explicit scope - show current database and schema only
+        catalog = current_database
+        schema = current_schema
+
+    where = ["1=1"]
+    if catalog:
+        where.append(f"database_name = '{catalog}'")
+    if schema:
+        where.append(f"schema_name = '{schema}'")
+    where_clause = " AND ".join(where)
+
+    query = f"""
+        SELECT *
+        from _fs_global._fs_information_schema._fs_stages
+        where {where_clause}
+    """
+
+    return sqlglot.parse_one(query, read="duckdb")
+
+
 # see https://docs.snowflake.com/en/sql-reference/sql/show-objects
 SQL_CREATE_VIEW_SHOW_OBJECTS = """
 create view if not exists _fs_global._fs_information_schema._fs_show_objects as
 select
     to_timestamp(0)::timestamptz as created_on,
     table_name as name,
-    case when table_type='BASE TABLE' then 'TABLE' else table_type end as 'kind',
     table_catalog as database_name,
     table_schema as schema_name,
+    case when table_type='BASE TABLE' then 'TABLE' else table_type end as 'kind',
     '' as comment,
     '' as cluster_by,
     -- TODO: implement rows and bytes as rows * 1024
@@ -313,9 +410,9 @@ select
     'SYSADMIN' as owner,
     1 as retention_time,
     'ROLE' as owner_role_type,
-    null as budget,
     'N' as is_hybrid,
-    'N' as is_dynamic
+    'N' as is_dynamic,
+    'N' as is_iceberg
 from information_schema.tables
 where not (table_schema == '_fs_information_schema')
 """
@@ -377,7 +474,9 @@ where not table_catalog in ('system')
 """
 
 
-def show_tables_etc(expression: exp.Expression, current_database: str | None = None) -> exp.Expression:
+def show_tables_etc(
+    expression: exp.Expression, current_database: str | None, current_schema: str | None
+) -> exp.Expression:
     """Transform SHOW OBJECTS/TABLES/VIEWS to a query against the _fs_information_schema views."""
     if not (
         isinstance(expression, exp.Show)
@@ -396,10 +495,14 @@ def show_tables_etc(expression: exp.Expression, current_database: str | None = N
     elif scope_kind == "SCHEMA" and table:
         catalog = table.db or current_database
         schema = table.name
-    else:
-        # all objects / tables - will show everything in the "account"
+    elif scope_kind == "ACCOUNT":
+        # show all objects / tables in the account
         catalog = None
         schema = None
+    else:
+        # no explicit scope - show current database and schema only
+        catalog = current_database
+        schema = current_schema
 
     if expression.args["terse"] and show == "VIEWS":
         columns = ["created_on, name, 'VIEW' as kind, database_name, schema_name"]
@@ -430,84 +533,71 @@ def show_tables_etc(expression: exp.Expression, current_database: str | None = N
     return sqlglot.parse_one(query, read="duckdb")
 
 
-# returns zero rows
-SQL_SHOW_PROCEDURES = """
-SELECT
-    '2012-08-01 07:00:00 UTC'::timestamptz as 'created_on',
-    'SYSTEM$CLASSIFY' as 'name',
-    '' as 'schema_name',
-    'Y' as 'is_builtin',
-    'N' as 'is_aggregate',
-    'N' as 'is_ansi',
-    2 as 'min_num_arguments',
-    2 as 'max_num_arguments',
-    'SYSTEM$CLASSIFY(VARCHAR, OBJECT) RETURN OBJECT' as 'arguments',
-    'classify stored proc' as 'description',
-    '' as 'catalog_name',
-    'N' as 'is_table_function',
-    'N' as 'valid_for_clustering',
-    NULL as 'is_secure',
-    '' as 'secrets',
-    '' as 'external_access_integrations',
-WHERE 0 = 1;
-"""
-
-
-def show_procedures(expression: exp.Expression) -> exp.Expression:
-    """Transform SHOW PROCEDURES.
+def show_users(expression: exp.Expression) -> exp.Expression:
+    """Transform SHOW USERS to a query against the global database's information_schema._fs_users table.
 
-    See https://docs.snowflake.com/en/sql-reference/sql/show-procedures
+    https://docs.snowflake.com/en/sql-reference/sql/show-users
     """
-    if (
-        isinstance(expression, exp.Show)
-        and isinstance(expression.this, str)
-        and expression.this.upper() == "PROCEDURES"
-    ):
-        return sqlglot.parse_one(SQL_SHOW_PROCEDURES, read="duckdb")
+    if isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "USERS":
+        return sqlglot.parse_one("SELECT * FROM _fs_global._fs_information_schema._fs_users", read="duckdb")
 
     return expression
 
 
-SQL_SHOW_SCHEMAS = """
-select
-    to_timestamp(0)::timestamptz as 'created_on',
-    case
-        when schema_name = '_fs_information_schema' then 'information_schema'
-        else schema_name
-    end as 'name',
-    NULL as 'kind',
-    catalog_name as 'database_name',
-    NULL as 'schema_name'
-from information_schema.schemata
-where not catalog_name in ('memory', 'system', 'temp', '_fs_global')
-  and not schema_name in ('main', 'pg_catalog')
-"""
-
-
-def show_schemas(expression: exp.Expression, current_database: str | None = None) -> exp.Expression:
-    """Transform SHOW SCHEMAS to a query against the information_schema.schemata table.
-
-    See https://docs.snowflake.com/en/sql-reference/sql/show-schemas
-    """
-    if isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "SCHEMAS":
-        if (ident := expression.find(exp.Identifier)) and isinstance(ident.this, str):
-            database = ident.this
-        else:
-            database = current_database
+# returns zero rows
+SQL_SHOW_WAREHOUSES = """
+SELECT
+    'FAKESNOW_WAREHOUSE' as name,
+    'STARTED' as state,
+    'STANDARD' as type,
+    'X-Small' as size,
+    1 as min_cluster_count,
+    1 as max_cluster_count,
+    1 as started_clusters,
+    0 as running,
+    0 as queued,
+    'N' as is_default,
+    'N' as is_current,
+    600 as auto_suspend,
+    'true' as auto_resume,
+    -- nb: deliberate space before '100' to match Snowflake's output
+    ' 100' as available,
+    '0' as provisioning,
+    '0' as quiescing,
+    '0' as other,
+    '1970-01-01 00:00:00.000000 UTC'::timestamptz as created_on,
+    '1970-01-01 00:00:00.000000 UTC'::timestamptz as resumed_on,
+    '1970-01-01 00:00:00.000000 UTC'::timestamptz as updated_on,
+    'SYSADMIN' as owner,
+    '' as comment,
+    'false' as enable_query_acceleration,
+    8 as query_acceleration_max_scale_factor,
+    'null' as resource_monitor,
 
-        return sqlglot.parse_one(
-            f"{SQL_SHOW_SCHEMAS} and catalog_name = '{database}'" if database else SQL_SHOW_SCHEMAS, read="duckdb"
-        )
+    -- deprecated - these 5 cols are for internal use
+    0 as actives,
+    0 as pendings,
+    0 as failed,
+    0 as suspended,
+    '123456789012' as uuid,
 
-    return expression
+    'STANDARD' as scaling_policy,
+    NULL as budget,
+    'ROLE' as owner_role_type,
+    NULL as resource_constraint;
+"""
 
 
-def show_users(expression: exp.Expression) -> exp.Expression:
-    """Transform SHOW USERS to a query against the global database's information_schema._fs_users table.
+def show_warehouses(expression: exp.Expression) -> exp.Expression:
+    """Transform SHOW WAREHOUSES.
 
-    https://docs.snowflake.com/en/sql-reference/sql/show-users
+    See https://docs.snowflake.com/en/sql-reference/sql/show-warehouses
     """
-    if isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "USERS":
-        return sqlglot.parse_one("SELECT * FROM _fs_global._fs_information_schema._fs_users", read="duckdb")
+    if (
+        isinstance(expression, exp.Show)
+        and isinstance(expression.this, str)
+        and expression.this.upper() == "WAREHOUSES"
+    ):
+        return sqlglot.parse_one(SQL_SHOW_WAREHOUSES, read="duckdb")
 
     return expression