interloper-sql 0.1.0__tar.gz → 0.2.0rc1__tar.gz

{interloper_sql-0.1.0 → interloper_sql-0.2.0rc1}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.3
 Name: interloper-sql
-Version: 0.1.0
+Version: 0.2.0rc1
 Summary: Interloper SQLAlchemy IO managers
 Author: Guillaume Onfroy
 Author-email: Guillaume Onfroy <guillaume@digitlcloud.com>
 Requires-Dist: sqlalchemy>=2.0
-Requires-Dist: interloper
+Requires-Dist: interloper-core
 Requires-Dist: pymysql>=1.1 ; extra == 'mysql'
 Requires-Dist: psycopg2-binary>=2.9 ; extra == 'postgres'
 Requires-Python: >=3.10

{interloper_sql-0.1.0 → interloper_sql-0.2.0rc1}/pyproject.toml

@@ -3,14 +3,14 @@
 # ###############
 [project]
 name = "interloper-sql"
-version = "0.1.0"
+version = "0.2.0-rc.1"
 description = "Interloper SQLAlchemy IO managers"
 readme = "README.md"
 authors = [{ name = "Guillaume Onfroy", email = "guillaume@digitlcloud.com" }]
 requires-python = ">=3.10"
 dependencies = [
     "sqlalchemy>=2.0",
-    "interloper",
+    "interloper-core",
 ]
 
 [project.optional-dependencies]
@@ -22,7 +22,7 @@ requires = ["uv_build>=0.9.5,<0.10.0"]
 build-backend = "uv_build"
 
 [tool.uv.sources]
-interloper = { workspace = true }
+interloper-core = { workspace = true }
 
 # ###############
 # RUFF

{interloper_sql-0.1.0 → interloper_sql-0.2.0rc1}/src/interloper_sql/__init__.py

@@ -1,3 +1,5 @@
+"""Interloper SQL integration for relational database IO via SQLAlchemy."""
+
 from interloper_sql.io import MySQLIO, PostgresIO, SqlIO, SqliteIO
 
 __all__ = [

{interloper_sql-0.1.0 → interloper_sql-0.2.0rc1}/src/interloper_sql/io/base.py

@@ -242,6 +242,35 @@ class SqlIO(DatabaseIO):
             result = conn.execute(sa_table.select().where(sa_table.c[column] == value))
             return [dict(row._mapping) for row in result]
 
+    # ------------------------------------------------------------------
+    # Introspection
+    # ------------------------------------------------------------------
+
+    def _count_by_partition(
+        self, table: str, schema: str | None, column: str,
+    ) -> dict[str, int]:
+        """Return row counts grouped by partition column via SQL ``GROUP BY``.
+
+        Args:
+            table: Target table name
+            schema: Database schema
+            column: Column to group by
+
+        Returns:
+            Mapping from partition value (as string) to row count.
+
+        Raises:
+            TableNotFoundError: If the table does not exist.
+        """
+        from sqlalchemy import func
+
+        sa_table = self._require_table(table, schema)
+        col = sa_table.c[column]
+        stmt = sa_table.select().with_only_columns(col, func.count()).group_by(col)
+        with self._engine.connect() as conn:
+            result = conn.execute(stmt)
+            return {str(row[0]): row[1] for row in result}
+
     # ------------------------------------------------------------------
     # Lifecycle
     # ------------------------------------------------------------------