mirror of
https://github.com/trushildhokiya/allininx-2.git
synced 2025-03-15 20:18:40 +00:00
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
|
from typing import Optional
|
||
|
|
|
||
|
|
from sqlalchemy import and_, select
|
||
|
|
from sqlalchemy.orm import selectinload
|
||
|
|
|
||
|
|
from reworkd_platform.db.crud.base import BaseCrud
|
||
|
|
from reworkd_platform.db.models.auth import OrganizationUser
|
||
|
|
from reworkd_platform.db.models.user import UserSession
|
||
|
|
|
||
|
|
|
||
|
|
class UserCrud(BaseCrud):
|
||
|
|
async def get_user_session(self, token: str) -> UserSession:
|
||
|
|
query = (
|
||
|
|
select(UserSession)
|
||
|
|
.filter(UserSession.session_token == token)
|
||
|
|
.options(selectinload(UserSession.user))
|
||
|
|
)
|
||
|
|
return (await self.session.execute(query)).scalar_one()
|
||
|
|
|
||
|
|
async def get_user_organization(
|
||
|
|
self, user_id: str, organization_id: str
|
||
|
|
) -> Optional[OrganizationUser]:
|
||
|
|
query = select(OrganizationUser).filter(
|
||
|
|
and_(
|
||
|
|
OrganizationUser.user_id == user_id,
|
||
|
|
OrganizationUser.organization_id == organization_id,
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
# TODO: Only returns the first organization
|
||
|
|
return (await self.session.execute(query)).scalar()
|