DataOps-platform/mcp-servers/task-manager/get_config.py

#!/usr/bin/env python3
"""
读取Flask配置文件中的production配置
用于MCP服务器获取数据库连接信息
"""
import sys
import os
import json

# 添加项目根目录到Python路径
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '../..'))
sys.path.insert(0, project_root)

try:
    from app.config.config import config
    
    # 从config字典中获取production配置类（注意：这是类，不是实例）
    ProductionConfig = config['production']
    
    # 直接访问类的属性（类属性，不需要实例化）
    database_url = ProductionConfig.SQLALCHEMY_DATABASE_URI
    
    # 输出数据库URI
    result = {
        'database_url': database_url,
        'status': 'success'
    }
    
    print(json.dumps(result))
    
except Exception as e:
    # 如果读取失败，返回错误信息
    result = {
        'status': 'error',
        'error': str(e),
        'database_url': None
    }
    print(json.dumps(result))
    sys.exit(1)