zaturn 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

zaturn/config.py

@@ -59,8 +59,12 @@ for s in source_list:
     elif source.startswith('postgresql://'):
         source_type = 'postgresql'
         source_name = source.split('/')[-1].split('?')[0]
-    elif source.startswith("mysql+pymysql://"):
+    elif source.startswith("mysql://") or source.startswith("mysql+pymysql://"):
         source_type = 'mysql'
+        s = s.replace('mysql://', 'mysql+pymysql://')
+        source_name = source.split('/')[-1].split('?')[0]
+    elif source.startswith('clickhouse://'):
+        source_type = 'clickhouse'
         source_name = source.split('/')[-1].split('?')[0]
     elif source.endswith(".duckdb"):
         source_type = "duckdb"

zaturn/core.py

@@ -62,7 +62,7 @@ def _list_tables(source_id: str):
                     if col.startswith("Tables_in_"):
                         return result[col].to_list()
                 
-            case "duckdb" | "csv" | "parquet":
+            case "duckdb" | "csv" | "parquet" | "clickhouse":
                 result = query_utils.execute_query(source, "SHOW TABLES")
                 return result['name'].to_list()
 
@@ -86,7 +86,7 @@ def describe_table(source_id: str, table_name: str) -> str:
         match source['type']:
             case 'sqlite':
                 result = query_utils.execute_query(source,
-                    f"PRAGMA table_info({table_name});"
+                    f'PRAGMA table_info({table_name});'
                 )
                 return result.to_markdown(index=False)
                 
@@ -96,7 +96,7 @@ def describe_table(source_id: str, table_name: str) -> str:
                 )
                 return result.to_markdown(index=False)
                 
-            case "mysql" | "duckdb" | "csv" | "parquet":
+            case "mysql" | "duckdb" | "csv" | "parquet" | "clickhouse":
                 result = query_utils.execute_query(source,
                     f"DESCRIBE {table_name};"
                 )

zaturn/query_utils.py

@@ -1,3 +1,4 @@
+import clickhouse_connect
 import duckdb
 import numpy as np
 import os
@@ -12,16 +13,11 @@ from zaturn import config
 def execute_query(source: dict, query: str):
     """Run the query using the appropriate engine and read only config"""
     url = source['url']
-    
+                
     match source['type']:
         case "sqlite":
-            if "mode=ro" in url:
-                pass
-            elif '?' in url:
-                url += '&mode=ro'
-            else:
-                url += '?mode=ro'
             with sqlalchemy.create_engine(url).connect() as conn:
+                conn.execute(sqlalchemy.text('PRAGMA query_only = ON;'))
                 result = conn.execute(sqlalchemy.text(query))
                 return pd.DataFrame(result)
 
@@ -31,6 +27,7 @@ def execute_query(source: dict, query: str):
                 session.autoflush = False
                 session.autocommit = False
                 session.flush = lambda *args: None
+                session.execute(sqlalchemy.text('SET SESSION TRANSACTION READ ONLY;'))
                 result = session.execute(sqlalchemy.text(query))
                 return pd.DataFrame(result)
 
@@ -46,6 +43,11 @@ def execute_query(source: dict, query: str):
                     result = conn.execute(sqlalchemy.text(query))
                     return pd.DataFrame(result)
 
+        case "clickhouse":
+            client = clickhouse_connect.get_client(dsn=url)
+            client.query('SET readonly=1;')
+            return client.query_df(query, use_extended_dtypes=False)
+
         case "duckdb":
             conn = duckdb.connect(url, read_only=True)
             return conn.execute(query).df()

