Spaces:
Sleeping
Sleeping
File size: 3,909 Bytes
551658a | 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 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 | from pydantic import BaseModel
from typing import Optional, List
from datetime import datetime
from enum import Enum
class MediaStatus(str, Enum):
PENDING = "pending"
UPLOADING = "uploading"
SUCCESS = "success"
FAILED = "failed"
class MediaTagBase(BaseModel):
tag_name: str
tag_color: Optional[str] = "#3b82f6"
sort_order: Optional[int] = 0
class MediaTagCreate(MediaTagBase):
pass
class MediaTagUpdate(BaseModel):
tag_name: Optional[str] = None
tag_color: Optional[str] = None
sort_order: Optional[int] = None
class MediaTagResponse(MediaTagBase):
id: int
company_code: str
created_at: datetime
updated_at: datetime
class Config:
from_attributes = True
class MediaTagMappingResponse(BaseModel):
id: int
media_id: str
tag_id: int
created_at: datetime
class Config:
from_attributes = True
class MediaBase(BaseModel):
file_name: str
r2_key: Optional[str] = ""
r2_url: Optional[str] = ""
thumbnail_r2_key: Optional[str] = ""
thumbnail_r2_url: Optional[str] = ""
media_type: Optional[str] = "image"
mime_type: Optional[str] = ""
file_size: Optional[int] = 0
width: Optional[int] = None
height: Optional[int] = None
duration: Optional[float] = None
directory_id: Optional[int] = None
status: Optional[MediaStatus] = MediaStatus.PENDING
error_message: Optional[str] = None
batch_id: Optional[str] = None
tags: Optional[List[int]] = []
class MediaCreate(MediaBase):
pass
class MediaUpdate(BaseModel):
file_name: Optional[str] = None
directory_id: Optional[int] = None
status: Optional[MediaStatus] = None
error_message: Optional[str] = None
tags: Optional[List[int]] = None
class MediaResponse(MediaBase):
id: int
media_id: str
company_code: str
created_at: datetime
updated_at: datetime
is_deleted: Optional[bool] = False
deleted_at: Optional[datetime] = None
class Config:
from_attributes = True
class MediaBatchResponse(BaseModel):
batch_id: str
total: int
successful: int
failed: int
results: List[dict]
class MediaBatchCreate(BaseModel):
files: List[str]
directory_id: Optional[int] = None
class MediaBatchDeleteRequest(BaseModel):
media_ids: List[str]
class MediaBatchUpdateRequest(BaseModel):
media_ids: List[str]
directory_id: Optional[int] = None
tags: Optional[List[int]] = None
class MediaBatchTagRequest(BaseModel):
media_ids: List[str]
tag_ids: List[int]
class MediaSearchRequest(BaseModel):
keyword: Optional[str] = None
media_type: Optional[str] = None
directory_id: Optional[int] = None
tag_ids: Optional[List[int]] = None
start_date: Optional[str] = None
end_date: Optional[str] = None
min_size: Optional[int] = None
max_size: Optional[int] = None
status: Optional[MediaStatus] = None
class MediaDirectoryBase(BaseModel):
name: str
parent_id: Optional[int] = None
path: Optional[str] = ""
level: Optional[int] = 0
sort_order: Optional[int] = 0
cover_r2_key: Optional[str] = None
cover_r2_url: Optional[str] = None
description: Optional[str] = None
is_public: Optional[bool] = False
media_count: Optional[int] = 0
total_size: Optional[int] = 0
class MediaDirectoryCreate(MediaDirectoryBase):
pass
class MediaDirectoryUpdate(BaseModel):
name: Optional[str] = None
parent_id: Optional[int] = None
sort_order: Optional[int] = None
cover_r2_key: Optional[str] = None
cover_r2_url: Optional[str] = None
description: Optional[str] = None
is_public: Optional[bool] = None
class MediaDirectoryResponse(MediaDirectoryBase):
id: int
company_code: str
created_at: datetime
updated_at: datetime
class Config:
from_attributes = True
|