fakesnow 0.9.0__py3-none-any.whl → 0.9.2__py3-none-any.whl

fakesnow/fakes.py

@@ -30,6 +30,7 @@ import fakesnow.expr as expr
 import fakesnow.info_schema as info_schema
 import fakesnow.macros as macros
 import fakesnow.transforms as transforms
+from fakesnow.global_database import create_global_database
 
 SCHEMA_UNSET = "schema_unset"
 SQL_SUCCESS = "SELECT 'Statement executed successfully.' as 'status'"
@@ -39,6 +40,7 @@ SQL_CREATED_TABLE = Template("SELECT 'Table ${name} successfully created.' as 's
 SQL_DROPPED = Template("SELECT '${name} successfully dropped.' as 'status'")
 SQL_INSERTED_ROWS = Template("SELECT ${count} as 'number of rows inserted'")
 SQL_UPDATED_ROWS = Template("SELECT ${count} as 'number of rows updated', 0 as 'number of multi-joined rows updated'")
+SQL_DELETED_ROWS = Template("SELECT ${count} as 'number of rows deleted'")
 
 
 class FakeSnowflakeCursor:
@@ -194,6 +196,8 @@ class FakeSnowflakeCursor:
             .transform(transforms.identifier)
             .transform(lambda e: transforms.show_schemas(e, self._conn.database))
             .transform(lambda e: transforms.show_objects_tables(e, self._conn.database))
+            .transform(transforms.show_users)
+            .transform(transforms.create_user)
         )
         sql = transformed.sql(dialect="duckdb")
         result_sql = None
@@ -265,6 +269,10 @@ class FakeSnowflakeCursor:
             (affected_count,) = self._duck_conn.fetchall()[0]
             result_sql = SQL_UPDATED_ROWS.substitute(count=affected_count)
 
+        elif cmd == "DELETE":
+            (affected_count,) = self._duck_conn.fetchall()[0]
+            result_sql = SQL_DELETED_ROWS.substitute(count=affected_count)
+
         elif cmd == "DESCRIBE TABLE":
             # DESCRIBE TABLE has already been run above to detect and error if the table exists
             # We now rerun DESCRIBE TABLE but transformed with columns to match Snowflake
@@ -480,6 +488,8 @@ class FakeSnowflakeConnection:
         self.db_path = db_path
         self._paramstyle = "pyformat"
 
