Add _rowVariant and _cellVariants

This commit is contained in:
long2ice
2020-07-23 15:28:31 +08:00
parent c4997b9c21
commit 08d0216a9f
5 changed files with 60 additions and 5 deletions

View File

@ -3,6 +3,7 @@ import io
import xlsxwriter
from fastapi import APIRouter, Depends
from fastapi.responses import UJSONResponse
from pydantic import Field
from starlette.responses import StreamingResponse
from starlette.status import HTTP_409_CONFLICT
from tortoise import Model
@ -78,9 +79,13 @@ async def get_resource(
creator = pydantic_model_creator(
model, include=resource.resource_fields.keys(), exclude=model._meta.m2m_fields
)
return GetManyOut(
total=await qs.count(), data=list(map(lambda x: creator.from_orm(x).dict(), result))
)
data = []
for item in result:
item_dict = creator.from_orm(item).dict()
item_dict["_rowVariant"] = item_dict.pop("rowVariant", None)
item_dict["_cellVariants"] = item_dict.pop("cellVariants", None)
data.append(item_dict)
return GetManyOut(total=await qs.count(), data=data)
@router.get("/{resource}/form", dependencies=[Depends(read_checker)])