feat: implement OrganizationRepository with CRUD operations
This commit is contained in:
parent
1ce5843436
commit
845737abca
|
|
@ -0,0 +1,49 @@
|
||||||
|
"""Organization repository for database operations."""
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from collections.abc import Sequence
|
||||||
|
|
||||||
|
from sqlalchemy import select
|
||||||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
|
from app.models.organization import Organization, OrganizationCreate
|
||||||
|
from app.models.organization_member import OrganizationMember
|
||||||
|
|
||||||
|
|
||||||
|
class OrganizationRepository:
|
||||||
|
"""Provides CRUD helpers for Organization model."""
|
||||||
|
|
||||||
|
def __init__(self, session: AsyncSession) -> None:
|
||||||
|
self._session = session
|
||||||
|
|
||||||
|
@property
|
||||||
|
def session(self) -> AsyncSession:
|
||||||
|
return self._session
|
||||||
|
|
||||||
|
async def list(self) -> Sequence[Organization]:
|
||||||
|
result = await self._session.scalars(select(Organization))
|
||||||
|
return result.all()
|
||||||
|
|
||||||
|
async def get_by_id(self, organization_id: int) -> Organization | None:
|
||||||
|
return await self._session.get(Organization, organization_id)
|
||||||
|
|
||||||
|
async def get_by_name(self, name: str) -> Organization | None:
|
||||||
|
stmt = select(Organization).where(Organization.name == name)
|
||||||
|
result = await self._session.scalars(stmt)
|
||||||
|
return result.first()
|
||||||
|
|
||||||
|
async def list_for_user(self, user_id: int) -> Sequence[Organization]:
|
||||||
|
stmt = (
|
||||||
|
select(Organization)
|
||||||
|
.join(OrganizationMember, OrganizationMember.organization_id == Organization.id)
|
||||||
|
.where(OrganizationMember.user_id == user_id)
|
||||||
|
.order_by(Organization.id)
|
||||||
|
)
|
||||||
|
result = await self._session.scalars(stmt)
|
||||||
|
return result.unique().all()
|
||||||
|
|
||||||
|
async def create(self, data: OrganizationCreate) -> Organization:
|
||||||
|
organization = Organization(name=data.name)
|
||||||
|
self._session.add(organization)
|
||||||
|
await self._session.flush()
|
||||||
|
return organization
|
||||||
Loading…
Reference in New Issue