update start.sh
This commit is contained in:
parent
d74e0a00bf
commit
b52341b6ad
22
app.py
22
app.py
|
|
@ -1,28 +1,18 @@
|
|||
from fastmcp import FastMCP
|
||||
from spider.mail_qq import start as a1
|
||||
import subprocess
|
||||
import asyncio
|
||||
import tempfile
|
||||
import os
|
||||
import logging
|
||||
|
||||
mcp = FastMCP("spider-server")
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
# 日志会输出到 stderr,不会污染 stdout
|
||||
logging.info("服务器启动")
|
||||
@mcp.tool()
|
||||
|
||||
|
||||
# @mcp.tool()
|
||||
def mail_qq_spider(account: str) -> str:
|
||||
"""qq邮箱爬虫,第一个参数是用户名"""
|
||||
return a1(account)
|
||||
|
||||
# 添加一个简单的工具
|
||||
@mcp.tool()
|
||||
def hello(name: str) -> str:
|
||||
"""Say hello to someone"""
|
||||
return f"Hello, {name}!"
|
||||
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def start_vnc_server(index: int) -> str:
|
||||
"""启动VNC服务
|
||||
|
|
@ -94,7 +84,11 @@ async def close_vnc_server(index: int) -> str:
|
|||
# 单参数模板
|
||||
@mcp.resource("data://spider_code")
|
||||
def get_spider_code_list() -> str:
|
||||
return f'[{"spider_code": "01","name": "阳光采购爬虫"}, {"spider_code": "02","name": "QQ邮箱爬虫"},]'
|
||||
spider_list = [
|
||||
{"spider_code": "01","name": "阳光采购爬虫"},
|
||||
{"spider_code": "02","name": "QQ邮箱爬虫"},
|
||||
]
|
||||
return json.dumps(spider_list, ensure_ascii=False)
|
||||
# 运行服务器
|
||||
if __name__ == "__main__":
|
||||
mcp.run(transport="sse")
|
||||
|
|
|
|||
Loading…
Reference in New Issue