{zaturn-0.1.5.dist-info → zaturn-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.1.5
+Version: 0.1.7
 Summary: AI Data Analysis MCP
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>
@@ -9,6 +9,7 @@ Project-URL: Issues, https://github.com/kdqed/zaturn/issues
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: clickhouse-connect>=0.8.17
 Requires-Dist: cryptography>=44.0.2
 Requires-Dist: duckdb>=1.2.1
 Requires-Dist: fastmcp>=0.4.1
@@ -56,7 +57,7 @@ https://github.com/user-attachments/assets/d42dc433-e5ec-4b3e-bef0-5cfc097396ab
 
 ### Multiple Data Sources 
 Zaturn can currently connect to the following data sources: 
-- SQL Databases: PostgreSQL, SQLite, DuckDB, MySQL
+- SQL Databases: PostgreSQL, SQLite, DuckDB, MySQL, ClickHouse
 - Files: CSV, Parquet
 
 Connectors for more data sources are being added.
@@ -105,6 +106,7 @@ OR add a `sources.txt` to the Zaturn config directory:
 ```
 postgresql://username:password@host:port/dbname
 mysql+pymysql://username:password@host:3306/dbname
+clickhouse://username:password@host:port/dbname
 sqlite:////full/path/to/sample_dbs/northwind.db
 /full/path/to/sample_dbs/titanic.parquet
 /full/path/to/sample_dbs/ny_aq.csv

zaturn-0.1.7.dist-info/RECORD

@@ -0,0 +1,12 @@
+zaturn/__init__.py,sha256=v4t5fkRuIJFE-SBxCa5pBjZv0EoC0eWK75nU9iaa7Rg,267
+zaturn/config.py,sha256=qdFy7GyGyXO-rEyqb8kO3wym3DEk2nQEiz-JscDKjaw,3446
+zaturn/core.py,sha256=MNB37smt4wkdHCTsZECesrksnx0FnCnHBw4kkWOTvTQ,4671
+zaturn/query_utils.py,sha256=4o37eGV8X1sbYvme_jKaPX8hdlynfzMK_0qCSf6XZlo,3031
+zaturn/visualizations.py,sha256=0ON70D_mK4o0oyfEKqAhr2jkKWz_5-kNKkD6_TGBR9k,5014
+zaturn/example_data/all_pokemon_data.csv,sha256=SUlGHHWbehuLg-ch1YUrQ6-xBtqHGw6rIkyn70fAgCk,130893
+zaturn-0.1.7.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
+zaturn-0.1.7.dist-info/METADATA,sha256=b_pULTQiosZBRg6lZSAWVD23HxRmjercbL3bj06wOHE,7179
+zaturn-0.1.7.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+zaturn-0.1.7.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
+zaturn-0.1.7.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
+zaturn-0.1.7.dist-info/RECORD,,

{zaturn-0.1.5.dist-info → zaturn-0.1.7.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.0)
+Generator: setuptools (80.3.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

zaturn-0.1.5.dist-info/RECORD

@@ -1,12 +0,0 @@
-zaturn/__init__.py,sha256=v4t5fkRuIJFE-SBxCa5pBjZv0EoC0eWK75nU9iaa7Rg,267
-zaturn/config.py,sha256=gF5M6Agmixw2A4vpWqIF3ICVnClPeASA51dhp1bkk04,3221
-zaturn/core.py,sha256=9zcKb0FbkgGkDtWfBA6_O5NQf6GVKG69HrqOr5nhsLU,4641
-zaturn/query_utils.py,sha256=zyQjcRnPKGHZdf0XHzQeMxHw9vieZIwXhBbVGP87ml4,2801
-zaturn/visualizations.py,sha256=0ON70D_mK4o0oyfEKqAhr2jkKWz_5-kNKkD6_TGBR9k,5014
-zaturn/example_data/all_pokemon_data.csv,sha256=SUlGHHWbehuLg-ch1YUrQ6-xBtqHGw6rIkyn70fAgCk,130893
-zaturn-0.1.5.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
-zaturn-0.1.5.dist-info/METADATA,sha256=Vl68nwLT5NVIdtRRMnV6ncyE4oTIoC3GF_eGJcLN-uA,7077
-zaturn-0.1.5.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-zaturn-0.1.5.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
-zaturn-0.1.5.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
-zaturn-0.1.5.dist-info/RECORD,,