Snowflake MCP Server

🚀 Snowflake MCP 服务器

Snowflake MCP 服务器是一个基于 Model Context Protocol (MCP) 的工具，它允许 Claude 安全地对 Snowflake 数据库执行只读查询，而不会对数据进行任何修改。

🚀 快速开始

Snowflake MCP 服务器为 Claude 提供了安全查询 Snowflake 数据库的途径。通过简单的配置和安装，你可以快速搭建并运行该服务器。

✨ 主要特性

• 灵活的身份验证：支持通过服务账户和私有密钥进行身份验证，也可使用外部浏览器进行交互式会话身份验证，连接到 Snowflake。
• 连接池管理：带有自动背景刷新的连接池，确保与 Snowflake 的持久连接。
• 多视图和数据库查询：支持在单个会话中查询多个视图和数据库。
• 多 SQL 语句类型支持：支持多种 SQL 语句类型，如 SELECT、SHOW、DESCRIBE、EXPLAIN、WITH。
• MCP 兼容处理器：提供 MCP 兼容的处理器，用于查询 Snowflake 数据。
• 只读操作与安全检查：仅支持只读操作，并带有安全检查，防止数据修改。
• Python 3.12+ 支持：支持 Python 3.12 及以上版本。
• 标准输入输出集成：基于标准输入输出 (stdio) 的 MCP 服务器，方便与 Claude Desktop 集成。

📦 安装指南

安装要求

python-dotenv
mypy
ruff
pytest
pydantic
snowflake-connector-python

安装步骤

pip install python-dotenv mypy ruff pytest pydantic snowflake-connector-python

💻 使用示例

环境变量配置

在项目根目录创建一个 .env 文件，并添加以下内容：

SNOWFLAKE_AUTH_TYPE=private_key 或 external_browser
SNOWFLAKE_PRIVATE_KEY_PATH=私钥路径（如使用 private_key 身份验证）
SNOWFLAKE_ACCOUNT_NAME=账户名称

运行服务器

python mcp_server_snowflake/main.py

📚 详细文档

功能模块

• mcp_server_snowflake.mcp_server：MCP 服务器主逻辑。
• mcp_server_snowflake.models：数据模型和验证。
• mcp_server_snowflake.auth：身份验证处理。
• mcp_server_snowflake.utils：工具函数和配置管理。

安全注意事项

• 查询限制：所有查询都会自动添加 LIMIT 子句，防止大数据集的风险。
• 写入操作禁止：禁止执行任何写入操作（如 INSERT、UPDATE、DELETE）。
• 输入验证：输入验证严格过滤 SQL 注入风险。
• 凭据安全：私钥和凭据不会持久化到磁盘。
• 通信安全：使用 HTTPS 进行与 Snowflake 的通信。

开发信息

代码检查工具

# 静态类型检查
mypy mcp_server_snowflake/

# 代码格式检查
ruff check .

# 代码自动格式化
ruff format .

# 单元测试
pytest

贡献指南

欢迎贡献！请随意提交 Pull Request。

🔧 技术细节

• Snowflake 连接：使用 Snowflake Python 连接器 实现与 Snowflake 的连接。
• MCP 协议交互：基于 MCP 协议 与 Claude 进行交互。
• 数据验证：使用 Pydantic 进行数据验证。
• 环境变量管理：使用 python-dotenv 管理环境变量。