mirror of
https://github.com/koloideal/DutyLog.git
synced 2026-06-10 10:25:29 +03:00
initial
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from src.dutylog.infrastructure.database.models.user import User
|
||||
|
||||
|
||||
class UsersDAO:
|
||||
def __init__(self, session: AsyncSession):
|
||||
self.session = session
|
||||
|
||||
async def get_user(self, user_id: int) -> User | None:
|
||||
result = await self.session.execute(
|
||||
select(User).where(User.id == user_id)
|
||||
)
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
async def create_user(self, user_id: int, username: str | None) -> User:
|
||||
user = User(id=user_id, username=username)
|
||||
self.session.add(user)
|
||||
await self.session.commit()
|
||||
return user
|
||||
Reference in New Issue
Block a user