110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, UploadFile, File, Form
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy.exc import IntegrityError
|
|
from typing import List
|
|
|
|
from app.db.database import get_db
|
|
from app.core.deps import require_admin
|
|
from app.models.models import Collection, Character
|
|
from app.schemas.schemas import CollectionOut
|
|
from app.services.storage import upload_image, delete_object_by_url
|
|
|
|
router = APIRouter(dependencies=[Depends(require_admin)])
|
|
|
|
ALLOWED_TYPES = {"image/jpeg", "image/png", "image/webp", "image/gif"}
|
|
|
|
|
|
@router.post("/collections", response_model=CollectionOut, status_code=201)
|
|
async def create_collection(
|
|
name: str = Form(...),
|
|
files: List[UploadFile] = File(...),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
name = name.strip()
|
|
if not name:
|
|
raise HTTPException(status_code=400, detail="Collection name required")
|
|
if not files:
|
|
raise HTTPException(status_code=400, detail="At least one image is required")
|
|
|
|
collection = Collection(name=name)
|
|
db.add(collection)
|
|
try:
|
|
db.flush()
|
|
except IntegrityError:
|
|
db.rollback()
|
|
raise HTTPException(status_code=409, detail="A collection with this name already exists")
|
|
|
|
for f in files:
|
|
if f.content_type not in ALLOWED_TYPES:
|
|
db.rollback()
|
|
raise HTTPException(
|
|
status_code=415,
|
|
detail=f"Unsupported file type: {f.content_type}",
|
|
)
|
|
data = await f.read()
|
|
_, url = upload_image(data, f.filename or "image", f.content_type or "")
|
|
char_name = (f.filename or "image").rsplit(".", 1)[0]
|
|
db.add(
|
|
Character(
|
|
name=char_name,
|
|
filename=f.filename or "image",
|
|
s3_url=url,
|
|
collection_id=collection.id,
|
|
)
|
|
)
|
|
|
|
db.commit()
|
|
db.refresh(collection)
|
|
return collection
|
|
|
|
|
|
@router.post("/collections/{collection_id}/characters", response_model=CollectionOut)
|
|
async def add_characters(
|
|
collection_id: int,
|
|
files: List[UploadFile] = File(...),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
collection = db.get(Collection, collection_id)
|
|
if not collection:
|
|
raise HTTPException(status_code=404, detail="Collection not found")
|
|
for f in files:
|
|
if f.content_type not in ALLOWED_TYPES:
|
|
raise HTTPException(status_code=415, detail=f"Unsupported file type: {f.content_type}")
|
|
data = await f.read()
|
|
_, url = upload_image(data, f.filename or "image", f.content_type or "")
|
|
char_name = (f.filename or "image").rsplit(".", 1)[0]
|
|
db.add(
|
|
Character(
|
|
name=char_name,
|
|
filename=f.filename or "image",
|
|
s3_url=url,
|
|
collection_id=collection.id,
|
|
)
|
|
)
|
|
db.commit()
|
|
db.refresh(collection)
|
|
return collection
|
|
|
|
|
|
@router.delete("/collections/{collection_id}", status_code=204)
|
|
def delete_collection(collection_id: int, db: Session = Depends(get_db)):
|
|
collection = db.get(Collection, collection_id)
|
|
if not collection:
|
|
raise HTTPException(status_code=404, detail="Collection not found")
|
|
urls = [c.s3_url for c in collection.characters]
|
|
db.delete(collection)
|
|
db.commit()
|
|
for u in urls:
|
|
delete_object_by_url(u)
|
|
|
|
|
|
@router.delete("/characters/{character_id}", status_code=204)
|
|
def delete_character(character_id: int, db: Session = Depends(get_db)):
|
|
ch = db.get(Character, character_id)
|
|
if not ch:
|
|
raise HTTPException(status_code=404, detail="Character not found")
|
|
url = ch.s3_url
|
|
db.delete(ch)
|
|
db.commit()
|
|
delete_object_by_url(url)
|