agent功能开发增加MCP后端

2025-11-07 21:46:50 +08:00
parent d1d8d1a25d
commit 3a058fd805
5 changed files with 2098 additions and 6 deletions
--- a/mcp_database.py
+++ b/mcp_database.py
@@ -544,3 +544,240 @@ async def get_stock_comparison(
                "comparison_type": metric,
                "stocks": [convert_row(row) for row in results]
            }
+
+
+async def get_user_favorite_stocks(user_id: str, limit: int = 100) -> List[Dict[str, Any]]:
+    """
+    获取用户自选股列表
+
+    Args:
+        user_id: 用户ID
+        limit: 返回条数
+
+    Returns:
+        自选股列表（包含最新行情数据）
+    """
+    pool = await get_pool()
+
+    async with pool.acquire() as conn:
+        async with conn.cursor(aiomysql.DictCursor) as cursor:
+            # 查询用户自选股（假设有 user_favorites 表）
+            # 如果没有此表，可以根据实际情况调整
+            query = """
+                SELECT
+                    f.user_id,
+                    f.stock_code,
+                    b.SECNAME as stock_name,
+                    b.F030V as industry,
+                    t.F007N as current_price,
+                    t.F010N as change_pct,
+                    t.F012N as turnover_rate,
+                    t.F026N as pe_ratio,
+                    t.TRADEDATE as latest_trade_date,
+                    f.created_at as favorite_time
+                FROM user_favorites f
+                INNER JOIN ea_baseinfo b ON f.stock_code = b.SECCODE
+                LEFT JOIN (
+                    SELECT SECCODE, MAX(TRADEDATE) as max_date
+                    FROM ea_trade
+                    GROUP BY SECCODE
+                ) latest ON b.SECCODE = latest.SECCODE
+                LEFT JOIN ea_trade t ON b.SECCODE = t.SECCODE
+                    AND t.TRADEDATE = latest.max_date
+                WHERE f.user_id = %s AND f.is_deleted = 0
+                ORDER BY f.created_at DESC
+                LIMIT %s
+            """
+
+            await cursor.execute(query, [user_id, limit])
+            results = await cursor.fetchall()
+
+            return [convert_row(row) for row in results]
+
+
+async def get_user_favorite_events(user_id: str, limit: int = 100) -> List[Dict[str, Any]]:
+    """
+    获取用户自选事件列表
+
+    Args:
+        user_id: 用户ID
+        limit: 返回条数
+
+    Returns:
+        自选事件列表
+    """
+    pool = await get_pool()
+
+    async with pool.acquire() as conn:
+        async with conn.cursor(aiomysql.DictCursor) as cursor:
+            # 查询用户自选事件（假设有 user_event_favorites 表）
+            query = """
+                SELECT
+                    f.user_id,
+                    f.event_id,
+                    e.title,
+                    e.description,
+                    e.event_date,
+                    e.importance,
+                    e.related_stocks,
+                    e.category,
+                    f.created_at as favorite_time
+                FROM user_event_favorites f
+                INNER JOIN events e ON f.event_id = e.id
+                WHERE f.user_id = %s AND f.is_deleted = 0
+                ORDER BY e.event_date DESC
+                LIMIT %s
+            """
+
+            await cursor.execute(query, [user_id, limit])
+            results = await cursor.fetchall()
+
+            return [convert_row(row) for row in results]
+
+
+async def add_favorite_stock(user_id: str, stock_code: str) -> Dict[str, Any]:
+    """
+    添加自选股
+
+    Args:
+        user_id: 用户ID
+        stock_code: 股票代码
+
+    Returns:
+        操作结果
+    """
+    pool = await get_pool()
+
+    async with pool.acquire() as conn:
+        async with conn.cursor(aiomysql.DictCursor) as cursor:
+            # 检查是否已存在
+            check_query = """
+                SELECT id, is_deleted
+                FROM user_favorites
+                WHERE user_id = %s AND stock_code = %s
+            """
+            await cursor.execute(check_query, [user_id, stock_code])
+            existing = await cursor.fetchone()
+
+            if existing:
+                if existing['is_deleted'] == 1:
+                    # 恢复已删除的记录
+                    update_query = """
+                        UPDATE user_favorites
+                        SET is_deleted = 0, updated_at = NOW()
+                        WHERE id = %s
+                    """
+                    await cursor.execute(update_query, [existing['id']])
+                    return {"success": True, "message": "已恢复自选股"}
+                else:
+                    return {"success": False, "message": "该股票已在自选中"}
+
+            # 插入新记录
+            insert_query = """
+                INSERT INTO user_favorites (user_id, stock_code, created_at, updated_at, is_deleted)
+                VALUES (%s, %s, NOW(), NOW(), 0)
+            """
+            await cursor.execute(insert_query, [user_id, stock_code])
+            return {"success": True, "message": "添加自选股成功"}
+
+
+async def remove_favorite_stock(user_id: str, stock_code: str) -> Dict[str, Any]:
+    """
+    删除自选股
+
+    Args:
+        user_id: 用户ID
+        stock_code: 股票代码
+
+    Returns:
+        操作结果
+    """
+    pool = await get_pool()
+
+    async with pool.acquire() as conn:
+        async with conn.cursor(aiomysql.DictCursor) as cursor:
+            query = """
+                UPDATE user_favorites
+                SET is_deleted = 1, updated_at = NOW()
+                WHERE user_id = %s AND stock_code = %s AND is_deleted = 0
+            """
+            result = await cursor.execute(query, [user_id, stock_code])
+
+            if result > 0:
+                return {"success": True, "message": "删除自选股成功"}
+            else:
+                return {"success": False, "message": "未找到该自选股"}
+
+
+async def add_favorite_event(user_id: str, event_id: int) -> Dict[str, Any]:
+    """
+    添加自选事件
+
+    Args:
+        user_id: 用户ID
+        event_id: 事件ID
+
+    Returns:
+        操作结果
+    """
+    pool = await get_pool()
+
+    async with pool.acquire() as conn:
+        async with conn.cursor(aiomysql.DictCursor) as cursor:
+            # 检查是否已存在
+            check_query = """
+                SELECT id, is_deleted
+                FROM user_event_favorites
+                WHERE user_id = %s AND event_id = %s
+            """
+            await cursor.execute(check_query, [user_id, event_id])
+            existing = await cursor.fetchone()
+
+            if existing:
+                if existing['is_deleted'] == 1:
+                    # 恢复已删除的记录
+                    update_query = """
+                        UPDATE user_event_favorites
+                        SET is_deleted = 0, updated_at = NOW()
+                        WHERE id = %s
+                    """
+                    await cursor.execute(update_query, [existing['id']])
+                    return {"success": True, "message": "已恢复自选事件"}
+                else:
+                    return {"success": False, "message": "该事件已在自选中"}
+
+            # 插入新记录
+            insert_query = """
+                INSERT INTO user_event_favorites (user_id, event_id, created_at, updated_at, is_deleted)
+                VALUES (%s, %s, NOW(), NOW(), 0)
+            """
+            await cursor.execute(insert_query, [user_id, event_id])
+            return {"success": True, "message": "添加自选事件成功"}
+
+
+async def remove_favorite_event(user_id: str, event_id: int) -> Dict[str, Any]:
+    """
+    删除自选事件
+
+    Args:
+        user_id: 用户ID
+        event_id: 事件ID
+
+    Returns:
+        操作结果
+    """
+    pool = await get_pool()
+
+    async with pool.acquire() as conn:
+        async with conn.cursor(aiomysql.DictCursor) as cursor:
+            query = """
+                UPDATE user_event_favorites
+                SET is_deleted = 1, updated_at = NOW()
+                WHERE user_id = %s AND event_id = %s AND is_deleted = 0
+            """
+            result = await cursor.execute(query, [user_id, event_id])
+
+            if result > 0:
+                return {"success": True, "message": "删除自选事件成功"}
+            else:
+                return {"success": False, "message": "未找到该自选事件"}