cinchdb 0.1.17__py3-none-any.whl → 0.1.18__py3-none-any.whl

cinchdb/managers/column.py

@@ -18,7 +18,8 @@ class ColumnManager:
     PROTECTED_COLUMNS = {"id", "created_at", "updated_at"}
 
     def __init__(
-        self, project_root: Path, database: str, branch: str, tenant: str = "main"
+        self, project_root: Path, database: str, branch: str, tenant: str = "main",
+        encryption_manager=None
     ):
         """Initialize column manager.
 
@@ -27,14 +28,16 @@ class ColumnManager:
             database: Database name
             branch: Branch name
             tenant: Tenant name (default: main)
+            encryption_manager: EncryptionManager instance for encrypted connections
         """
         self.project_root = Path(project_root)
         self.database = database
         self.branch = branch
         self.tenant = tenant
+        self.encryption_manager = encryption_manager
         self.db_path = get_tenant_db_path(project_root, database, branch, tenant)
         self.change_tracker = ChangeTracker(project_root, database, branch)
-        self.table_manager = TableManager(project_root, database, branch, tenant)
+        self.table_manager = TableManager(project_root, database, branch, tenant, encryption_manager)
 
     def list_columns(self, table_name: str) -> List[Column]:
         """List all columns in a table.
@@ -415,7 +418,7 @@ class ColumnManager:
 
         create_sql = f"CREATE TABLE {temp_table} ({', '.join(col_defs)})"
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             # Create new table
             conn.execute(create_sql)
 
@@ -488,7 +491,7 @@ class ColumnManager:
 
         # If making NOT NULL, check for NULL values
         if not nullable:
-            with DatabaseConnection(self.db_path) as conn:
+            with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
                 cursor = conn.execute(
                     f"SELECT COUNT(*) FROM {table_name} WHERE {column_name} IS NULL"
                 )

cinchdb/managers/data.py

@@ -20,7 +20,8 @@ class DataManager:
     """Manages data operations within a database tenant."""
 
     def __init__(
-        self, project_root: Path, database: str, branch: str, tenant: str = "main"
+        self, project_root: Path, database: str, branch: str, tenant: str = "main",
+        encryption_manager=None
     ):
         """Initialize data manager.
 
@@ -29,14 +30,16 @@ class DataManager:
             database: Database name
             branch: Branch name
             tenant: Tenant name (default: main)
+            encryption_manager: EncryptionManager instance for encrypted connections
         """
         self.project_root = Path(project_root)
         self.database = database
         self.branch = branch
         self.tenant = tenant
+        self.encryption_manager = encryption_manager
         self.db_path = get_tenant_db_path(project_root, database, branch, tenant)
-        self.table_manager = TableManager(project_root, database, branch, tenant)
-        self.query_manager = QueryManager(project_root, database, branch, tenant)
+        self.table_manager = TableManager(project_root, database, branch, tenant, encryption_manager)
+        self.query_manager = QueryManager(project_root, database, branch, tenant, encryption_manager)
 
     def select(
         self,
@@ -73,7 +76,7 @@ class DataManager:
         if offset:
             query += f" OFFSET {offset}"
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(query, params)
             rows = cursor.fetchall()
 
@@ -130,7 +133,7 @@ class DataManager:
             VALUES ({", ".join(placeholders)})
         """
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             try:
                 conn.execute(query, record_data)
                 conn.commit()
@@ -233,7 +236,7 @@ class DataManager:
 
         params = {**update_data, "id": data["id"]}
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             try:
                 conn.execute(query, params)
                 conn.commit()
@@ -273,7 +276,7 @@ class DataManager:
 
         query = f"DELETE FROM {table_name} WHERE {where_clause}"
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             try:
                 cursor = conn.execute(query, params)
                 deleted_count = cursor.rowcount
@@ -320,7 +323,7 @@ class DataManager:
         table_name = self._get_table_name(type(instances[0]))
         created_instances = []
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             try:
                 for instance in instances:
                     data = instance.model_dump()
@@ -370,7 +373,7 @@ class DataManager:
         if where_clause:
             query += f" WHERE {where_clause}"
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(query, params)
             result = cursor.fetchone()
             return result["count"] if result else 0
@@ -521,7 +524,7 @@ class DataManager:
         
         sql = f"DELETE FROM {table} WHERE {where_clause}"
         
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(sql, params)
             conn.commit()
             return cursor.rowcount
