Update project configuration and codebase
This commit is contained in:
@@ -1,4 +1,3 @@
|
||||
from typing import Optional
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
@@ -7,9 +6,7 @@ from schemas.users import NewUserDTO
|
||||
|
||||
|
||||
class UserRepository:
|
||||
async def get_user_by_id(
|
||||
self, session: AsyncSession, user_id: int
|
||||
) -> Optional[User]:
|
||||
async def get_user_by_id(self, session: AsyncSession, user_id: int) -> User | None:
|
||||
stmt = select(User).where(User.id == user_id)
|
||||
result = await session.execute(stmt)
|
||||
|
||||
@@ -28,7 +25,7 @@ class UserRepository:
|
||||
|
||||
async def update_user_topic(
|
||||
self, session: AsyncSession, user_id: int, thread_id: int
|
||||
) -> Optional[User]:
|
||||
) -> User | None:
|
||||
user = await self.get_user_by_id(session, user_id=user_id)
|
||||
if not user:
|
||||
return None
|
||||
@@ -37,9 +34,7 @@ class UserRepository:
|
||||
await session.commit()
|
||||
return user
|
||||
|
||||
async def update_user_link(
|
||||
self, session: AsyncSession, user_id: int, link: str
|
||||
) -> Optional[User]:
|
||||
async def update_user_link(self, session: AsyncSession, user_id: int, link: str) -> User | None:
|
||||
user = await self.get_user_by_id(session, user_id=user_id)
|
||||
if not user:
|
||||
return None
|
||||
@@ -50,7 +45,7 @@ class UserRepository:
|
||||
|
||||
async def update_user_balance(
|
||||
self, session: AsyncSession, user_id: int, balance: int
|
||||
) -> Optional[User]:
|
||||
) -> User | None:
|
||||
user = await self.get_user_by_id(session, user_id=user_id)
|
||||
if not user:
|
||||
return None
|
||||
@@ -61,7 +56,7 @@ class UserRepository:
|
||||
|
||||
async def increase_user_balance(
|
||||
self, session: AsyncSession, user_id: int, amount: int
|
||||
) -> Optional[User]:
|
||||
) -> User | None:
|
||||
user = await self.get_user_by_id(session, user_id=user_id)
|
||||
if not user:
|
||||
return None
|
||||
|
||||
Reference in New Issue
Block a user