15 lines
506 B
Python
15 lines
506 B
Python
from fastapi import APIRouter
|
|
from app.schemas import TaskCreateResponse
|
|
from app.schemas import DownloadRequest
|
|
from app.tasks import download_youtube
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.post("/", response_model=TaskCreateResponse)
|
|
async def download_video(data: DownloadRequest):
|
|
print("Get Request. Try to set task")
|
|
task = download_youtube.delay(url=str(data.url), quality=data.quality)
|
|
print("Set task. Return task response")
|
|
return TaskCreateResponse(task_id=task.id, status=task.status)
|