ad5c8be0a3
Add missing file upload functionality to V2 API that was removed during V1 to V2 migration. Update frontend to use v2 API endpoints. Backend changes: - Add /api/v2/upload endpoint in main.py for file uploads - Import necessary dependencies (UploadFile, hashlib, TaskFile) - Upload endpoint creates task, saves file, and returns task info - Add UploadResponse schema to task.py schemas - Update tasks router imports for consistency Frontend changes: - Update API_VERSION from 'v1' to 'v2' in api.ts - Update UploadResponse type to match V2 API response format (task_id instead of batch_id, single file instead of array) This fixes the 404 error when uploading files from the frontend. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
120 lines
3.1 KiB
Python
120 lines
3.1 KiB
Python
"""
|
|
Tool_OCR - Task Management Schemas
|
|
"""
|
|
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, Field
|
|
from enum import Enum
|
|
|
|
|
|
class TaskStatusEnum(str, Enum):
|
|
"""Task status enumeration"""
|
|
PENDING = "pending"
|
|
PROCESSING = "processing"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
|
|
|
|
class TaskCreate(BaseModel):
|
|
"""Task creation request"""
|
|
filename: Optional[str] = Field(None, description="Original filename")
|
|
file_type: Optional[str] = Field(None, description="File MIME type")
|
|
|
|
|
|
class TaskUpdate(BaseModel):
|
|
"""Task update request"""
|
|
status: Optional[TaskStatusEnum] = None
|
|
error_message: Optional[str] = None
|
|
processing_time_ms: Optional[int] = None
|
|
result_json_path: Optional[str] = None
|
|
result_markdown_path: Optional[str] = None
|
|
result_pdf_path: Optional[str] = None
|
|
|
|
|
|
class TaskFileResponse(BaseModel):
|
|
"""Task file response schema"""
|
|
id: int
|
|
original_name: Optional[str] = None
|
|
stored_path: Optional[str] = None
|
|
file_size: Optional[int] = None
|
|
mime_type: Optional[str] = None
|
|
file_hash: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TaskResponse(BaseModel):
|
|
"""Task response schema"""
|
|
id: int
|
|
user_id: int
|
|
task_id: str
|
|
filename: Optional[str] = None
|
|
file_type: Optional[str] = None
|
|
status: TaskStatusEnum
|
|
result_json_path: Optional[str] = None
|
|
result_markdown_path: Optional[str] = None
|
|
result_pdf_path: Optional[str] = None
|
|
error_message: Optional[str] = None
|
|
processing_time_ms: Optional[int] = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
completed_at: Optional[datetime] = None
|
|
file_deleted: bool = False
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TaskDetailResponse(TaskResponse):
|
|
"""Detailed task response with files"""
|
|
files: List[TaskFileResponse] = []
|
|
|
|
|
|
class TaskListResponse(BaseModel):
|
|
"""Paginated task list response"""
|
|
tasks: List[TaskResponse]
|
|
total: int
|
|
page: int
|
|
page_size: int
|
|
has_more: bool
|
|
|
|
|
|
class TaskStatsResponse(BaseModel):
|
|
"""User task statistics"""
|
|
total: int
|
|
pending: int
|
|
processing: int
|
|
completed: int
|
|
failed: int
|
|
|
|
|
|
class TaskHistoryQuery(BaseModel):
|
|
"""Task history query parameters"""
|
|
status: Optional[TaskStatusEnum] = None
|
|
filename: Optional[str] = None
|
|
date_from: Optional[datetime] = None
|
|
date_to: Optional[datetime] = None
|
|
page: int = Field(default=1, ge=1)
|
|
page_size: int = Field(default=50, ge=1, le=100)
|
|
order_by: str = Field(default="created_at")
|
|
order_desc: bool = Field(default=True)
|
|
|
|
|
|
class UploadFileInfo(BaseModel):
|
|
"""Uploaded file information"""
|
|
filename: str
|
|
file_size: int
|
|
file_type: str
|
|
|
|
|
|
class UploadResponse(BaseModel):
|
|
"""File upload response"""
|
|
task_id: str = Field(..., description="Created task ID")
|
|
filename: str = Field(..., description="Original filename")
|
|
file_size: int = Field(..., description="File size in bytes")
|
|
file_type: str = Field(..., description="File MIME type")
|
|
status: TaskStatusEnum = Field(..., description="Initial task status")
|