Spaces:
Sleeping
Sleeping
File size: 2,951 Bytes
57528c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 | from typing import Optional
from fastapi import APIRouter, File, Form, Query, UploadFile
from app.core.errors import not_found
from app.schemas.common import StatusMessage
from app.schemas.data import (
GroupInfoResponse,
SaveRdsRequest,
SaveRdsResponse,
UpdateDataFileRequest,
UpdateDataFileResponse,
UploadResponse,
)
from app.schemas.project import SelectDataFileResponse
from app.services import workspace_service
router = APIRouter(prefix="/projects/{project_id}/data", tags=["data"])
@router.post("/upload/", response_model=UploadResponse)
def upload_data(
project_id: str,
files: UploadFile = File(...),
group_index: str = Form("2"),
wavenumber_range: Optional[str] = Form(default=None, alias="Wavenumber range"),
) -> dict:
group_index_value = workspace_service.validate_group_index(group_index)
return workspace_service.upload_data_file(project_id, files, group_index_value, wavenumber_range)
@router.get("/group_info/", response_model=GroupInfoResponse)
def group_info(
project_id: str,
group_index: Optional[str] = Query(default=None),
) -> dict:
workspace_service.validate_group_index(group_index or "2")
try:
return workspace_service.get_group_info(project_id)
except KeyError:
not_found("project_not_found", "项目不存在")
@router.post("/save_rds/", response_model=SaveRdsResponse)
def save_rds(project_id: str, payload: SaveRdsRequest) -> dict:
try:
return workspace_service.save_rds_data(project_id, payload.group_order)
except KeyError:
not_found("project_not_found", "项目不存在")
@router.post("/{file_id}/select/", response_model=SelectDataFileResponse)
def select_data_file(project_id: str, file_id: int) -> dict:
try:
return workspace_service.select_data_file(project_id, file_id)
except KeyError as exc:
if str(exc).strip("'") == "file_not_found":
not_found("file_not_found", "数据文件不存在")
not_found("project_not_found", "项目不存在")
@router.post("/{file_id}/update/", response_model=UpdateDataFileResponse)
def update_data_file(
project_id: str,
file_id: int,
payload: UpdateDataFileRequest,
) -> dict:
try:
return workspace_service.update_data_file(project_id, file_id, payload.wavenumber_range)
except KeyError as exc:
if str(exc).strip("'") == "file_not_found":
not_found("file_not_found", "数据文件不存在")
not_found("project_not_found", "项目不存在")
@router.delete("/{file_id}/delete/", response_model=StatusMessage)
def delete_data_file(project_id: str, file_id: int) -> dict:
try:
return workspace_service.delete_data_file(project_id, file_id)
except KeyError as exc:
if str(exc).strip("'") == "file_not_found":
not_found("file_not_found", "数据文件不存在")
not_found("project_not_found", "项目不存在")
|