Files
Aislo/B03_FileInput/B03_FileInput_Schema.py
T
eomsangdon 60bbe89edb fix(B03): rerun analysis on reupload
Clear derived browser caches so B05 loads the newly generated route and section data.
2026-08-29 20:17:37 +09:00

154 lines
5.0 KiB
Python

"""B03 파일 입력 요청·응답 검증 모델."""
from pathlib import PurePath
from typing import Any
from pydantic import BaseModel, ConfigDict, Field, model_validator
from config.config_system import UPLOAD_ALLOWED_EXT, UPLOAD_CHUNK_SIZE_BYTES, UPLOAD_MAX_MB
class FileUploadDescriptor(BaseModel):
"""업로드 전에 검증할 단일 파일의 이름과 크기."""
model_config = ConfigDict(extra="forbid", str_strip_whitespace=True)
original_filename: str = Field(min_length=1, max_length=255)
size_bytes: int = Field(gt=0)
@model_validator(mode="after")
def validate_file(self) -> "FileUploadDescriptor":
filename = self.original_filename
if PurePath(filename).name != filename or "/" in filename or "\\" in filename:
raise ValueError("파일명에는 경로가 포함될 수 없습니다.")
extension = PurePath(filename).suffix.lower()
if extension not in UPLOAD_ALLOWED_EXT:
allowed = ", ".join(UPLOAD_ALLOWED_EXT)
raise ValueError(f"허용되지 않은 파일 확장자입니다. 허용 형식: {allowed}")
maximum_bytes = UPLOAD_MAX_MB * 1024 * 1024
if self.size_bytes > maximum_bytes:
raise ValueError(f"파일 크기는 {UPLOAD_MAX_MB}MB를 초과할 수 없습니다.")
return self
class UploadedFileResult(BaseModel):
"""저장 완료된 단일 입력 파일 정보."""
input_file_id: int
original_filename: str
file_type: str
relative_path: str
size_bytes: int
metadata: dict[str, Any]
class FileUploadResponse(BaseModel):
"""B03 다중 파일 업로드 성공 응답."""
status: str = "success"
project_id: str
files: list[UploadedFileResult]
class ChunkSessionCreateRequest(FileUploadDescriptor):
"""청크 업로드 세션 생성 요청."""
chunk_size_bytes: int = Field(default=UPLOAD_CHUNK_SIZE_BYTES, gt=0)
complete_upload: bool = False
# 파일 지문 — 같은 이름으로 **같은 내용**이 다시 올라오는지 전송 전에 가린다.
# 화면이 파일 크기 + 앞·중간·끝 조각으로 만든다([[fileFingerprint]]).
fingerprint: str | None = Field(default=None, max_length=128)
class ChunkSessionCreateResponse(BaseModel):
"""청크 업로드 세션 생성 응답."""
status: str = "success"
project_id: str
upload_session_id: str
original_filename: str
file_size_bytes: int
chunk_size_bytes: int
total_chunks: int
completed_chunks: int = 0
# 이미 같은 내용이 올라와 있어 전송을 건너뛰어도 되는지. true면 세션을 만들지 않는다.
already_uploaded: bool = False
class ChunkUploadResponse(BaseModel):
"""단일 청크 저장 응답."""
status: str = "success"
upload_session_id: str
chunk_index: int
completed_chunks: int
total_chunks: int
chunk_hash: str
class UploadFinalizeRequest(BaseModel):
"""청크 업로드 완료 및 병합 요청."""
model_config = ConfigDict(extra="forbid", str_strip_whitespace=True)
session_id: str = Field(min_length=1, max_length=36)
total_chunks: int = Field(gt=0)
complete_upload: bool = True
# 세션 생성 때 쓴 지문을 그대로 다시 받아 입력 파일에 남긴다. 다음에 같은 파일이
# 올라오면 이 값으로 전송을 건너뛴다(upload_sessions에 컬럼을 더하지 않으려는 선택).
fingerprint: str | None = Field(default=None, max_length=128)
class UploadStatusResponse(BaseModel):
"""청크 업로드 상태 조회 응답."""
status: str = "success"
upload_session_id: str
upload_status: str
original_filename: str
file_size_bytes: int
chunk_size_bytes: int
total_chunks: int
completed_chunks: int
completed_chunk_indexes: list[int]
class UploadOverviewFile(BaseModel):
"""재접속 현황 — 업로드 완료된 입력 파일 한 건(서버 정본)."""
input_file_id: int
file_type: str
original_filename: str
file_size_mb: float
status: str
uploaded_at: str | None = None
class UploadOverviewSession(BaseModel):
"""재접속 현황 — 중단된 청크 업로드 세션 한 건(이어올리기 안내용)."""
upload_session_id: str
original_filename: str
file_size_bytes: int
total_chunks: int
completed_chunks: int
progress_percent: float
updated_at: str | None = None
class UploadOverviewResponse(BaseModel):
"""B03 재접속 시 업로드 현황 — localStorage가 아니라 이 응답이 정본이다.
`required_complete`는 필수 파일(LAS/LAZ 1 + csv·prj·tfw)이 모두 업로드된 상태,
`analysis_complete`는 WF1 분석(stage 1)까지 끝난 상태다. 두 값이 모두 참이면
화면은 "완료" 배지를 띄우고, 완료 슬롯에 재업로드하면 교체 경고 모달을 띄운다.
"""
status: str = "success"
files: list[UploadOverviewFile]
pending_sessions: list[UploadOverviewSession]
required_complete: bool
analysis_complete: bool