Xiaoxia AI
64 lines
2.4 KiB
Python
64 lines
2.4 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
from packages.adapters.sqlalchemy_impl.models import GeneratedVideoModel
|
|
from packages.domain import GeneratedVideo
|
|
|
|
|
|
class SQLAlchemyGeneratedVideoRepository:
|
|
def __init__(self, session: Session):
|
|
self.session = session
|
|
|
|
def create(self, video: GeneratedVideo) -> GeneratedVideo:
|
|
model = GeneratedVideoModel(
|
|
id=video.id,
|
|
workspace_id=video.workspace_id,
|
|
project_id=video.project_id,
|
|
generation_task_id=video.generation_task_id,
|
|
name=video.name,
|
|
file_url=video.file_url,
|
|
file_size=video.file_size,
|
|
duration=video.duration,
|
|
thumbnail_url=video.thumbnail_url,
|
|
width=video.width,
|
|
height=video.height,
|
|
fps=video.fps,
|
|
generated_at=video.generated_at,
|
|
created_at=video.created_at,
|
|
)
|
|
self.session.add(model)
|
|
self.session.commit()
|
|
return video
|
|
|
|
def get(self, video_id: str) -> GeneratedVideo | None:
|
|
model = self.session.query(GeneratedVideoModel).filter(GeneratedVideoModel.id == video_id).first()
|
|
if model is None:
|
|
return None
|
|
return GeneratedVideo(
|
|
id=model.id,
|
|
workspace_id=model.workspace_id,
|
|
project_id=model.project_id,
|
|
generation_task_id=model.generation_task_id,
|
|
name=model.name,
|
|
file_url=model.file_url,
|
|
file_size=int(model.file_size or 0),
|
|
duration=model.duration,
|
|
thumbnail_url=model.thumbnail_url,
|
|
width=int(model.width or 0),
|
|
height=int(model.height or 0),
|
|
fps=model.fps,
|
|
generated_at=model.generated_at,
|
|
created_at=model.created_at,
|
|
)
|
|
|
|
def list_by_project(self, project_id: str) -> list[GeneratedVideo]:
|
|
models = self.session.query(GeneratedVideoModel).filter(GeneratedVideoModel.project_id == project_id).all()
|
|
return [self.get(model.id) for model in models if self.get(model.id) is not None]
|
|
|
|
def list_by_generation_task(self, generation_task_id: str) -> list[GeneratedVideo]:
|
|
models = (
|
|
self.session.query(GeneratedVideoModel)
|
|
.filter(GeneratedVideoModel.generation_task_id == generation_task_id)
|
|
.all()
|
|
)
|
|
return [self.get(model.id) for model in models if self.get(model.id) is not None]
|