feat: initialize aivideo project
This commit is contained in:
104
backend/app/modules/video_tasks/router.py
Normal file
104
backend/app/modules/video_tasks/router.py
Normal file
@@ -0,0 +1,104 @@
|
||||
from fastapi import APIRouter, Depends
|
||||
from sqlalchemy.orm import Session
|
||||
|
||||
from app.common.db.session import get_db
|
||||
from app.common.responses.api_response import success_response
|
||||
from app.common.security.deps import get_current_user, require_admin_permission
|
||||
from app.models.entities import User
|
||||
from app.modules.video_tasks.schema import CreateVideoTaskRequest
|
||||
from app.modules.video_tasks.service import VideoTasksService
|
||||
|
||||
|
||||
router = APIRouter(tags=["video-tasks"])
|
||||
|
||||
|
||||
@router.post("/api/v1/video-tasks")
|
||||
def create_task(
|
||||
payload: CreateVideoTaskRequest,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).create_task(current_user.id, payload))
|
||||
|
||||
|
||||
@router.get("/api/v1/video-tasks")
|
||||
def list_tasks(
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).list_tasks(current_user.id))
|
||||
|
||||
|
||||
@router.get("/api/v1/video-tasks/{task_no}")
|
||||
def get_task_detail(
|
||||
task_no: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).get_task_detail(current_user.id, task_no))
|
||||
|
||||
|
||||
@router.post("/api/v1/video-tasks/{task_no}/retry")
|
||||
def retry_task(
|
||||
task_no: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).retry_task(current_user.id, task_no))
|
||||
|
||||
|
||||
@router.post("/api/v1/video-tasks/{task_no}/cancel")
|
||||
def cancel_task(
|
||||
task_no: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).cancel_task(current_user.id, task_no))
|
||||
|
||||
|
||||
@router.delete("/api/v1/video-tasks/{task_no}")
|
||||
def delete_task(
|
||||
task_no: str,
|
||||
current_user: User = Depends(get_current_user),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).delete_task(current_user.id, task_no))
|
||||
|
||||
|
||||
admin_router = APIRouter(prefix="/api/v1/admin/video-tasks", tags=["admin-video-tasks"])
|
||||
|
||||
|
||||
@admin_router.get("")
|
||||
def admin_list_tasks(
|
||||
_=Depends(require_admin_permission()),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).admin_list_tasks())
|
||||
|
||||
|
||||
@admin_router.get("/{task_id}")
|
||||
def admin_get_task(
|
||||
task_id: int,
|
||||
_=Depends(require_admin_permission()),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).admin_get_task(task_id))
|
||||
|
||||
|
||||
@admin_router.post("/{task_id}/retry")
|
||||
def admin_retry_task(
|
||||
task_id: int,
|
||||
_=Depends(require_admin_permission()),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).admin_retry_task(task_id))
|
||||
|
||||
|
||||
@admin_router.post("/{task_id}/refund")
|
||||
def admin_refund_task(
|
||||
task_id: int,
|
||||
_=Depends(require_admin_permission()),
|
||||
db: Session = Depends(get_db),
|
||||
):
|
||||
return success_response(VideoTasksService(db).admin_refund_task(task_id))
|
||||
|
||||
Reference in New Issue
Block a user