mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-10-04 01:10:36 +00:00
zhangtao
013ba8074a
移除SQLITE_DB_NAME配置，统一使用DATABASE_NAME作为数据库名称 重构CRUD基类，增加out_schema参数用于序列化输出 优化数据库表查询逻辑，支持多种数据库类型 修复序列化工具类的方法签名和实现
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from typing import Dict, List, Optional, Sequence
|
|
|
|
from app.core.base_crud import CRUDBase
|
|
from app.api.v1.module_system.auth.schema import AuthSchema
|
|
from .model import McpModel
|
|
from .schema import McpCreateSchema, McpUpdateSchema
|
|
|
|
|
|
class McpCRUD(CRUDBase[McpModel, McpCreateSchema, McpUpdateSchema]):
|
|
"""MCP 服务器数据层"""
|
|
|
|
def __init__(self, auth: AuthSchema) -> None:
|
|
"""初始化CRUD"""
|
|
self.auth = auth
|
|
super().__init__(model=McpModel, auth=auth)
|
|
|
|
async def get_by_id_crud(self, id: int) -> Optional[McpModel]:
|
|
"""详情"""
|
|
return await self.get(id=id)
|
|
|
|
async def get_by_name_crud(self, name: str) -> Optional[McpModel]:
|
|
"""通过名称获取MCP服务器"""
|
|
return await self.get(name=name)
|
|
|
|
async def get_list_crud(self, search: Optional[Dict] = None, order_by: Optional[List[Dict[str, str]]] = None) -> Sequence[McpModel]:
|
|
"""列表查询"""
|
|
return await self.list(search=search or {}, order_by=order_by or [{'id': 'asc'}])
|
|
|
|
async def create_crud(self, data: McpCreateSchema) -> Optional[McpModel]:
|
|
"""创建"""
|
|
return await self.create(data=data)
|
|
|
|
async def update_crud(self, id: int, data: McpUpdateSchema) -> Optional[McpModel]:
|
|
"""更新"""
|
|
return await self.update(id=id, data=data)
|
|
|
|
async def delete_crud(self, ids: List[int]) -> None:
|
|
"""批量删除"""
|
|
return await self.delete(ids=ids)
|