connect database
This commit is contained in:
22
app/infrastructure/database/crud.py
Normal file
22
app/infrastructure/database/crud.py
Normal file
@@ -0,0 +1,22 @@
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from .models import User
|
||||
|
||||
|
||||
async def get_user_by_tg_id(session: AsyncSession, user_tg_id: int) -> User | None:
|
||||
result = await session.execute(select(User).where(User.tg_id == user_tg_id))
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
|
||||
async def create_user(
|
||||
session: AsyncSession,
|
||||
tg_id: int,
|
||||
fullname: str,
|
||||
phone: str = "",
|
||||
role: str = "user",
|
||||
) -> User:
|
||||
user = User(tg_id=tg_id, fullname=fullname, role=role, phone=phone)
|
||||
session.add(user)
|
||||
await session.commit()
|
||||
return user
|
||||
Reference in New Issue
Block a user