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