@@ -585,7 +588,7 @@ class DataManager:
         
         sql = f"UPDATE {table} SET {', '.join(set_clauses)} WHERE {where_clause}"
         
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(sql, all_params)
             conn.commit()
             return cursor.rowcount
@@ -616,7 +619,7 @@ class DataManager:
         sql = f"UPDATE {table} SET {set_clause} WHERE id = ?"
         params.append(record_id)
         
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(sql, params)
             conn.commit()
             
@@ -645,7 +648,7 @@ class DataManager:
         """
         sql = f"DELETE FROM {table} WHERE id = ?"
         
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(sql, [record_id])
             conn.commit()
             return cursor.rowcount > 0

cinchdb/managers/query.py

@@ -16,7 +16,8 @@ class QueryManager:
     """Manages SQL query execution with support for typed returns."""
 
     def __init__(
-        self, project_root: Path, database: str, branch: str, tenant: str = "main"
+        self, project_root: Path, database: str, branch: str, tenant: str = "main",
+        encryption_manager=None
     ):
         """Initialize query manager.
 
@@ -25,13 +26,15 @@ class QueryManager:
             database: Database name
             branch: Branch name
             tenant: Tenant name (default: main)
+            encryption_manager: EncryptionManager instance for encrypted connections
         """
         self.project_root = Path(project_root)
         self.database = database
         self.branch = branch
         self.tenant = tenant
+        self.encryption_manager = encryption_manager
         # Initialize tenant manager for lazy tenant handling
-        self.tenant_manager = TenantManager(project_root, database, branch)
+        self.tenant_manager = TenantManager(project_root, database, branch, encryption_manager)
     
     def _is_write_query(self, sql: str) -> bool:
         """Check if a SQL query is a write operation.
@@ -84,7 +87,7 @@ class QueryManager:
             self.tenant, is_write=False
         )
         
-        with DatabaseConnection(db_path) as conn:
+        with DatabaseConnection(db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(sql, params)
             rows = cursor.fetchall()
             return [dict(row) for row in rows]
@@ -214,7 +217,7 @@ class QueryManager:
             self.tenant, is_write=True
         )
 
-        with DatabaseConnection(db_path) as conn:
+        with DatabaseConnection(db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(sql, params)
             affected_rows = cursor.rowcount
             conn.commit()
@@ -243,7 +246,7 @@ class QueryManager:
             self.tenant, is_write=is_write
         )
 
-        with DatabaseConnection(db_path) as conn:
+        with DatabaseConnection(db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             try:
                 for params in params_list:
                     cursor = conn.execute(sql, params)

cinchdb/managers/table.py

@@ -24,7 +24,8 @@ class TableManager:
     PROTECTED_TABLE_PREFIXES = ("__", "sqlite_")
 
     def __init__(
-        self, project_root: Path, database: str, branch: str, tenant: str = "main"
+        self, project_root: Path, database: str, branch: str, tenant: str = "main",
+        encryption_manager=None
     ):
         """Initialize table manager.
 
@@ -33,14 +34,16 @@ class TableManager:
             database: Database name
             branch: Branch name
             tenant: Tenant name (default: main)
+            encryption_manager: EncryptionManager instance for encrypted connections
         """
         self.project_root = Path(project_root)
         self.database = database
         self.branch = branch
         self.tenant = tenant
+        self.encryption_manager = encryption_manager
         self.db_path = get_tenant_db_path(project_root, database, branch, tenant)
         self.change_tracker = ChangeTracker(project_root, database, branch)
-        self.tenant_manager = TenantManager(project_root, database, branch)
+        self.tenant_manager = TenantManager(project_root, database, branch, encryption_manager)
 
     def list_tables(self) -> List[Table]:
         """List all tables in the tenant.
@@ -50,7 +53,7 @@ class TableManager:
         """
         tables = []
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             # Get all tables first, then filter in Python (more reliable than SQL LIKE)
             cursor = conn.execute(
                 """
@@ -230,7 +233,7 @@ class TableManager:
         columns = []
         foreign_keys = {}
 
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             # Get foreign key information first
             fk_cursor = conn.execute(f"PRAGMA foreign_key_list({table_name})")
             for fk_row in fk_cursor.fetchall():
@@ -423,7 +426,7 @@ class TableManager:
         Returns:
             True if table exists
         """
-        with DatabaseConnection(self.db_path) as conn:
+        with DatabaseConnection(self.db_path, tenant_id=self.tenant, encryption_manager=self.encryption_manager) as conn:
             cursor = conn.execute(
                 "SELECT name FROM sqlite_master WHERE type='table' AND name=?",
                 (table_name,),