108 lines
2.6 KiB
Python
108 lines
2.6 KiB
Python
|
||
from pydantic import BaseModel, EmailStr
|
||
from typing import Optional, Union, List
|
||
import datetime
|
||
|
||
class LoginRequest(BaseModel):
|
||
username: str
|
||
password: str
|
||
|
||
class LoginResponse(BaseModel):
|
||
user_id: int
|
||
username: str
|
||
caption: str
|
||
email: EmailStr
|
||
token: str
|
||
role_id: int
|
||
|
||
class UserInfo(BaseModel):
|
||
user_id: int
|
||
username: str
|
||
caption: str
|
||
email: EmailStr
|
||
created_at: datetime.datetime
|
||
meetings_created: int
|
||
meetings_attended: int
|
||
|
||
class UserListResponse(BaseModel):
|
||
users: list[UserInfo]
|
||
total: int
|
||
|
||
class CreateUserRequest(BaseModel):
|
||
username: str
|
||
password: Optional[str] = None
|
||
caption: str
|
||
email: EmailStr
|
||
role_id: int
|
||
|
||
class UpdateUserRequest(BaseModel):
|
||
username: Optional[str] = None
|
||
caption: Optional[str] = None
|
||
email: Optional[str] = None
|
||
role_id: Optional[int] = None
|
||
|
||
class AttendeeInfo(BaseModel):
|
||
user_id: int
|
||
caption: str
|
||
|
||
class TranscriptionTaskStatus(BaseModel):
|
||
task_id: str
|
||
status: str # 'pending', 'processing', 'completed', 'failed'
|
||
progress: int # 0-100
|
||
meeting_id: int
|
||
created_at: Optional[str] = None
|
||
updated_at: Optional[str] = None
|
||
completed_at: Optional[str] = None
|
||
error_message: Optional[str] = None
|
||
|
||
class Meeting(BaseModel):
|
||
meeting_id: int
|
||
title: str
|
||
meeting_time: Optional[datetime.datetime]
|
||
summary: Optional[str]
|
||
created_at: datetime.datetime
|
||
attendees: Union[List[str], List[AttendeeInfo]] # Support both formats
|
||
creator_id: int
|
||
creator_username: str
|
||
audio_file_path: Optional[str] = None
|
||
transcription_status: Optional[TranscriptionTaskStatus] = None
|
||
|
||
class TranscriptSegment(BaseModel):
|
||
segment_id: int
|
||
meeting_id: int
|
||
speaker_id: Optional[int] = None # AI解析的原始结果
|
||
speaker_tag: str
|
||
start_time_ms: int
|
||
end_time_ms: int
|
||
text_content: str
|
||
|
||
class CreateMeetingRequest(BaseModel):
|
||
user_id: int
|
||
title: str
|
||
meeting_time: Optional[datetime.datetime]
|
||
attendee_ids: list[int]
|
||
|
||
class UpdateMeetingRequest(BaseModel):
|
||
title: str
|
||
meeting_time: Optional[datetime.datetime]
|
||
summary: Optional[str]
|
||
attendee_ids: list[int]
|
||
|
||
class SpeakerTagUpdateRequest(BaseModel):
|
||
speaker_id: int # 使用原始speaker_id（整数）
|
||
new_tag: str
|
||
|
||
class BatchSpeakerTagUpdateRequest(BaseModel):
|
||
updates: List[SpeakerTagUpdateRequest]
|
||
|
||
class TranscriptUpdateRequest(BaseModel):
|
||
segment_id: int
|
||
text_content: str
|
||
|
||
class BatchTranscriptUpdateRequest(BaseModel):
|
||
updates: List[TranscriptUpdateRequest]
|
||
|
||
class PasswordChangeRequest(BaseModel):
|
||
old_password: str
|
||
new_password: str
|