fakesnow 0.9.0__tar.gz → 0.9.2__tar.gz

{fakesnow-0.9.0/fakesnow.egg-info → fakesnow-0.9.2}/PKG-INFO

@@ -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 → fakesnow-0.9.2}/README.md

@@ -119,6 +119,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 → fakesnow-0.9.2}/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-0.9.2/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-0.9.0 → fakesnow-0.9.2}/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-0.9.0 → fakesnow-0.9.2}/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 → fakesnow-0.9.2/fakesnow.egg-info}/PKG-INFO

@@ -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 → fakesnow-0.9.2}/fakesnow.egg-info/SOURCES.txt

@@ -9,6 +9,7 @@ fakesnow/cli.py
 fakesnow/expr.py
 fakesnow/fakes.py
 fakesnow/fixtures.py
+fakesnow/global_database.py
 fakesnow/info_schema.py
 fakesnow/macros.py
 fakesnow/py.typed
@@ -24,4 +25,5 @@ tests/test_cli.py
 tests/test_expr.py
 tests/test_fakes.py
 tests/test_patch.py
-tests/test_transforms.py
+tests/test_transforms.py
+tests/test_users.py

{fakesnow-0.9.0 → fakesnow-0.9.2}/fakesnow.egg-info/requires.txt

@@ -1,7 +1,7 @@
-duckdb~=0.9.2
+duckdb~=0.10.0
 pyarrow
 snowflake-connector-python
-sqlglot~=21.0.1
+sqlglot~=21.1.0
 
 [dev]
 black~=23.9

{fakesnow-0.9.0 → fakesnow-0.9.2}/pyproject.toml

@@ -1,17 +1,17 @@
 [project]
 name = "fakesnow"
 description = "Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally."
-version = "0.9.0"
+version = "0.9.2"
 readme = "README.md"
 license = { file = "LICENSE" }
 classifiers = ["License :: OSI Approved :: MIT License"]
 keywords = ["snowflake", "snowflakedb", "fake", "local", "mock", "testing"]
 requires-python = ">=3.9"
 dependencies = [
-    "duckdb~=0.9.2",
+    "duckdb~=0.10.0",
     "pyarrow",
     "snowflake-connector-python",
-    "sqlglot~=21.0.1",
+    "sqlglot~=21.1.0",
 ]
 
 [project.urls]

{fakesnow-0.9.0 → fakesnow-0.9.2}/tests/test_fakes.py

@@ -459,6 +459,19 @@ def test_description_update(dcur: snowflake.connector.cursor.DictCursor):
     # fmt: on
 
 
+def test_description_delete(dcur: snowflake.connector.cursor.DictCursor):
+    dcur.execute("create table example (x int)")
+    dcur.execute("insert into example values (1), (2), (3)")
+    dcur.execute("delete from example where x>1")
+    assert dcur.fetchall() == [{"number of rows deleted": 2}]
+    # TODO: Snowflake is actually precision=19, is_nullable=False
+    # fmt: off
+    assert dcur.description == [
+        ResultMetadata(name='number of rows deleted', type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True),
+    ]
+    # fmt: on
+
+
 def test_equal_null(cur: snowflake.connector.cursor.SnowflakeCursor):
     cur.execute("select equal_null(NULL, NULL), equal_null(1, 1), equal_null(1, 2), equal_null(1, NULL)")
     assert cur.fetchall() == [(True, True, False, False)]

{fakesnow-0.9.0 → fakesnow-0.9.2}/tests/test_transforms.py

@@ -140,7 +140,7 @@ def test_flatten() -> None:
         )
         .transform(flatten)
         .sql(dialect="duckdb")
-        == """SELECT t.id, flat.value -> '$.fruit' FROM (SELECT 1, JSON('[{"fruit":"banana"}]') UNION SELECT 2, JSON('[{"fruit":"coconut"}, {"fruit":"durian"}]')) AS t(id, fruits), LATERAL UNNEST(LIST_REVERSE(CAST(t.fruits AS JSON[]))) AS flat(VALUE)"""  # noqa: E501
+        == """SELECT t.id, flat.value -> '$.fruit' FROM (SELECT 1, JSON('[{"fruit":"banana"}]') UNION SELECT 2, JSON('[{"fruit":"coconut"}, {"fruit":"durian"}]')) AS t(id, fruits), LATERAL UNNEST(CAST(t.fruits AS JSON[])) AS flat(VALUE)"""  # noqa: E501
     )
 
 

fakesnow-0.9.2/tests/test_users.py

@@ -0,0 +1,19 @@
+import snowflake.connector.cursor
+
+
+def test_show_users_base_case(cur: snowflake.connector.cursor.SnowflakeCursor):
+    result = cur.execute("SHOW USERS")
+    assert result
+    assert result.fetchall() == []
+
+
+def test_show_users_with_users(cur: snowflake.connector.cursor.SnowflakeCursor):
+    result = cur.execute("CREATE USER foo")
+    result = cur.execute("CREATE USER bar")
+
+    result = cur.execute("SHOW USERS")
+    assert result
+
+    rows = result.fetchall()
+    names = [row[0] for row in rows]
+    assert names == ["foo", "bar"]