zaturn 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

zaturn/mcp/__init__.py

@@ -56,6 +56,10 @@ for s in source_list:
         source_type = 'mysql'
         s = s.replace('mysql://', 'mysql+pymysql://')
         source_name = source.split('/')[-1].split('?')[0]
+    elif source.startswith("mssql://"):
+        source_type = 'mssql'
+        s = s.replace('mssql://', 'mssql+pymssql://')
+        source_name = source.split('/')[-1].split('?')[0]    
     elif source.startswith('clickhouse://'):
         source_type = 'clickhouse'
         source_name = source.split('/')[-1].split('?')[0]

zaturn/tools/query_utils.py

@@ -30,6 +30,10 @@ def list_tables(source):
                 for col in list(result):
                     if col.startswith("Tables_in_"):
                         return result[col].to_list()
+
+            case "mssql":
+                result = execute_query(source, "SELECT name FROM sys.tables")
+                return result['name'].to_list()
                 
             case "duckdb" | "csv" | "parquet" | "clickhouse":
                 result = execute_query(source, "SHOW TABLES")
@@ -46,7 +50,7 @@ def describe_table(source, table_name):
                 f'PRAGMA table_info("{table_name}");'
             )
             
-        case 'postgresql':
+        case 'postgresql' | 'mssql':
             return execute_query(source,
                 f"SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE table_name = '{table_name}';"
             )
@@ -81,6 +85,14 @@ def execute_query(source: dict, query: str):
                 result = session.execute(sqlalchemy.text(query))
                 return pd.DataFrame(result)
 
+        case "mssql":
+            engine = sqlalchemy.create_engine(url)
+            with Session(engine) as session:
+                # no known way to ensure read-only here
+                # please use read-only credentials
+                result = session.execute(sqlalchemy.text(query))
+                return pd.DataFrame(result)
+
         case "postgresql":
             engine = sqlalchemy.create_engine(url)
             with engine.connect() as conn:

{zaturn-0.2.0.dist-info → zaturn-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.2.0
+Version: 0.2.1
 Summary: AI Data Analysis MCP & Studio
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>
@@ -25,6 +25,7 @@ Requires-Dist: platformdirs>=4.3.7
 Requires-Dist: plotly[express]>=6.0.1
 Requires-Dist: psycopg2-binary>=2.9.10
 Requires-Dist: pyarrow>=19.0.1
+Requires-Dist: pymssql>=2.3.7
 Requires-Dist: pymysql>=1.1.1
 Requires-Dist: python-lsp-server>=1.12.2
 Requires-Dist: python-slugify>=8.0.4
@@ -58,7 +59,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, ClickHouse
+- SQL Databases: PostgreSQL, SQLite, DuckDB, MySQL, ClickHouse, SQL Server
 - Files: CSV, Parquet
 
 Connectors for more data sources are being added.

{zaturn-0.2.0.dist-info → zaturn-0.2.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-zaturn/mcp/__init__.py,sha256=-fyROX8LHX1UXiLYE3kk17pXIILW-DVrFviechA57Pg,3213
+zaturn/mcp/__init__.py,sha256=lBfs5Ztpih-Dn5vuTxJUTzjD2SM2icAZZM9GgRVozYU,3399
 zaturn/studio/__init__.py,sha256=A1QxWydtsE0VmsJlvF7O9Fx1-WKjV3MBPje7knFSDzw,83
 zaturn/studio/agent_wrapper.py,sha256=4mEMRfE1Tfv63D7OIabqUbEJdIgO2yo8UXO6PKvNrmI,4603
 zaturn/studio/app.py,sha256=LEoSjs5wpZGX1B7pcJ2e_6CaxOFjbO6JitxlkJv__30,9208
@@ -28,12 +28,12 @@ zaturn/studio/templates/css/style.css,sha256=cKImLAjLp6XjHdT16v9yJS0DHG2wWx8djR1
 zaturn/tools/__init__.py,sha256=vOcHneRuZll9k-2e3sra7m8qfdc4IoAFYc2g_xK79HE,262
 zaturn/tools/config.py,sha256=8YCwFp1emhEfbbJd_KTQ_Nrl4yt9tJoDZHgMXatJGjk,3382
 zaturn/tools/core.py,sha256=4fHxSIbO22cTIejJk02GzUBfWimh1Ql9mZ3JgI1qLBY,2893
-zaturn/tools/query_utils.py,sha256=x769ZrpbnpxHwdQf6PbAUx6-qMGWuqfMOTA3kVjRUcE,4808
+zaturn/tools/query_utils.py,sha256=OnXbONmEgsHcScrJ76b7P3BRLOKgi8TMkhxBcLQK9cY,5306
 zaturn/tools/visualizations.py,sha256=aPXCN1YIjCz6uiwJjduKCiLLtnBFX92hTvjy5EqQYk8,9509
 zaturn/tools/example_data/all_pokemon_data.csv,sha256=SUlGHHWbehuLg-ch1YUrQ6-xBtqHGw6rIkyn70fAgCk,130893
-zaturn-0.2.0.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
-zaturn-0.2.0.dist-info/METADATA,sha256=9_odOtbZ5NNo5nu5hDOXIEWG1TlIlxr39DdDVakp_pg,3957
-zaturn-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-zaturn-0.2.0.dist-info/entry_points.txt,sha256=MWMWX0dE_ZQM4StGKCLxJym_D91F82RcBi2LBj0hEho,82
-zaturn-0.2.0.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
-zaturn-0.2.0.dist-info/RECORD,,
+zaturn-0.2.1.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
+zaturn-0.2.1.dist-info/METADATA,sha256=NsCeZRR_EKj5TekqsM4R4fgicuUbwWf29PxHm5PQ_7k,3999
+zaturn-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+zaturn-0.2.1.dist-info/entry_points.txt,sha256=MWMWX0dE_ZQM4StGKCLxJym_D91F82RcBi2LBj0hEho,82
+zaturn-0.2.1.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
+zaturn-0.2.1.dist-info/RECORD,,