mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-10-02 00:27:18 +00:00
Wu Clan
15a9527866
* Refactor the service layer db session call method * Fix lint * Update paging interfaces * Fix some errors
113 lines
3.7 KiB
Python
113 lines
3.7 KiB
Python
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends, Path, Query
|
|
|
|
from backend.app.admin.schema.data_rule import (
|
|
CreateDataRuleParam,
|
|
DeleteDataRuleParam,
|
|
GetDataRuleColumnDetail,
|
|
GetDataRuleDetail,
|
|
UpdateDataRuleParam,
|
|
)
|
|
from backend.app.admin.service.data_rule_service import data_rule_service
|
|
from backend.common.pagination import DependsPagination, PageData
|
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
|
from backend.common.security.jwt import DependsJwtAuth
|
|
from backend.common.security.permission import RequestPermission
|
|
from backend.common.security.rbac import DependsRBAC
|
|
from backend.database.db import CurrentSession, CurrentSessionTransaction
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get('/models', summary='获取数据规则可用模型', dependencies=[DependsJwtAuth])
|
|
async def get_data_rule_models() -> ResponseSchemaModel[list[str]]:
|
|
models = await data_rule_service.get_models()
|
|
return response_base.success(data=models)
|
|
|
|
|
|
@router.get('/models/{model}/columns', summary='获取数据规则可用模型列', dependencies=[DependsJwtAuth])
|
|
async def get_data_rule_model_columns(
|
|
model: Annotated[str, Path(description='模型名称')],
|
|
) -> ResponseSchemaModel[list[GetDataRuleColumnDetail]]:
|
|
models = await data_rule_service.get_columns(model=model)
|
|
return response_base.success(data=models)
|
|
|
|
|
|
@router.get('/all', summary='获取所有数据规则', dependencies=[DependsJwtAuth])
|
|
async def get_all_data_rules(db: CurrentSession) -> ResponseSchemaModel[list[GetDataRuleDetail]]:
|
|
data = await data_rule_service.get_all(db=db)
|
|
return response_base.success(data=data)
|
|
|
|
|
|
@router.get('/{pk}', summary='获取数据规则详情', dependencies=[DependsJwtAuth])
|
|
async def get_data_rule(
|
|
db: CurrentSession,
|
|
pk: Annotated[int, Path(description='数据规则 ID')],
|
|
) -> ResponseSchemaModel[GetDataRuleDetail]:
|
|
data = await data_rule_service.get(db=db, pk=pk)
|
|
return response_base.success(data=data)
|
|
|
|
|
|
@router.get(
|
|
'',
|
|
summary='分页获取所有数据规则',
|
|
dependencies=[
|
|
DependsJwtAuth,
|
|
DependsPagination,
|
|
],
|
|
)
|
|
async def get_data_rules_paginated(
|
|
db: CurrentSession,
|
|
name: Annotated[str | None, Query(description='规则名称')] = None,
|
|
) -> ResponseSchemaModel[PageData[GetDataRuleDetail]]:
|
|
page_data = await data_rule_service.get_list(db=db, name=name)
|
|
return response_base.success(data=page_data)
|
|
|
|
|
|
@router.post(
|
|
'',
|
|
summary='创建数据规则',
|
|
dependencies=[
|
|
Depends(RequestPermission('data:rule:add')),
|
|
DependsRBAC,
|
|
],
|
|
)
|
|
async def create_data_rule(db: CurrentSessionTransaction, obj: CreateDataRuleParam) -> ResponseModel:
|
|
await data_rule_service.create(db=db, obj=obj)
|
|
return response_base.success()
|
|
|
|
|
|
@router.put(
|
|
'/{pk}',
|
|
summary='更新数据规则',
|
|
dependencies=[
|
|
Depends(RequestPermission('data:rule:edit')),
|
|
DependsRBAC,
|
|
],
|
|
)
|
|
async def update_data_rule(
|
|
db: CurrentSessionTransaction,
|
|
pk: Annotated[int, Path(description='数据规则 ID')],
|
|
obj: UpdateDataRuleParam,
|
|
) -> ResponseModel:
|
|
count = await data_rule_service.update(db=db, pk=pk, obj=obj)
|
|
if count > 0:
|
|
return response_base.success()
|
|
return response_base.fail()
|
|
|
|
|
|
@router.delete(
|
|
'',
|
|
summary='批量删除数据规则',
|
|
dependencies=[
|
|
Depends(RequestPermission('data:rule:del')),
|
|
DependsRBAC,
|
|
],
|
|
)
|
|
async def delete_data_rules(db: CurrentSessionTransaction, obj: DeleteDataRuleParam) -> ResponseModel:
|
|
count = await data_rule_service.delete(db=db, obj=obj)
|
|
if count > 0:
|
|
return response_base.success()
|
|
return response_base.fail()
|