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 fastmcp import FastMCP
|
||||||
from spider.mail_qq import start as a1
|
from spider.mail_qq import start as a1
|
||||||
import subprocess
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import tempfile
|
|
||||||
import os
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
mcp = FastMCP("spider-server")
|
mcp = FastMCP("spider-server")
|
||||||
logging.basicConfig(level=logging.INFO)
|
logging.basicConfig(level=logging.INFO)
|
||||||
# 日志会输出到 stderr,不会污染 stdout
|
# 日志会输出到 stderr,不会污染 stdout
|
||||||
logging.info("服务器启动")
|
logging.info("服务器启动")
|
||||||
@mcp.tool()
|
|
||||||
|
|
||||||
|
# @mcp.tool()
|
||||||
def mail_qq_spider(account: str) -> str:
|
def mail_qq_spider(account: str) -> str:
|
||||||
"""qq邮箱爬虫,第一个参数是用户名"""
|
"""qq邮箱爬虫,第一个参数是用户名"""
|
||||||
return a1(account)
|
return a1(account)
|
||||||
|
|
||||||
# 添加一个简单的工具
|
|
||||||
@mcp.tool()
|
|
||||||
def hello(name: str) -> str:
|
|
||||||
"""Say hello to someone"""
|
|
||||||
return f"Hello, {name}!"
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@mcp.tool()
|
@mcp.tool()
|
||||||
async def start_vnc_server(index: int) -> str:
|
async def start_vnc_server(index: int) -> str:
|
||||||
"""启动VNC服务
|
"""启动VNC服务
|
||||||
|
|
@ -94,7 +84,11 @@ async def close_vnc_server(index: int) -> str:
|
||||||
# 单参数模板
|
# 单参数模板
|
||||||
@mcp.resource("data://spider_code")
|
@mcp.resource("data://spider_code")
|
||||||
def get_spider_code_list() -> str:
|
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__":
|
if __name__ == "__main__":
|
||||||
mcp.run(transport="sse")
|
mcp.run(transport="sse")
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue