Initial commit

tgbot/database/db_users.py

@@ -0,0 +1,106 @@
+# - *- 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)
+            )