Xiaoxia AI
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
from packages.adapters.sqlalchemy_impl.models import ProjectModel
|
|
from packages.domain import Project
|
|
|
|
|
|
class SQLAlchemyProjectRepository:
|
|
def __init__(self, session: Session):
|
|
self.session = session
|
|
|
|
def _to_entity(self, model: ProjectModel) -> Project:
|
|
return Project(
|
|
id=model.id,
|
|
workspace_id=model.workspace_id,
|
|
name=model.name,
|
|
description=model.description,
|
|
created_at=model.created_at,
|
|
)
|
|
|
|
def list_by_workspace(self, workspace_id: str) -> list[Project]:
|
|
models = self.session.query(ProjectModel).filter(ProjectModel.workspace_id == workspace_id).all()
|
|
return [self._to_entity(model) for model in models]
|
|
|
|
def find_by_id(self, project_id: str) -> Project | None:
|
|
model = self.session.query(ProjectModel).filter(ProjectModel.id == project_id).first()
|
|
if model is None:
|
|
return None
|
|
return self._to_entity(model)
|
|
|
|
def create(self, project: Project) -> Project:
|
|
model = ProjectModel(
|
|
id=project.id,
|
|
workspace_id=project.workspace_id,
|
|
name=project.name,
|
|
description=project.description,
|
|
created_at=project.created_at,
|
|
)
|
|
self.session.add(model)
|
|
self.session.commit()
|
|
return project
|