mirror of
https://github.com/djimboy/djimbo_template_aio3.git
synced 2026-07-25 01:34:30 +00:00
dj
107 lines
3.8 KiB
Python
107 lines
3.8 KiB
Python
# - *- coding: utf- 8 - *-
|
|
from sqlalchemy import BigInteger, Integer, String, or_
|
|
from sqlalchemy import update as sqlalchemy_update
|
|
from sqlalchemy.dialects.sqlite import insert
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from tgbot.database.core import Base, session_scope
|
|
from tgbot.database.repository import BaseRepository
|
|
from tgbot.utils.const_functions import get_unix
|
|
|
|
|
|
# Модель пользователя Telegram
|
|
class UserModel(Base):
|
|
__tablename__ = "storage_users"
|
|
|
|
increment: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
user_id: Mapped[int] = mapped_column(BigInteger, nullable=False, unique=True, index=True)
|
|
user_login: Mapped[str] = mapped_column(String(255), nullable=False, default="")
|
|
user_name: Mapped[str] = mapped_column(String(255), nullable=False, default="")
|
|
user_surname: Mapped[str] = mapped_column(String(255), nullable=False, default="")
|
|
user_fullname: Mapped[str] = mapped_column(String(511), nullable=False, default="")
|
|
user_unix: Mapped[int] = mapped_column(Integer, nullable=False, default=get_unix)
|
|
|
|
|
|
ModelBase = UserModel
|
|
BaseModel = UserModel
|
|
|
|
|
|
# Репозиторий пользователей
|
|
class UsersRepository(BaseRepository[UserModel]):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.table_model = UserModel
|
|
self.storage_name = UserModel.__tablename__
|
|
|
|
# Для совместимости add ведет себя как upsert
|
|
async def add(
|
|
self,
|
|
user_id: int,
|
|
user_login: str,
|
|
user_name: str,
|
|
user_surname: str,
|
|
user_fullname: str,
|
|
) -> UserModel:
|
|
return await self.upsert(
|
|
user_id=user_id,
|
|
user_login=user_login,
|
|
user_name=user_name,
|
|
user_surname=user_surname,
|
|
user_fullname=user_fullname,
|
|
)
|
|
|
|
# Создание пользователя или обновление его данных по user_id
|
|
async def upsert(
|
|
self,
|
|
user_id: int,
|
|
user_login: str,
|
|
user_name: str,
|
|
user_surname: str,
|
|
user_fullname: str,
|
|
) -> UserModel:
|
|
statement = insert(UserModel).values(
|
|
user_id=user_id,
|
|
user_login=user_login,
|
|
user_name=user_name,
|
|
user_surname=user_surname,
|
|
user_fullname=user_fullname,
|
|
user_unix=get_unix(),
|
|
)
|
|
statement = statement.on_conflict_do_update(
|
|
index_elements=[UserModel.user_id],
|
|
set_={
|
|
"user_login": statement.excluded.user_login,
|
|
"user_name": statement.excluded.user_name,
|
|
"user_surname": statement.excluded.user_surname,
|
|
"user_fullname": statement.excluded.user_fullname,
|
|
},
|
|
where=or_(
|
|
UserModel.user_login != statement.excluded.user_login,
|
|
UserModel.user_name != statement.excluded.user_name,
|
|
UserModel.user_surname != statement.excluded.user_surname,
|
|
UserModel.user_fullname != statement.excluded.user_fullname,
|
|
),
|
|
)
|
|
|
|
async with session_scope() as session:
|
|
await session.execute(statement)
|
|
|
|
user = await self.get(user_id=user_id)
|
|
|
|
if user is None:
|
|
raise RuntimeError("Пользователь не сохранился")
|
|
|
|
return user
|
|
|
|
# Обновление пользователя по Telegram ID
|
|
async def update(self, user_id: int, **kwargs) -> None:
|
|
if not kwargs:
|
|
return
|
|
|
|
async with session_scope() as session:
|
|
await session.execute(
|
|
sqlalchemy_update(UserModel)
|
|
.where(UserModel.user_id == user_id)
|
|
.values(**kwargs)
|
|
)
|