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

zaturn/mcp/__init__.py

@@ -1,10 +1,11 @@
 import argparse
+import importlib.resources
 import os
 import platformdirs
-import pkg_resources
 import sys
 
 from fastmcp import FastMCP
+from fastmcp.tools.tool import Tool
 
 from zaturn.tools import ZaturnTools
 
@@ -30,19 +31,17 @@ if not source_list:
     source_list = args.sources
 
 if not source_list:
-    source_list = [
-        pkg_resources.resource_filename(
-            'zaturn',
-            os.path.join('mcp', 'example_data', 'all_pokemon_data.csv')
-        )
-    ]
+    with importlib.resources.path(
+        'zaturn.tools.example_data', 'all_pokemon_data.csv'
+    ) as source_path:
+        source_list = [str(source_path)]
+    
     print("No data sources provided. Loading example dataset for demonstration.")
     print(f"\nTo load your datasets, add them to {SOURCES_FILE} (one source URL or full file path per line)")
     print("\nOr use command line args to specify data sources:")
     print("zaturn_mcp sqlite:///path/to/mydata.db /path/to/my_file.csv")
     print(f"\nNOTE: Sources in command line args will be ignored if sources are found in {SOURCES_FILE}")
 
-    
 SOURCES = {}
 for s in source_list:
     source = s.lower()
@@ -56,6 +55,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]
@@ -86,8 +89,8 @@ for s in source_list:
 def ZaturnMCP(sources):
     zaturn_tools = ZaturnTools(sources)
     zaturn_mcp = FastMCP()
-    for tool in zaturn_tools.tools:
-        zaturn_mcp.add_tool(tool)
+    for tool_function in zaturn_tools.tools:
+        zaturn_mcp.add_tool(Tool.from_function(tool_function))
 
     return zaturn_mcp
 

zaturn/tools/config.py

@@ -1,8 +1,5 @@
-import argparse
 import os
 import platformdirs
-import pkg_resources
-import sys
 
 # Basic Setup
 USER_DATA_DIR = platformdirs.user_data_dir('zaturn', 'zaturn')
@@ -13,83 +10,3 @@ SOURCES_FILE = os.path.join(USER_DATA_DIR, 'sources.txt')
 os.makedirs(QUERIES_DIR, exist_ok=True)
 os.makedirs(VISUALS_DIR, exist_ok=True)
 
-# Parse command line args
-parser = argparse.ArgumentParser(
-    description="Zaturn: A read-only BI tool for analyzing various data sources"
-)
-parser.add_argument('--noimg', action='store_const', 
-    const=True, default=False, 
-    help='Return image file paths instead of images for visuals. Use when MCP client cannot render images.',
-)
-parser.add_argument('sources', nargs=argparse.REMAINDER, default=[], 
-    help='Data source (can be specified multiple times). Can be SQLite, MySQL, PostgreSQL connection string, or a path to CSV, Parquet, or DuckDB file.'
-)
-args = parser.parse_args()
-
-# Read and parse sources
-source_list = []
-if os.path.exists(SOURCES_FILE):
-    with open(SOURCES_FILE) as f:
-        source_list = [line.strip('\n') for line in f.readlines() if line.strip('\n')]
-
-if not source_list:
-    source_list = args.sources
-
-if not source_list:
-    source_list = [
-        pkg_resources.resource_filename(
-            'zaturn',
-            os.path.join('mcp', 'example_data', 'all_pokemon_data.csv')
-        )
-    ]
-    print("No data sources provided. Loading example dataset for demonstration.")
-    print(f"\nTo load your datasets, add them to {SOURCES_FILE} (one source URL or full file path per line)")
-    print("\nOr use command line args to specify data sources:")
-    print("zaturn_mcp sqlite:///path/to/mydata.db /path/to/my_file.csv")
-    print(f"\nNOTE: Sources in command line args will be ignored if sources are found in {SOURCES_FILE}")
-    
-CLI_SOURCES = {}
-for s in source_list:
-    source = s.lower()
-    if source.startswith('sqlite://'):
-        source_type = 'sqlite'
-        source_name = source.split('/')[-1].split('?')[0].split('.db')[0]
-    elif source.startswith('postgresql://'):
-        source_type = 'postgresql'
-        source_name = source.split('/')[-1].split('?')[0]
-    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"
-        source_name = source.split('/')[-1].split('.')[0]
-    elif source.endswith(".csv"):
-        source_type = "csv"
-        source_name = source.split('/')[-1].split('.')[0]
-    elif source.endswith(".parquet") or source.endswith(".pq"):
-        source_type = "parquet"
-        source_name = source.split('/')[-1].split('.')[0]
-    else:
-        continue
-
-    source_id = f'{source_name}-{source_type}'
-    if source_id in CLI_SOURCES:
-        i = 2
-        while True:
-            source_id = f'{source_name}{i}-{source_type}'
-            if source_id not in SOURCES:
-                break
-            i += 1
-
-    CLI_SOURCES[source_id] = {'url': s, 'type': source_type}
-
-
-# Other Settings
-CLI_RETURN_IMAGES = not args.noimg
-
-
-

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.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.2.0
+Version: 0.2.2
 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.
@@ -77,7 +78,7 @@ More visualization capabilities are being added.
 
 ## Installation & Setup
 
-See [https://zaturn.pro/install](https://zaturn.pro/install)
+See [https://zaturn.pro/docs/install](https://zaturn.pro/docs/install)
 
 
 ## Roadmap
@@ -97,7 +98,6 @@ Analyst:
   I can get it done right now. Actually, you can do it right now.
   You know what? The boss can do it right now.
 ```
-- A native notebook interface 
 
 ## Help And Feedback
 

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

@@ -1,4 +1,4 @@
-zaturn/mcp/__init__.py,sha256=-fyROX8LHX1UXiLYE3kk17pXIILW-DVrFviechA57Pg,3213
+zaturn/mcp/__init__.py,sha256=8pTcOyJ3tK7TJcP4HBDVQA_MVuMrfYJwa8sWKfCvXIE,3466
 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
@@ -26,14 +26,14 @@ zaturn/studio/templates/setup_prompt.html,sha256=K8l3ewlEyrUFiMmw2qPyDFbFNkq1RCB
 zaturn/studio/templates/user_message.html,sha256=WbvdVr4AVzVx6MG0eQ_ulMeHB9rpScZc-EZlCoYFEgw,90
 zaturn/studio/templates/css/style.css,sha256=cKImLAjLp6XjHdT16v9yJS0DHG2wWx8djR18SHfMqvE,6264
 zaturn/tools/__init__.py,sha256=vOcHneRuZll9k-2e3sra7m8qfdc4IoAFYc2g_xK79HE,262
-zaturn/tools/config.py,sha256=8YCwFp1emhEfbbJd_KTQ_Nrl4yt9tJoDZHgMXatJGjk,3382
+zaturn/tools/config.py,sha256=Ia-WzK_G6vQqB7norvaffb3fKLjtmXupVqAWsxdu6Gs,354
 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.2.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
+zaturn-0.2.2.dist-info/METADATA,sha256=BSnrqH9k7MNfqmRVPEO6q3884y-8Ym1X6ctklZz6NZc,3978
+zaturn-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+zaturn-0.2.2.dist-info/entry_points.txt,sha256=MWMWX0dE_ZQM4StGKCLxJym_D91F82RcBi2LBj0hEho,82
+zaturn-0.2.2.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
+zaturn-0.2.2.dist-info/RECORD,,