This commit is contained in:
macbook_max
2026-10-15 20:36:59 +08:00
commit 6bf949949c
7923 changed files with 1369582 additions and 0 deletions
+117
View File
@@ -0,0 +1,117 @@
"""视频模块路由(/videos/* + /tags/* + /search/*,对应《03-接口文档》第 5 章)"""
import redis.asyncio as aioredis
from fastapi import APIRouter, Depends, Form, Query, Request, UploadFile
from sqlalchemy.ext.asyncio import AsyncSession
from app.core import ratelimit
from app.core.response import ok
from app.database import get_db
from app.deps import Identity, get_identity, require_user
from app.redis_client import get_redis
from app.schemas.video import EditVideoReq, PublishVideoReq, UploadTokenReq, WatchCompleteReq, WatchReq
from app.services import video_service
router = APIRouter(prefix="/videos", tags=["视频"])
router_tags = APIRouter(prefix="/tags", tags=["视频"])
router_search = APIRouter(prefix="/search", tags=["搜索"])
# 搜索限流:30 次/分钟(登录按用户,游客按 IP)
SEARCH_RATE_LIMIT = 30
# 上传凭证 + 发布限流:5 次/小时(同一动作桶)
UPLOAD_RATE_LIMIT = 5
# 注意:/videos/feed、/videos/categories、/videos/search、/videos/upload/* 须先于 /videos/{video_id} 注册
@router.get("/feed")
async def get_feed(cursor: str | None = Query(None), limit: int = Query(10, ge=1, le=20),
category_id: int | None = Query(None),
type: str = Query("recommend", pattern="^(recommend|latest|hot)$"),
db: AsyncSession = Depends(get_db), redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(get_identity)):
return ok(await video_service.get_feed(db, redis, identity, type, cursor, limit, category_id))
@router.get("/categories")
async def get_categories(db: AsyncSession = Depends(get_db)):
return ok(await video_service.list_categories(db))
@router.get("/search")
async def search_videos(request: Request, keyword: str = Query(min_length=1, max_length=30),
sort: str = Query("hot", pattern="^(hot|latest)$"),
page: int = Query(1, ge=1), page_size: int = Query(20, ge=1, le=50),
db: AsyncSession = Depends(get_db), redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(get_identity)):
subject = identity.subject or (request.client.host if request.client else "anon")
await ratelimit.hit_subject(redis, subject, "search", SEARCH_RATE_LIMIT, 60)
return ok(await video_service.search_videos(db, redis, identity, keyword, sort, page, page_size))
@router_search.get("/hot")
async def get_hot_search(redis: aioredis.Redis = Depends(get_redis)):
return ok(await video_service.get_hot_search(redis))
@router_tags.get("/{tag_name}/videos")
async def get_tag_videos(tag_name: str, sort: str = Query("hot", pattern="^(hot|latest)$"),
page: int = Query(1, ge=1), page_size: int = Query(20, ge=1, le=50),
db: AsyncSession = Depends(get_db), identity: Identity = Depends(get_identity)):
return ok(await video_service.get_tag_videos(db, identity, tag_name, sort, page, page_size))
@router.post("/upload/token")
async def get_upload_token(req: UploadTokenReq, redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(require_user)):
await ratelimit.hit_subject(redis, identity.subject, "upload", UPLOAD_RATE_LIMIT, 3600, "上传过于频繁,请稍后再试")
return ok(await video_service.get_upload_token(redis, identity, req))
@router.post("/upload/file")
async def upload_file(file: UploadFile, object_key: str = Form(...),
redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(require_user)):
return ok(await video_service.register_upload(redis, identity, object_key, file))
@router.post("")
async def publish_video(req: PublishVideoReq, db: AsyncSession = Depends(get_db),
redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(require_user)):
await ratelimit.hit_subject(redis, identity.subject, "upload", UPLOAD_RATE_LIMIT, 3600, "发布过于频繁,请稍后再试")
return ok(await video_service.publish_video(db, redis, identity, req))
@router.get("/{video_id}")
async def get_detail(video_id: int, db: AsyncSession = Depends(get_db),
redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(get_identity)):
return ok(await video_service.get_detail(db, redis, identity, video_id))
@router.put("/{video_id}")
async def update_video(video_id: int, req: EditVideoReq, db: AsyncSession = Depends(get_db),
redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(require_user)):
return ok(await video_service.update_video(db, redis, identity, video_id, req))
@router.delete("/{video_id}")
async def delete_video(video_id: int, db: AsyncSession = Depends(get_db),
redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(require_user)):
await video_service.delete_video(db, redis, identity, video_id)
return ok()
@router.post("/{video_id}/watch")
async def report_watch(video_id: int, req: WatchReq, db: AsyncSession = Depends(get_db),
redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(get_identity)):
return ok(await video_service.report_watch(db, redis, identity, video_id, req.position))
@router.post("/{video_id}/watch/complete")
async def report_complete(video_id: int, req: WatchCompleteReq, db: AsyncSession = Depends(get_db),
redis: aioredis.Redis = Depends(get_redis),
identity: Identity = Depends(get_identity)):
return ok(await video_service.report_complete(db, redis, identity, video_id, req.watch_duration))