analytics-toolkit 1.3.8.2__tar.gz → 1.3.8.3__tar.gz

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: analytics-toolkit
-Version: 1.3.8.2
+Version: 1.3.8.3
 Summary: Shared Python utilities for SQL, Excel, and date helpers.
 Author: analytics_toolkit contributors
 License: MIT License
@@ -75,7 +75,7 @@ Python toolkit for AB-test analysis, SQL workflows, Excel reports, and date help
 
 [Documentation Overview](https://github.com/Karapsin/analytics_toolkit/blob/main/docs/README.md)
 
-**Version:** `1.3.8.2`<br>
+**Version:** `1.3.8.3`<br>
 **Depends:** Python (`>=3.8,<3.15`)<br>
 **Imports:** [clickhouse-connect](https://pypi.org/project/clickhouse-connect/) (`>=0.5.14,<1`), [lz4](https://pypi.org/project/lz4/) (`>=4.3.2,<5`), [numpy](https://pypi.org/project/numpy/) (`>=1.24.2,<2`), [openpyxl](https://pypi.org/project/openpyxl/) (`>=3.1.1,<4`), [orjson](https://pypi.org/project/orjson/) (`>=3.8.7,<4`), [pandas](https://pypi.org/project/pandas/) (`>=1.4.4,<3`), [psycopg2-binary](https://pypi.org/project/psycopg2-binary/) (`>=2.9.5,<3`), [python-dateutil](https://pypi.org/project/python-dateutil/) (`>=2.8.2,<3`), [pytz](https://pypi.org/project/pytz/) (`>=2022.7`), [requests](https://pypi.org/project/requests/) (`>=2.28.2,<3`), [scipy](https://pypi.org/project/scipy/) (`>=1.10.1,<2`), [sqlglot](https://pypi.org/project/sqlglot/) (`>=20,<31`), [sqlparse](https://pypi.org/project/sqlparse/) (`>=0.4.3,<1`), [tqdm](https://pypi.org/project/tqdm/) (`>=4.65.0,<5`), [trino](https://pypi.org/project/trino/) (`>=0.320,<1`), [zstandard](https://pypi.org/project/zstandard/) (`>=0.20.0,<1`)<br>
 **Suggests:** [apache-airflow](https://pypi.org/project/apache-airflow/) (`>=2.4,<3`; optional extra `airflow`)<br>

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/README.md

@@ -6,7 +6,7 @@ Python toolkit for AB-test analysis, SQL workflows, Excel reports, and date help
 
 [Documentation Overview](https://github.com/Karapsin/analytics_toolkit/blob/main/docs/README.md)
 
-**Version:** `1.3.8.2`<br>
+**Version:** `1.3.8.3`<br>
 **Depends:** Python (`>=3.8,<3.15`)<br>
 **Imports:** [clickhouse-connect](https://pypi.org/project/clickhouse-connect/) (`>=0.5.14,<1`), [lz4](https://pypi.org/project/lz4/) (`>=4.3.2,<5`), [numpy](https://pypi.org/project/numpy/) (`>=1.24.2,<2`), [openpyxl](https://pypi.org/project/openpyxl/) (`>=3.1.1,<4`), [orjson](https://pypi.org/project/orjson/) (`>=3.8.7,<4`), [pandas](https://pypi.org/project/pandas/) (`>=1.4.4,<3`), [psycopg2-binary](https://pypi.org/project/psycopg2-binary/) (`>=2.9.5,<3`), [python-dateutil](https://pypi.org/project/python-dateutil/) (`>=2.8.2,<3`), [pytz](https://pypi.org/project/pytz/) (`>=2022.7`), [requests](https://pypi.org/project/requests/) (`>=2.28.2,<3`), [scipy](https://pypi.org/project/scipy/) (`>=1.10.1,<2`), [sqlglot](https://pypi.org/project/sqlglot/) (`>=20,<31`), [sqlparse](https://pypi.org/project/sqlparse/) (`>=0.4.3,<1`), [tqdm](https://pypi.org/project/tqdm/) (`>=4.65.0,<5`), [trino](https://pypi.org/project/trino/) (`>=0.320,<1`), [zstandard](https://pypi.org/project/zstandard/) (`>=0.20.0,<1`)<br>
 **Suggests:** [apache-airflow](https://pypi.org/project/apache-airflow/) (`>=2.4,<3`; optional extra `airflow`)<br>

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/analytics_toolkit/sql/__init__.py

@@ -32,7 +32,7 @@ from .dml.io.read_sql import read_sql as read
 from .dml.io.cancel_queries import cancel_queries
 from .dml.table import (
     drop_tables,
-    drop_paritions,
+    drop_partitions,
     gp_create_partitions,
     gp_vacuum,
 )
@@ -52,7 +52,7 @@ _TIMED_PUBLIC_SQL_FUNCTION_NAMES = (
     "execute_read",
     "execute",
     "drop_tables",
-    "drop_paritions",
+    "drop_partitions",
     "gp_create_partitions",
     "gp_vacuum",
     "load_df",
@@ -81,7 +81,7 @@ __all__ = [
     "generate_dummy_connections",
     "create_sql_table",
     "drop_tables",
-    "drop_paritions",
+    "drop_partitions",
     "gp_create_partitions",
     "gp_vacuum",
     "load_df",

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/analytics_toolkit/sql/dml/__init__.py

@@ -3,7 +3,7 @@ from .io.execute_sql import execute_sql
 from .io.read_sql import read_sql
 from .load.load_df import build_load_df_plan, load_df
 from .table import (
-    drop_paritions,
+    drop_partitions,
     gp_create_partitions,
     gp_vacuum,
 )
@@ -13,7 +13,7 @@ __all__ = [
     "build_load_df_plan",
     "build_transfer_table_plan",
     "cancel_queries",
-    "drop_paritions",
+    "drop_partitions",
     "execute_sql",
     "gp_create_partitions",
     "gp_vacuum",

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/analytics_toolkit/sql/dml/table/__init__.py

@@ -35,7 +35,7 @@ from .table_validation import (
     validate_stage_target_key_overlap,
     validate_stage_uniqueness,
 )
-from .partitions import drop_paritions
+from .partitions import drop_partitions
 
 __all__ = [
     "analyze_table",
@@ -52,7 +52,7 @@ __all__ = [
     "build_insert_from_table_sql",
     "count_table_rows",
     "CreateTableFromSqlOptions",
-    "drop_paritions",
+    "drop_partitions",
     "DropManyPartitionsOptions",
     "drop_tables",
     "drop_table_with_retry",

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/analytics_toolkit/sql/dml/table/api.py

@@ -47,7 +47,7 @@ from .partitions import (
     build_drop_many_partitions_plan,
     build_drop_many_partitions_sqls,
     build_gp_create_partitions_plan,
-    drop_paritions,
+    drop_partitions,
     gp_create_partitions,
 )
 from .write_modes import (

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/analytics_toolkit/sql/dml/table/partitions.py

@@ -213,7 +213,7 @@ def gp_create_partitions(
     return None
 
 @timed_public_sql_function
-def drop_paritions(
+def drop_partitions(
     db_key: str,
     table: str,
     partition_keys_list: list[str],
@@ -255,7 +255,7 @@ def drop_paritions(
     def operation(connection_ref: dict[str, Any], attempt: int) -> None:
         with tracked_sql_operation(
             metadata=metadata,
-            operation_name="drop_paritions",
+            operation_name="drop_partitions",
             alias=options.connection_key,
             backend=options.backend,
             phase="drop_partitions",
@@ -271,7 +271,7 @@ def drop_paritions(
 
     def context(attempt: int) -> SqlOperationContext:
         return SqlOperationContext(
-            operation="drop_paritions",
+            operation="drop_partitions",
             alias=options.connection_key,
             backend=options.backend,
             phase="drop_partitions",
@@ -413,7 +413,7 @@ def build_drop_many_partitions_plan(
         query_label=options.query_label,
     )
     plan = SqlPlan(
-        operation="drop_paritions",
+        operation="drop_partitions",
         target_alias=options.connection_key,
         target_backend=options.backend,
         target_table=options.target_table,

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/analytics_toolkit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: analytics-toolkit
-Version: 1.3.8.2
+Version: 1.3.8.3
 Summary: Shared Python utilities for SQL, Excel, and date helpers.
 Author: analytics_toolkit contributors
 License: MIT License
@@ -75,7 +75,7 @@ Python toolkit for AB-test analysis, SQL workflows, Excel reports, and date help
 
 [Documentation Overview](https://github.com/Karapsin/analytics_toolkit/blob/main/docs/README.md)
 
-**Version:** `1.3.8.2`<br>
+**Version:** `1.3.8.3`<br>
 **Depends:** Python (`>=3.8,<3.15`)<br>
 **Imports:** [clickhouse-connect](https://pypi.org/project/clickhouse-connect/) (`>=0.5.14,<1`), [lz4](https://pypi.org/project/lz4/) (`>=4.3.2,<5`), [numpy](https://pypi.org/project/numpy/) (`>=1.24.2,<2`), [openpyxl](https://pypi.org/project/openpyxl/) (`>=3.1.1,<4`), [orjson](https://pypi.org/project/orjson/) (`>=3.8.7,<4`), [pandas](https://pypi.org/project/pandas/) (`>=1.4.4,<3`), [psycopg2-binary](https://pypi.org/project/psycopg2-binary/) (`>=2.9.5,<3`), [python-dateutil](https://pypi.org/project/python-dateutil/) (`>=2.8.2,<3`), [pytz](https://pypi.org/project/pytz/) (`>=2022.7`), [requests](https://pypi.org/project/requests/) (`>=2.28.2,<3`), [scipy](https://pypi.org/project/scipy/) (`>=1.10.1,<2`), [sqlglot](https://pypi.org/project/sqlglot/) (`>=20,<31`), [sqlparse](https://pypi.org/project/sqlparse/) (`>=0.4.3,<1`), [tqdm](https://pypi.org/project/tqdm/) (`>=4.65.0,<5`), [trino](https://pypi.org/project/trino/) (`>=0.320,<1`), [zstandard](https://pypi.org/project/zstandard/) (`>=0.20.0,<1`)<br>
 **Suggests:** [apache-airflow](https://pypi.org/project/apache-airflow/) (`>=2.4,<3`; optional extra `airflow`)<br>

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "analytics-toolkit"
-version = "1.3.8.2"
+version = "1.3.8.3"
 description = "Shared Python utilities for SQL, Excel, and date helpers."
 readme = "README.md"
 requires-python = ">=3.8,<3.15"

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/tests/test_sql_architecture.py

@@ -14,7 +14,7 @@ def test_sql_facade_is_the_supported_public_surface() -> None:
     public_names = {
         "cancel_queries",
         "create_sql_table",
-        "drop_paritions",
+        "drop_partitions",
         "drop_tables",
         "execute",
         "generate_dummy_connections",
@@ -38,6 +38,7 @@ def test_sql_facade_is_the_supported_public_surface() -> None:
         "ch_drop_table",
         "create_table_from_sql",
         "execute_sql",
+        "drop_paritions",
         "drop_many_partitions",
         "drop_table",
         "gp_cancel_all_running_queries",
@@ -83,7 +84,7 @@ def test_single_db_sql_public_operations_use_db_key() -> None:
     single_db_operations = {
         "create_sql_table",
         "cancel_queries",
-        "drop_paritions",
+        "drop_partitions",
         "drop_tables",
         "execute",
         "execute_read",

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/tests/test_sql_backend_adapters.py

@@ -58,7 +58,7 @@ def test_sql_public_api_exports_are_stable() -> None:
     public_names = {
         "async_sql",
         "create_sql_table",
-        "drop_paritions",
+        "drop_partitions",
         "drop_tables",
         "execute",
         "execute_read",
@@ -96,6 +96,8 @@ def test_sql_public_api_exports_are_stable() -> None:
     assert "read_sql" not in sql_module.__all__
     assert "drop_table" not in sql_module.__all__
     assert not hasattr(sql_module, "drop_table")
+    assert "drop_paritions" not in sql_module.__all__
+    assert not hasattr(sql_module, "drop_paritions")
     assert "drop_many_partitions" not in sql_module.__all__
     assert not hasattr(sql_module, "drop_many_partitions")
     assert "transfer_table" not in sql_module.__all__

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/tests/test_sql_drop_many_partitions.py

@@ -16,14 +16,18 @@ dml_table_module = importlib.import_module("analytics_toolkit.sql.dml.table")
 table_ops_module = importlib.import_module("analytics_toolkit.sql.dml.table.partitions")
 
 
-def test_drop_paritions_is_public_and_timed() -> None:
-    assert sql_module.drop_paritions is table_ops_module.drop_paritions
-    assert dml_table_module.drop_paritions is table_ops_module.drop_paritions
-    assert "drop_paritions" in sql_module.__all__
-    assert "drop_paritions" in dml_table_module.__all__
+def test_drop_partitions_is_public_and_timed() -> None:
+    assert sql_module.drop_partitions is table_ops_module.drop_partitions
+    assert dml_table_module.drop_partitions is table_ops_module.drop_partitions
+    assert "drop_partitions" in sql_module.__all__
+    assert "drop_partitions" in dml_table_module.__all__
     assert "build_drop_many_partitions_sqls" in dml_table_module.__all__
-    assert "drop_paritions" in sql_module._TIMED_PUBLIC_SQL_FUNCTION_NAMES
-    assert getattr(sql_module.drop_paritions, "__sql_public_timing__", False)
+    assert "drop_partitions" in sql_module._TIMED_PUBLIC_SQL_FUNCTION_NAMES
+    assert getattr(sql_module.drop_partitions, "__sql_public_timing__", False)
+    assert "drop_paritions" not in sql_module.__all__
+    assert not hasattr(sql_module, "drop_paritions")
+    assert "drop_paritions" not in dml_table_module.__all__
+    assert not hasattr(dml_table_module, "drop_paritions")
 
 
 def test_build_drop_many_partitions_sqls_renders_backend_sql() -> None:
@@ -67,7 +71,7 @@ def test_build_drop_many_partitions_sqls_renders_backend_sql() -> None:
     ]
 
 
-def test_drop_paritions_executes_greenplum_in_order_and_commits(
+def test_drop_partitions_executes_greenplum_in_order_and_commits(
     monkeypatch: pytest.MonkeyPatch,
 ) -> None:
     connection = FakeDbapiConnection()
@@ -77,7 +81,7 @@ def test_drop_paritions_executes_greenplum_in_order_and_commits(
         lambda key: connection,
     )
 
-    result = table_ops_module.drop_paritions(
+    result = table_ops_module.drop_partitions(
         "gp",
         "sandbox.events",
         ["2025-05-01", "2025-05-02"],
@@ -93,7 +97,7 @@ def test_drop_paritions_executes_greenplum_in_order_and_commits(
     assert connection.close_calls == 1
 
 
-def test_drop_paritions_executes_trino_delete_without_commit(
+def test_drop_partitions_executes_trino_delete_without_commit(
     monkeypatch: pytest.MonkeyPatch,
 ) -> None:
     connection = FakeDbapiConnection()
@@ -103,7 +107,7 @@ def test_drop_paritions_executes_trino_delete_without_commit(
         lambda key: connection,
     )
 
-    table_ops_module.drop_paritions(
+    table_ops_module.drop_partitions(
         "trino",
         "sandbox.events",
         ["2025-05-01", "2025-05-02"],
@@ -118,7 +122,7 @@ def test_drop_paritions_executes_trino_delete_without_commit(
     assert connection.close_calls == 1
 
 
-def test_drop_paritions_executes_clickhouse_shard_drops(
+def test_drop_partitions_executes_clickhouse_shard_drops(
     monkeypatch: pytest.MonkeyPatch,
 ) -> None:
     connection = FakeClickHouseClient()
@@ -128,7 +132,7 @@ def test_drop_paritions_executes_clickhouse_shard_drops(
         lambda key: connection,
     )
 
-    table_ops_module.drop_paritions(
+    table_ops_module.drop_partitions(
         "ch",
         "sandbox.events",
         ["2025-05-01", "2025-05-02"],
@@ -143,7 +147,7 @@ def test_drop_paritions_executes_clickhouse_shard_drops(
     assert connection.close_calls == 1
 
 
-def test_drop_paritions_dry_run_returns_plan_without_connection(
+def test_drop_partitions_dry_run_returns_plan_without_connection(
     monkeypatch: pytest.MonkeyPatch,
 ) -> None:
     monkeypatch.setattr(
@@ -152,21 +156,21 @@ def test_drop_paritions_dry_run_returns_plan_without_connection(
         lambda key: pytest.fail("connection should not be opened"),
     )
 
-    plan = table_ops_module.drop_paritions(
+    plan = table_ops_module.drop_partitions(
         "gp",
         "sandbox.events",
         ["2025-05-01"],
         dry_run=True,
     )
 
-    assert plan.operation == "drop_paritions"
+    assert plan.operation == "drop_partitions"
     assert plan.metadata.statement_count == 1
     assert plan.sqls == ["ALTER TABLE sandbox.events DROP PARTITION FOR ('2025-05-01')"]
 
 
-def test_drop_paritions_validates_required_inputs() -> None:
+def test_drop_partitions_validates_required_inputs() -> None:
     with pytest.raises(InvalidSqlInputError):
-        table_ops_module.drop_paritions(
+        table_ops_module.drop_partitions(
             "gp",
             "sandbox.events",
             [],
@@ -178,14 +182,14 @@ def test_drop_paritions_validates_required_inputs() -> None:
             [],
         )
     with pytest.raises(InvalidSqlInputError):
-        table_ops_module.drop_paritions(
+        table_ops_module.drop_partitions(
             "gp",
             "sandbox.events",
             ["2025-05-01"],
             trino_partition_column="dt",
         )
     with pytest.raises(UnsupportedConnectionTypeError):
-        table_ops_module.drop_paritions(
+        table_ops_module.drop_partitions(
             "unknown",
             "sandbox.events",
             ["2025-05-01"],

{analytics_toolkit-1.3.8.2 → analytics_toolkit-1.3.8.3}/tests/test_sql_improvements.py

@@ -179,7 +179,7 @@ def test_public_sql_facade_exports_refactored_helpers() -> None:
 @pytest.mark.parametrize(
     "function_name",
     [
-        "drop_paritions",
+        "drop_partitions",
         "drop_tables",
         "create_sql_table",
         "execute",