from fastapi import APIRouter

from api.responses import success_response
from models import Studio


router = APIRouter()


@router.get("")
async def list_studios():
    studios = await Studio.all().values("id", "name", "contact_email")
    return success_response(
        data=studios,
        message="获取制片公司列表成功",
    )


@router.post("", status_code=201)
async def create_studio():
    return success_response(
        data={"method": "POST"},
        message="创建制片公司成功",
        code=201,
    )


@router.put("/{studio_id}")
async def update_studio(studio_id: int):
    return success_response(
        data={"id": studio_id, "method": "PUT"},
        message="更新制片公司成功",
    )


@router.delete("/{studio_id}")
async def delete_studio(studio_id: int):
    return success_response(
        data={"id": studio_id, "method": "DELETE"},
        message="删除制片公司成功",
    )