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

zaturn/config.py

@@ -59,8 +59,9 @@ 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'

zaturn/core.py

@@ -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)
                 

zaturn/query_utils.py

@@ -13,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)
 
@@ -32,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)
 

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zaturn
-Version: 0.1.6
+Version: 0.1.7
 Summary: AI Data Analysis MCP
 Author-email: Karthik Devan <krtdvn@gmail.com>
 Maintainer-email: Karthik Devan <krtdvn@gmail.com>

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.6.dist-info → zaturn-0.1.7.dist-info}/WHEEL

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

zaturn-0.1.6.dist-info/RECORD

@@ -1,12 +0,0 @@
-zaturn/__init__.py,sha256=v4t5fkRuIJFE-SBxCa5pBjZv0EoC0eWK75nU9iaa7Rg,267
-zaturn/config.py,sha256=V0sqe72zCLyHVJyH27vOR401hA7NnRhkfaJlIeyNoXo,3359
-zaturn/core.py,sha256=hASmXF6iMQA5M62tf94YjE5D3Rja9Vj5-8zMVKQ_tOY,4671
-zaturn/query_utils.py,sha256=zXy8-eDpqpehgdHBs0zjLPfpLDsbF4xtiCEU1dE0xms,3028
-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.6.dist-info/licenses/LICENSE,sha256=mZSuFlbEBZGl0-8ULRMLdRDbhau5hrWRNQOjytYeaug,1070
-zaturn-0.1.6.dist-info/METADATA,sha256=tO7Fc7dIV_YumwTavFxKWQMGEitbDBE82zDv1J_vEVw,7179
-zaturn-0.1.6.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-zaturn-0.1.6.dist-info/entry_points.txt,sha256=N1UZC2zvod92_Brs4A2xZiAnt-iGLBNryglXfwhxfj4,43
-zaturn-0.1.6.dist-info/top_level.txt,sha256=KLUnwQwVZkfd5YCnnqR35MOOs8KLhanPGelvmRo2MVA,7
-zaturn-0.1.6.dist-info/RECORD,,