20 lines
471 B
Python
20 lines
471 B
Python
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from db.models.users import User
|
|
|
|
|
|
async def get_user(session: AsyncSession, user_id: int) -> User | None:
|
|
stmt = select(User).where(User.id == user_id)
|
|
r = await session.execute(stmt)
|
|
|
|
return r.scalar_one_or_none()
|
|
|
|
|
|
async def create_user(session: AsyncSession, user_id: int) -> User:
|
|
user = User(id=user_id)
|
|
session.add(user)
|
|
await session.commit()
|
|
|
|
return user
|