用于代码图提取的 Python MCP 服务器

# Clone the repository
git clone https://github.com/yourusername/python-mcp-new.git
cd python-mcp-new

# Create a virtual environment
python -m venv venv
source venv/bin/activate # On Windows, use: venv\Scripts\activate

# Install dependencies
pip install -r requirements.txt

# Token limit for extraction
TOKEN_LIMIT=8000

{
 "mcpServers": {
 "python-code-explorer": {
 "command": "python",
 "args": [
 "/path/to/python-mcp-new/server.py"
 ],
 "env": {
 "TOKEN_LIMIT": "8000"
 }
 }
 }
}

from agent import get_python_code

# Get Python code structure for a specific file
result = get_python_code(
 target_file="/home/user/project/main.py",
 root_repo_path="/home/user/project" # Optional, defaults to target file directory
)

# Process the result
target_file = result["target_file"]
print(f"Main file: {target_file['file_path']}")
print(f"Docstring: {target_file['docstring']}")

# Display related files
for ref_file in result["referenced_files"]:
 print(f"Related file: {ref_file['file_path']}")
 print(f"Object: {ref_file['object_name']}")
 print(f"Type: {ref_file['object_type']}")

# See if we're close to the token limit
print(f"Token usage: {result['token_count']}/{result['token_limit']}")

{
 "target_file": {
 "file_path": "main.py",
 "code": "import os\nimport sys\nfrom utils.helpers import format_output\n\ndef main():\n args = sys.argv[1:]\n if not args:\n print('No arguments provided')\n return\n \n result = format_output(args[0])\n print(result)\n\nif __name__ == '__main__':\n main()",
 "type": "target",
 "docstring": ""
 },
 "referenced_files": [
 {
 "file_path": "utils/helpers.py",
 "object_name": "format_output",
 "object_type": "function",
 "code": "def format_output(text):\n \"\"\"Format the input text for display.\"\"\"\n if not text:\n return ''\n return f'Output: {text.upper()}'\n",
 "docstring": "Format the input text for display.",
 "truncated": false
 }
 ],
 "additional_files": [
 {
 "file_path": "config.py",
 "code": "# Configuration settings\n\nDEBUG = True\nVERSION = '1.0.0'\nMAX_RETRIES = 3\n",
 "type": "related_by_directory",
 "docstring": "Configuration settings for the application."
 }
 ],
 "total_files": 3,
 "token_count": 450,
 "token_limit": 8000
}

from agent import handle_mcp_request
import json

# List available tools
list_request = {
 "jsonrpc": "2.0",
 "id": 1,
 "method": "tools/list"
}

response = handle_mcp_request(list_request)
print(json.dumps(response, indent=2))

{
 "jsonrpc": "2.0",
 "id": 1,
 "result": {
 "tools": [
 {
 "name": "get_python_code",
 "description": "Return the code of a target Python file and related files based on import/export proximity.",
 "inputSchema": {
 "type": "object",
 "properties": {
 "target_file": {
 "type": "string",
 "description": "Path to the Python file to analyze."
 },
 "root_repo_path": {
 "type": "string",
 "description": "Root directory of the repository. If not provided, the directory of the target file will be used."
 }
 },
 "required": ["target_file"]
 }
 }
 ]
 }
}

from agent import handle_mcp_request
import json

# Call the get_python_code tool
tool_request = {
 "jsonrpc": "2.0",
 "id": 2,
 "method": "tools/call",
 "params": {
 "name": "get_python_code",
 "arguments": {
 "target_file": "/home/user/project/main.py",
 "root_repo_path": "/home/user/project" # Optional
 }
 }
}

response = handle_mcp_request(tool_request)
print(json.dumps(response, indent=2))

