snowflake-mcp 0.1.0__tar.gz

snowflake_mcp-0.1.0/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: snowflake-mcp
+Version: 0.1.0
+Summary: A Snowflake MCP client for Python
+Requires-Python: >=3.14
+Description-Content-Type: text/markdown
+Requires-Dist: dotenv>=0.9.9
+Requires-Dist: fastmcp>=3.4.2
+Requires-Dist: snowflake-connector-python[pandas]>=4.6.0

snowflake_mcp-0.1.0/pyproject.toml

@@ -0,0 +1,22 @@
+[project]
+name = "snowflake-mcp"
+version = "0.1.0"
+description = "A Snowflake MCP client for Python"
+readme = "README.md"
+requires-python = ">=3.14"
+dependencies = [
+    "dotenv>=0.9.9",
+    "fastmcp>=3.4.2",
+    "snowflake-connector-python[pandas]>=4.6.0",
+]
+
+
+[build-system]
+requires = ["setuptools>=65.5.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[tools.setuptools.packages.find]
+where = ["src"]
+
+[project.scripts]
+snowflake-mcp = "snowflake_mcp.main:main"

snowflake_mcp-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

snowflake_mcp-0.1.0/src/snowflake-mcp/main.py

@@ -0,0 +1,4 @@
+from snowflake_mcp import mcp
+
+if __name__ == "__main__":
+    mcp.run(transport="stdio")

snowflake_mcp-0.1.0/src/snowflake-mcp/snowflake_mcp.py

@@ -0,0 +1,115 @@
+from snowflake.connector import connect
+from typing import Optional
+from snowflake.connector import SnowflakeConnection
+from dotenv import load_dotenv
+import os
+from fastmcp import FastMCP
+
+mcp = FastMCP("snowflake-mcp", instructions="version: 0.1.0, This is a Snowflake MCP that can execute SQL queries on a Snowflake database.")
+
+load_dotenv()  # Load environment variables from .env file
+
+def get_query_type(sql):
+    first_word = sql.strip().split()[0].upper()
+
+    ddl = {
+        "CREATE", "ALTER", "DROP", "TRUNCATE",
+        "RENAME", "COMMENT"
+    }
+
+    dml = {
+        "INSERT", "UPDATE", "DELETE", "MERGE", "COPY"
+    }
+
+    query = {
+        "SELECT", "SHOW", "DESCRIBE", "DESC", "EXPLAIN"
+    }
+
+    tcl = {
+        "COMMIT", "ROLLBACK"
+    }
+
+    dcl = {
+        "GRANT", "REVOKE"
+    }
+
+    if first_word in ddl:
+        return "DDL"
+    elif first_word in dml:
+        return "DML"
+    elif first_word in dcl:
+        return "DCL"
+    elif first_word in tcl:
+        return "TCL"
+    elif first_word in query:
+        return "QUERY"
+
+    return "UNKNOWN"
+def get_snowflake_connection(user: str, password: str, account: str, warehouse: Optional[str] = "COMPUTE_WH", role: Optional[str] = "PUBLIC", database: Optional[str] = "SNOWFLAKE_SAMPLE_DATA", schema: Optional[str] = "INFORMATION_SCHEMA") -> SnowflakeConnection:
+    """
+    Establishes a connection to a Snowflake database.
+
+    Parameters:
+    user (str): The username for the Snowflake account.
+    password (str): The password for the Snowflake account.
+    account (str): The Snowflake account identifier.
+    warehouse (str): The name of the warehouse to use.
+    role (str): The role to use for the connection.
+    database (str): The name of the database to connect to.
+    schema (str): The name of the schema to use.
+
+    Returns:
+    connection: A Snowflake connection object.
+    """
+    try:
+        connection = connect(
+            user=user,
+            password=password,
+            account=account,
+            warehouse=warehouse,
+            role=role,
+            database=database,
+            schema=schema
+        )
+        return connection
+    except Exception as e:
+        print(f"Error connecting to Snowflake: {e}")
+        return None
+    
+
+@mcp.tool()
+def execute_query(query: str):
+    """
+    Executes a SQL query on the Snowflake database.
+
+    Parameters:
+    connection (SnowflakeConnection): The Snowflake connection object.
+    query (str): The SQL query to execute.
+
+    Returns:
+    result: The result of the query execution.
+    """
+
+    if get_query_type(query) != "QUERY":
+        print("Only SELECT queries are allowed.")
+        return None
+    try:
+        connection = get_snowflake_connection(
+            user=os.getenv("SNOWFLAKE_USER"),
+            password=os.getenv("SNOWFLAKE_PASSWORD"),
+            account=os.getenv("SNOWFLAKE_ACCOUNT"),
+        )
+        print(connection)
+        cursor = connection.cursor()
+        cursor.execute(query)
+        result = cursor.fetchall()
+        cursor.close()
+        return result
+    except Exception as e:
+        print(f"Error executing query: {e}")
+        return None
+    
+
+
+if __name__ == "__main__":
+    mcp.run(transport="stdio")

snowflake_mcp-0.1.0/src/snowflake_mcp.egg-info/PKG-INFO

@@ -0,0 +1,9 @@
+Metadata-Version: 2.4
+Name: snowflake-mcp
+Version: 0.1.0
+Summary: A Snowflake MCP client for Python
+Requires-Python: >=3.14
+Description-Content-Type: text/markdown
+Requires-Dist: dotenv>=0.9.9
+Requires-Dist: fastmcp>=3.4.2
+Requires-Dist: snowflake-connector-python[pandas]>=4.6.0

snowflake_mcp-0.1.0/src/snowflake_mcp.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+pyproject.toml
+src/__init__.py
+src/snowflake-mcp/__init__.py
+src/snowflake-mcp/main.py
+src/snowflake-mcp/snowflake_mcp.py
+src/snowflake_mcp.egg-info/PKG-INFO
+src/snowflake_mcp.egg-info/SOURCES.txt
+src/snowflake_mcp.egg-info/dependency_links.txt
+src/snowflake_mcp.egg-info/entry_points.txt
+src/snowflake_mcp.egg-info/requires.txt
+src/snowflake_mcp.egg-info/top_level.txt

snowflake_mcp-0.1.0/src/snowflake_mcp.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

snowflake_mcp-0.1.0/src/snowflake_mcp.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+snowflake-mcp = snowflake_mcp.main:main

snowflake_mcp-0.1.0/src/snowflake_mcp.egg-info/requires.txt

@@ -0,0 +1,3 @@
+dotenv>=0.9.9
+fastmcp>=3.4.2
+snowflake-connector-python[pandas]>=4.6.0

snowflake_mcp-0.1.0/src/snowflake_mcp.egg-info/top_level.txt

@@ -0,0 +1,2 @@
+__init__
+snowflake-mcp