+        create_global_database(duck_conn)
+
         # create database if needed
         if (
             create_database

fakesnow/global_database.py

@@ -0,0 +1,46 @@
+from duckdb import DuckDBPyConnection
+
+GLOBAL_DATABASE_NAME = "_fs_global"
+USERS_TABLE_FQ_NAME = f"{GLOBAL_DATABASE_NAME}._fs_users_ext"
+
+# replicates the output structure of https://docs.snowflake.com/en/sql-reference/sql/show-users
+SQL_CREATE_INFORMATION_SCHEMA_USERS_TABLE_EXT = f"""
+create table if not exists {USERS_TABLE_FQ_NAME} (
+    name varchar,
+    created_on TIMESTAMPTZ,
+    login_name varchar,
+    display_name varchar,
+    first_name varchar,
+    last_name varchar,
+    email varchar,
+    mins_to_unlock varchar,
+    days_to_expiry varchar,
+    comment varchar,
+    disabled varchar,
+    must_change_password varchar,
+    snowflake_lock varchar,
+    default_warehouse varchar,
+    default_namespace varchar,
+    default_role varchar,
+    default_secondary_roles varchar,
+    ext_authn_duo varchar,
+    ext_authn_uid varchar,
+    mins_to_bypass_mfa varchar,
+    owner varchar,
+    last_success_login TIMESTAMPTZ,
+    expires_at_time TIMESTAMPTZ,
+    locked_until_time TIMESTAMPTZ,
+    has_password varchar,
+    has_rsa_public_key varchar,
+)
+"""
+
+
+def create_global_database(conn: DuckDBPyConnection) -> None:
+    """Create a "global" database for storing objects which span database.
+
+    Including (but not limited to):
+     - Users
+    """
+    conn.execute(f"ATTACH IF NOT EXISTS ':memory:' AS {GLOBAL_DATABASE_NAME}")
+    conn.execute(SQL_CREATE_INFORMATION_SCHEMA_USERS_TABLE_EXT)

fakesnow/info_schema.py

@@ -74,7 +74,8 @@ select
     1 as retention_time,
     'STANDARD' as type
 from information_schema.schemata
-where catalog_name not in ('memory', 'system', 'temp') and schema_name = 'information_schema'
+where catalog_name not in ('memory', 'system', 'temp', '_fs_global')
+  and schema_name = 'information_schema'
 """
 )
 

fakesnow/transforms.py

@@ -7,6 +7,8 @@ from typing import cast
 import sqlglot
 from sqlglot import exp
 
+from fakesnow.global_database import USERS_TABLE_FQ_NAME
+
 MISSING_DATABASE = "missing_database"
 SUCCESS_NOP = sqlglot.parse_one("SELECT 'Statement executed successfully.'")
 
@@ -271,22 +273,15 @@ def flatten(expression: exp.Expression) -> exp.Expression:
         return exp.Lateral(
             this=exp.Unnest(
                 expressions=[
-                    exp.Anonymous(
-                        # duckdb unnests in reserve, so we reverse the list to match
-                        # the order of the original array (and snowflake)
-                        this="list_reverse",
-                        expressions=[
-                            exp.Cast(
-                                this=explode_expression,
-                                to=exp.DataType(
-                                    this=exp.DataType.Type.ARRAY,
-                                    expressions=[exp.DataType(this=exp.DataType.Type.JSON, nested=False, prefix=False)],
-                                    nested=True,
-                                ),
-                            )
-                        ],
+                    exp.Cast(
+                        this=explode_expression,
+                        to=exp.DataType(
+                            this=exp.DataType.Type.ARRAY,
+                            expressions=[exp.DataType(this=exp.DataType.Type.JSON, nested=False, prefix=False)],
+                            nested=True,
+                        ),
                     )
-                ]
+                ],
             ),
             alias=exp.TableAlias(this=alias.this, columns=[exp.Identifier(this="VALUE", quoted=False)]),
         )
@@ -963,3 +958,32 @@ def values_columns(expression: exp.Expression) -> exp.Expression:
         expression.set("alias", exp.TableAlias(this=exp.Identifier(this="_", quoted=False), columns=columns))
 
     return expression
+
+
+def show_users(expression: exp.Expression) -> exp.Expression:
+    """Transform SHOW USERS to a query against the global database's information_schema._fs_users table.
+
+    https://docs.snowflake.com/en/sql-reference/sql/show-users
+    """
+    if isinstance(expression, exp.Show) and isinstance(expression.this, str) and expression.this.upper() == "USERS":
+        return sqlglot.parse_one(f"SELECT * FROM {USERS_TABLE_FQ_NAME}", read="duckdb")
+
+    return expression
+
+
+def create_user(expression: exp.Expression) -> exp.Expression:
+    """Transform CREATE USER to a query against the global database's information_schema._fs_users table.
+
+    https://docs.snowflake.com/en/sql-reference/sql/create-user
+    """
+    # XXX: this is a placeholder. We need to implement the full CREATE USER syntax, but
+    #      sqlglot doesnt yet support Create for snowflake.
+    if isinstance(expression, exp.Command) and expression.this == "CREATE":
+        sub_exp = expression.expression.strip()
+        if sub_exp.upper().startswith("USER"):
+            _, name, *ignored = sub_exp.split(" ")
+            if ignored:
+                raise NotImplementedError(f"`CREATE USER` with {ignored} not yet supported")
+            return sqlglot.parse_one(f"INSERT INTO {USERS_TABLE_FQ_NAME} (name) VALUES ('{name}')", read="duckdb")
+
+    return expression

{fakesnow-0.9.0.dist-info → fakesnow-0.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.0
+Version: 0.9.2
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License: Apache License
                                    Version 2.0, January 2004
@@ -210,10 +210,10 @@ Classifier: License :: OSI Approved :: MIT License
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: duckdb ~=0.9.2
+Requires-Dist: duckdb ~=0.10.0
 Requires-Dist: pyarrow
 Requires-Dist: snowflake-connector-python
-Requires-Dist: sqlglot ~=21.0.1
+Requires-Dist: sqlglot ~=21.1.0
 Provides-Extra: dev
 Requires-Dist: black ~=23.9 ; extra == 'dev'
 Requires-Dist: build ~=1.0 ; extra == 'dev'
@@ -349,6 +349,7 @@ Partial support
 - [x] regex functions
 - [x] semi-structured data
 - [x] tags
+- [x] user management (See [tests/test_users.py](tests/test_users.py))
 
 For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 

{fakesnow-0.9.0.dist-info → fakesnow-0.9.2.dist-info}/RECORD

@@ -3,15 +3,16 @@ fakesnow/__main__.py,sha256=GDrGyNTvBFuqn_UfDjKs7b3LPtU6gDv1KwosVDrukIM,76
 fakesnow/checks.py,sha256=-QMvdcrRbhN60rnzxLBJ0IkUBWyLR8gGGKKmCS0w9mA,2383
 fakesnow/cli.py,sha256=9qfI-Ssr6mo8UmIlXkUAOz2z2YPBgDsrEVaZv9FjGFs,2201
 fakesnow/expr.py,sha256=CAxuYIUkwI339DQIBzvFF0F-m1tcVGKEPA5rDTzmH9A,892
-fakesnow/fakes.py,sha256=aRVnMMtqe2GM2YFEmoj8JEDEofb_iv6BrmVBf5zxiA0,27492
+fakesnow/fakes.py,sha256=iF9IN7nG74DNMDNTZ9CBUGqsYsqicZ7PgRNe4B_zAFs,27931
 fakesnow/fixtures.py,sha256=G-NkVeruSQAJ7fvSS2fR2oysUn0Yra1pohHlOvacKEk,455
-fakesnow/info_schema.py,sha256=TyFR4YfZkJhoPXqgbZr0J2xDo7mK9WMpbn-O1XHVhl4,4685
+fakesnow/global_database.py,sha256=WTVIP1VhNvdCeX7TQncX1TRpGQU5rBf5Pbxim40zeSU,1399
+fakesnow/info_schema.py,sha256=eiKEVBbaSGfsuPD8USuhHF9BDC8pU9sPiDHgy0orTN8,4701
 fakesnow/macros.py,sha256=pX1YJDnQOkFJSHYUjQ6ErEkYIKvFI6Ncz_au0vv1csA,265
 fakesnow/py.typed,sha256=B-DLSjYBi7pkKjwxCSdpVj2J02wgfJr-E7B1wOUyxYU,80
-fakesnow/transforms.py,sha256=8-QKw47_TwCamSai1kndiB-JhrKjlYcWP4y8-6X_m0U,35293
-fakesnow-0.9.0.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
-fakesnow-0.9.0.dist-info/METADATA,sha256=V9B-3KdQ5AmjluhsGnDyPzn0IxNHYQ3t972eIQIYV9M,17652
-fakesnow-0.9.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-fakesnow-0.9.0.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
-fakesnow-0.9.0.dist-info/top_level.txt,sha256=500evXI1IFX9so82cizGIEMHAb_dJNPaZvd2H9dcKTA,24
-fakesnow-0.9.0.dist-info/RECORD,,
+fakesnow/transforms.py,sha256=z9tZDOtvQDAoYmQDvDegUOwxgBFvjnSfKCo7kTc-ryA,36300
+fakesnow-0.9.2.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
+fakesnow-0.9.2.dist-info/METADATA,sha256=7rSjgiC1nVjW_38DFx5I6OfyFz71f_TbjapFAIZ-kkI,17724
+fakesnow-0.9.2.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+fakesnow-0.9.2.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
+fakesnow-0.9.2.dist-info/top_level.txt,sha256=500evXI1IFX9so82cizGIEMHAb_dJNPaZvd2H9dcKTA,24
+fakesnow-0.9.2.dist-info/RECORD,,