feat: вфыв
This commit is contained in:
47
services/user.py
Normal file
47
services/user.py
Normal file
@@ -0,0 +1,47 @@
|
||||
from sqlalchemy import select, update, delete, insert
|
||||
|
||||
from models import User, user_position
|
||||
from services.base import BaseService
|
||||
from schemas.user import *
|
||||
|
||||
|
||||
class UserService(BaseService):
|
||||
async def get_all(self) -> GetAllUsersResponse:
|
||||
stmt = (
|
||||
select(User)
|
||||
.order_by(User.id.desc())
|
||||
# .where(User.is_deleted == False)
|
||||
)
|
||||
users = (await self.session.scalars(stmt)).all()
|
||||
users_schemas = [UserSchema.model_validate(user) for user in users]
|
||||
return GetAllUsersResponse(users=users_schemas)
|
||||
|
||||
async def get_by_id(self, user_id: int) -> Optional[User]:
|
||||
return await self.session.scalar(select(User).where(User.id == user_id))
|
||||
|
||||
async def update(self, request: UpdateUserRequest) -> UpdateUserResponse:
|
||||
try:
|
||||
if not self.get_by_id(request.data.id):
|
||||
return UpdateUserResponse(ok=False, message='Указанный пользователь не найден')
|
||||
base_fields = request.data.model_dump_parent()
|
||||
stmt = update(User).values(**base_fields).where(User.id == request.data.id)
|
||||
await self.session.execute(stmt)
|
||||
await self.session.flush()
|
||||
|
||||
# Updating position
|
||||
stmt = delete(user_position).where(user_position.c.user_id == request.data.id)
|
||||
await self.session.execute(stmt)
|
||||
await self.session.flush()
|
||||
|
||||
if not request.data.position_key:
|
||||
await self.session.commit()
|
||||
return UpdateUserResponse(ok=True, message='Пользователь успешно обновлен')
|
||||
stmt = insert(user_position).values(**{
|
||||
'user_id': request.data.id,
|
||||
'position_key': request.data.position_key
|
||||
})
|
||||
await self.session.execute(stmt)
|
||||
await self.session.commit()
|
||||
return UpdateUserResponse(ok=True, message='Пользователь успешно обновлен')
|
||||
except Exception as e:
|
||||
pass
|
||||
Reference in New Issue
Block a user