andrei
dc47e4b003
Добавлена функциональность управления сессиями пользователей, включая создание сессий при входе, получение списка активных сессий, деактивацию отдельных сессий и деактивацию всех сессий пользователя.
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
import datetime
|
|
from typing import Optional, Sequence
|
|
|
|
from sqlalchemy import update, delete
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.future import select
|
|
|
|
from app.domain.models.sessions import Session
|
|
|
|
|
|
class SessionsRepository:
|
|
def __init__(self, db: AsyncSession):
|
|
self.db = db
|
|
|
|
async def create(self, session: Session) -> Session:
|
|
self.db.add(session)
|
|
await self.db.commit()
|
|
await self.db.refresh(session)
|
|
return session
|
|
|
|
async def get_by_id(self, session_id: int) -> Optional[Session]:
|
|
result = await self.db.execute(
|
|
select(Session).filter_by(id=session_id)
|
|
)
|
|
return result.scalars().all()
|
|
|
|
async def get_by_token(self, token: str) -> Optional[Session]:
|
|
result = await self.db.execute(
|
|
select(Session).filter_by(token=token, is_active=True)
|
|
)
|
|
return result.scalars().first()
|
|
|
|
async def get_by_user_id(self, user_id: int) -> Sequence[Session]:
|
|
result = await self.db.execute(
|
|
select(Session).filter_by(user_id=user_id, is_active=True)
|
|
)
|
|
return result.scalars().all()
|
|
|
|
async def deactivate_session(self, session_id: int) -> None:
|
|
await self.db.execute(
|
|
update(Session).filter_by(id=session_id).values(is_active=False)
|
|
)
|
|
await self.db.commit()
|
|
|
|
async def deactivate_all_sessions(self, user_id: int) -> None:
|
|
await self.db.execute(
|
|
update(Session).filter_by(user_id=user_id).values(is_active=False)
|
|
)
|
|
await self.db.commit()
|
|
|
|
async def cleanup_expired_sessions(self) -> None:
|
|
await self.db.execute(
|
|
delete(Session).filter(Session.expires_at < datetime.datetime.now())
|
|
)
|
|
await self.db.commit()
|