{
 "jsonrpc": "2.0",
 "id": 2,
 "result": {
 "content": [
 {
 "type": "text",
 "text": "Python code analysis for /home/user/project/main.py"
 },
 {
 "type": "resource",
 "resource": {
 "uri": "resource://python-code/main.py",
 "mimeType": "application/json",
 "data": {
 "target_file": {
 "file_path": "main.py",
 "code": "import os\nimport sys\nfrom utils.helpers import format_output\n\ndef main():\n args = sys.argv[1:]\n if not args:\n print('No arguments provided')\n return\n \n result = format_output(args[0])\n print(result)\n\nif __name__ == '__main__':\n main()",
 "type": "target",
 "docstring": ""
 },
 "referenced_files": [
 {
 "file_path": "utils/helpers.py",
 "object_name": "format_output",
 "object_type": "function",
 "code": "def format_output(text):\n \"\"\"Format the input text for display.\"\"\"\n if not text:\n return ''\n return f'Output: {text.upper()}'\n",
 "docstring": "Format the input text for display.",
 "truncated": false
 }
 ],
 "additional_files": [
 {
 "file_path": "config.py",
 "code": "# Configuration settings\n\nDEBUG = True\nVERSION = '1.0.0'\nMAX_RETRIES = 3\n",
 "type": "related_by_directory",
 "docstring": "Configuration settings for the application."
 }
 ],
 "total_files": 3,
 "token_count": 450,
 "token_limit": 8000
 }
 }
 }
 ],
 "isError": false
 }
}

from agent import handle_mcp_request

# Call with invalid file path
faulty_request = {
 "jsonrpc": "2.0",
 "id": 3,
 "method": "tools/call",
 "params": {
 "name": "get_python_code",
 "arguments": {
 "target_file": "/path/to/nonexistent.py"
 }
 }
}

response = handle_mcp_request(faulty_request)
print(json.dumps(response, indent=2))

{
 "jsonrpc": "2.0",
 "id": 3,
 "result": {
 "content": [
 {
 "type": "text",
 "text": "Error processing Python code: No such file or directory: '/path/to/nonexistent.py'"
 }
 ],
 "isError": true
 }
}

python -m unittest discover tests

# Start the server with default settings
python run_server.py

# Specify a custom name
python run_server.py --name "My Code Explorer"

# Use a specific .env file
python run_server.py --env-file .env.production

# Install the MCP CLI
pip install "mcp[cli]"

# Start the server in development mode with the Inspector UI
mcp dev server.py

# Install the server in Claude Desktop
mcp install server.py

# With custom configuration
mcp install server.py --name "Python Code Explorer" -f .env

from server import mcp

# Configure the server
mcp.name = "Custom Code Explorer"

# Run the server
mcp.run()

from mcp.client import Client, Transport

# Connect to the server
client = Client(Transport.subprocess(["python", "server.py"]))
client.initialize()

# List available tools
for tool in client.tools:
 print(f"Tool: {tool.name}")

# Use the get_code tool
result = client.tools.get_code(target_file="path/to/your/file.py")
print(f"Found {len(result['referenced_files'])} referenced files")

# Clean up
client.shutdown()

python examples/mcp_client_example.py [optional_target_file.py]

@mcp.tool()
def analyze_imports(target_file: str) -> Dict[str, Any]:
 """Analyze all imports in a Python file."""
 # Implementation code here
 return {
 "file": target_file,
 "imports": [], # List of imports found
 "analysis": "" # Analysis of the imports
 }
 
@mcp.tool()
def find_python_files(directory: str, pattern: str = "*.py") -> list[str]:
 """Find Python files matching a pattern in a directory."""
 from pathlib import Path
 return [str(p) for p in Path(directory).glob(pattern) if p.is_file()]

@mcp.resource("python_stats://{directory}")
def get_stats(directory: str) -> Dict[str, Any]:
 """Get statistics about Python files in a directory."""
 from pathlib import Path
 stats = {
 "directory": directory,
 "file_count": 0,
 "total_lines": 0,
 "average_lines": 0
 }
 
 files = list(Path(directory).glob("**/*.py"))
 stats["file_count"] = len(files)
 
 if files:
 total_lines = 0
 for file in files:
 with open(file, "r") as f:
 total_lines += len(f.readlines())
 stats["total_lines"] = total_lines
 stats["average_lines"] = total_lines / len(files)
 
 return stats