Fix issues with getting the owner
This commit is contained in:
@@ -1,11 +1,12 @@
|
||||
from sqlalchemy.ext.asyncio.session import AsyncSession
|
||||
from typing import Annotated
|
||||
from uuid import UUID
|
||||
|
||||
from fastapi import Depends
|
||||
from ..providers.db_provider import get_session
|
||||
from ..models.owner import Owner
|
||||
from sqlalchemy.ext.asyncio.session import AsyncSession
|
||||
from sqlmodel import select
|
||||
|
||||
from uuid import UUID
|
||||
from ..models.owner import Owner
|
||||
from ..providers.db_provider import get_session
|
||||
|
||||
|
||||
class OwnerRepository:
|
||||
@@ -22,6 +23,11 @@ class OwnerRepository:
|
||||
result = await self.session.execute(statement)
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
async def get_by_user_id(self, user_id: UUID):
|
||||
statement = select(Owner).where(Owner.user_id == user_id)
|
||||
result = await self.session.execute(statement)
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
async def save(self, owner: Owner) -> None:
|
||||
"""
|
||||
Save a owner to the database. If an owner with that ID already exists, do an upsert.
|
||||
|
||||
Reference in New Issue
Block a user