Pywss(发音 /piːwaɪz/,类似 p~whys)是一个轻量级的 Python Web 框架,它基于 Python3.6+ 特性构建。
与 Flask、Django 等主流框架不同的是,Pywss 的底层并没有实现 WSGI 接口协议。 其编程风格也更类似于 Gin、Iris 等框架,因此对于熟悉这些框架的开发者来说,Pywss 是一个非常值得探索的项目。
其关键特性有:
- 简单:拒绝海量参数,减少心智负担。了解上下文
pywss.Context
即刻启程。 - 快速:引入线程池机制,减少并发场景下线程创建/销毁开销。
- 优雅:
ctx.next
真的太优雅了。如果你也和我一样喜欢,那我觉得这件事情,泰裤辣!! - 标准化:集成了部分 OpenAPI(Swagger)能力,方便开发者快速生成 API 文档并进行调试。
- 支持WebSocket:开箱即用的 WebSocket 能力。
- 接口测试:开箱即用的 API 测试模块,不启动服务也能测试接口功能辣!
- MCP PRO:一站式集成 SSE、StreamHTTP 和 MCPO 协议,助你轻松构建多 MCP 工具🔥
在线文档 https://czasg.github.io/pywss/
pip3 install pywss
首先创建 main.py
文件,并写入以下代码:
import time
import pywss
def log_handler(ctx: pywss.Context):
start_time = time.time()
ctx.next()
print(
f"Route: {ctx.route}, "
f"Method: {ctx.method}, "
f"Status: {ctx.response_status_code}, "
f"Time: {time.time() - start_time:.3f}s"
)
def handler(ctx: pywss.Context):
ctx.write("hello~")
def main():
app = pywss.App()
app.get("/hello", handler) # curl localhost:8080/hello
app.any("*", log_handler, handler) # curl -X POST localhost:8080/hello
app.run()
if __name__ == '__main__':
main()
接着启动服务:
python3 main.py
至此,一个简单的 web 应用服务就完成了。
要快速构建 MCP 服务,只需继承 pywss.mcp.MCPServer
并遵循以下规则:
核心约束:
- 所有接口方法必须以
tool_
开头(如tool_query_user
) - 必须使用
@pywss.openapi.docs
声明请求参数,且request
必须从 pydantic.BaseModel 继承(自动生成 OpenAPI 文档)
请求处理:
- 通过
ctx.data.req
直接获取结构化请求体 - 使用
self.handle_success(ctx, data)
返回成功响应 - 使用
self.handle_error(ctx, code, message)
返回标准错误
# coding: utf-8
import pywss
from pydantic import BaseModel
from pywss.mcp import MCPServer
class DomainReq(BaseModel): # 定义 DomainReq 请求,必须从 pydantic.BaseModel 继承
domain: str
class DomainMCPServer(MCPServer): # 定义 DomainMCPServer 服务,必须从 pywss.mcp.MCPServer 继承
@pywss.openapi.docs(description="获取单个域名服务", request=DomainReq) # required,工具及其参数说明
def tool_get_domain(self, ctx: pywss.Context):
req: DomainReq = ctx.data.req # 框架已经封装好了请求,可以从 ctx.data.req 直接获取使用,异常请求会被拦截
self.handle_success(ctx, { # handle_success 封装了 jsonrpc2.0 输出规范
"domain": req.domain,
"color": req.color
})
class LogReq(BaseModel):
traceId: str
class LogMCPServer(MCPServer):
@pywss.openapi.docs(description="获取单个trace日志", request=LogReq)
def tool_get_trace_log(self, ctx: pywss.Context):
req: LogReq = ctx.data.req
self.handle_success(ctx, {
"traceId": req.traceId,
})
domainMCPServer = DomainMCPServer()
logMCPServer = LogMCPServer()
app = pywss.App()
app.openapi() # 开启 OpenAPI 文档
domainMCPServer.mount(app.group("/api/v1/domain")) # 挂载 MCP 服务,同时指定路由
logMCPServer.mount(app.group("/api/v1/log")) # 挂载 MCP 服务,同时指定路由
app.run()
接着启动服务:
python3 main.py
协议类型 | 服务类 | 请求方法 | 端点路径格式 | 示例路径 |
---|---|---|---|---|
SSE | domainMCPServer |
GET |
/api/v1/domain/sse |
GET /api/v1/domain/sse |
logMCPServer |
GET |
/api/v1/log/sse |
GET /api/v1/log/sse |
|
StreamHTTP | domainMCPServer |
POST |
/api/v1/domain/mcp |
POST /api/v1/domain/mcp |
logMCPServer |
POST |
/api/v1/log/mcp |
POST /api/v1/log/mcp |
|
MCPO | domainMCPServer |
POST |
/api/v1/domain/tools/{tool_name} |
POST /api/v1/domain/tools/get_domain |
logMCPServer |
POST |
/api/v1/log/tools/{tool_name} |
POST /api/v1/log/tools/get_trace_log |
更多功能见在线文档。