video-maker/backend/app/app/crud/crud_video.py

from typing import List

from fastapi.encoders import jsonable_encoder
from sqlalchemy.orm import Session
from sqlalchemy import desc
from app.crud.base import CRUDBase
from app.models.video import Video
from app.schemas.video import VideoCreate, VideoUpdate

from app.utils import random_name

class CRUDVideo(CRUDBase[Video, VideoCreate, VideoUpdate]):
    def create_with_owner(
        self, db: Session, *, obj_in: VideoCreate, owner_id: int
    ) -> Video:
        obj_in_data = jsonable_encoder(obj_in)
        db_obj = self.model(**obj_in_data, owner_id=owner_id)
        db.add(db_obj)
        db.commit()
        db.refresh(db_obj)
        return db_obj

    def get_multi_by_owner(
        self, db: Session, *, owner_id: int, skip: int = 0, limit: int = 100
    ) -> List[Video]:
        return (
            db.query(self.model)
            .filter(Video.owner_id == owner_id)
            .order_by(desc(Video.id))
            .offset(skip)
            .limit(limit)
            .all()
        )

    def generate_file_name(
      self, db: Session, *, n:int
    ) -> bool:
        while True:
            filename = random_name(n)
            if not db.query(self.model).filter(Video.stored_filename==filename).first():
              return filename


video = CRUDVideo(Video)