Rebuild the docs around tested examples; shrink README.v2.md to a pitch (#2978)
This commit is contained in:
parent
4caa41f6d5
commit
411a6d3980
1 changed files with 9510 additions and 2639 deletions
44
docs_src/lifespan/tutorial002.py
Normal file
44
docs_src/lifespan/tutorial002.py
Normal file
|
|
@ -0,0 +1,44 @@
|
|||
from collections.abc import AsyncIterator
|
||||
from contextlib import asynccontextmanager
|
||||
from dataclasses import dataclass
|
||||
|
||||
from mcp.server import MCPServer
|
||||
from mcp.server.mcpserver import Context
|
||||
|
||||
|
||||
class Database:
|
||||
def __init__(self) -> None:
|
||||
self.connected = False
|
||||
|
||||
async def connect(self) -> None:
|
||||
self.connected = True
|
||||
|
||||
async def disconnect(self) -> None:
|
||||
self.connected = False
|
||||
|
||||
|
||||
@dataclass
|
||||
class AppContext:
|
||||
db: Database
|
||||
|
||||
|
||||
database = Database()
|
||||
|
||||
|
||||
@asynccontextmanager
|
||||
async def app_lifespan(server: MCPServer) -> AsyncIterator[AppContext]:
|
||||
await database.connect()
|
||||
try:
|
||||
yield AppContext(db=database)
|
||||
finally:
|
||||
await database.disconnect()
|
||||
|
||||
|
||||
mcp = MCPServer("Bookshop", lifespan=app_lifespan)
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
def database_status(ctx: Context[AppContext]) -> str:
|
||||
"""Report whether the database connection is up."""
|
||||
db = ctx.request_context.lifespan_context.db
|
||||
return "connected" if db.connected else "disconnected"
|
||||
Loading…
Reference in a new issue