mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 13:12:24 +00:00
Wu Clan
7553ccf14d
* Update return model of query interface * Update tree data return type hints * Update code generation and task return types * Fix inconsistency between ORM query result and schema
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends, Path, Query
|
|
from fastapi.responses import StreamingResponse
|
|
|
|
from backend.app.generator.conf import generator_settings
|
|
from backend.app.generator.schema.gen import ImportParam
|
|
from backend.app.generator.service.gen_service import gen_service
|
|
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
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get('/tables', summary='获取数据库表')
|
|
async def get_all_tables(
|
|
table_schema: Annotated[str, Query(..., description='数据库名')] = 'fba',
|
|
) -> ResponseSchemaModel[list[str]]:
|
|
data = await gen_service.get_tables(table_schema=table_schema)
|
|
return response_base.success(data=data)
|
|
|
|
|
|
@router.post(
|
|
'/import',
|
|
summary='导入代码生成业务和模型列',
|
|
dependencies=[
|
|
Depends(RequestPermission('gen:code:import')),
|
|
DependsRBAC,
|
|
],
|
|
)
|
|
async def import_table(obj: ImportParam) -> ResponseModel:
|
|
await gen_service.import_business_and_model(obj=obj)
|
|
return response_base.success()
|
|
|
|
|
|
@router.get('/preview/{pk}', summary='生成代码预览', dependencies=[DependsJwtAuth])
|
|
async def preview_code(pk: Annotated[int, Path(..., description='业务ID')]) -> ResponseSchemaModel[dict[str, bytes]]:
|
|
data = await gen_service.preview(pk=pk)
|
|
return response_base.success(data=data)
|
|
|
|
|
|
@router.get('/generate/{pk}/path', summary='获取代码生成路径', dependencies=[DependsJwtAuth])
|
|
async def generate_path(pk: Annotated[int, Path(..., description='业务ID')]) -> ResponseSchemaModel[list[str]]:
|
|
data = await gen_service.get_generate_path(pk=pk)
|
|
return response_base.success(data=data)
|
|
|
|
|
|
@router.post(
|
|
'/generate/{pk}',
|
|
summary='代码生成',
|
|
description='文件磁盘写入，请谨慎操作',
|
|
dependencies=[
|
|
Depends(RequestPermission('gen:code:generate')),
|
|
DependsRBAC,
|
|
],
|
|
)
|
|
async def generate_code(pk: Annotated[int, Path(..., description='业务ID')]) -> ResponseModel:
|
|
await gen_service.generate(pk=pk)
|
|
return response_base.success()
|
|
|
|
|
|
@router.get('/download/{pk}', summary='下载代码', dependencies=[DependsJwtAuth])
|
|
async def download_code(pk: Annotated[int, Path(..., description='业务ID')]):
|
|
bio = await gen_service.download(pk=pk)
|
|
return StreamingResponse(
|
|
bio,
|
|
media_type='application/x-zip-compressed',
|
|
headers={'Content-Disposition': f'attachment; filename={generator_settings.DOWNLOAD_ZIP_FILENAME}.zip'},
|
|
)
|