feat(referral): add referral system with withdrawals

tgbot/database/db_users.py

@@ -15,8 +15,12 @@ from tgbot.utils.const_functions import get_unix
 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)
+    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="")
@@ -26,6 +30,13 @@ class UserModel(Base):
     user_give: Mapped[float] = mapped_column(Float, nullable=False, default=0)
     user_unix: Mapped[int] = mapped_column(Integer, nullable=False, default=get_unix)
 
+    # Поля для реферальной системы
+    user_referrer_id: Mapped[int] = mapped_column(BigInteger, nullable=True, index=True)
+    user_referral_balance: Mapped[float] = mapped_column(
+        Float, nullable=False, default=0
+    )
+    user_referral_hold: Mapped[float] = mapped_column(Float, nullable=False, default=0)
+
 
 ModelBase = User
 BaseModel = User
@@ -41,31 +52,34 @@ class UsersRepository(BaseRepository[UserModel, User]):
 
     # Добавление или обновление пользователя
     async def add(
-            self,
-            user_id: int,
-            user_login: str,
-            user_name: str,
-            user_surname: str = "",
-            user_fullname: str = "",
+        self,
+        user_id: int,
+        user_login: str,
+        user_name: str,
+        user_surname: str = "",
+        user_fullname: str = "",
     ) -> User:
         return await self.upsert(
             user_id=user_id,
             user_login=user_login,
             user_name=user_name,
             user_surname=user_surname,
-            user_fullname=user_fullname or " ".join(filter(None, [user_name, user_surname])),
+            user_fullname=user_fullname
+            or " ".join(filter(None, [user_name, user_surname])),
         )
 
     # Выполнение upsert пользователя по телеграм ID
     async def upsert(
-            self,
-            user_id: int,
-            user_login: str,
-            user_name: str,
-            user_surname: str = "",
-            user_fullname: str = "",
+        self,
+        user_id: int,
+        user_login: str,
+        user_name: str,
+        user_surname: str = "",
+        user_fullname: str = "",
     ) -> User:
-        user_fullname = user_fullname or " ".join(filter(None, [user_name, user_surname]))
+        user_fullname = user_fullname or " ".join(
+            filter(None, [user_name, user_surname])
+        )
         user_table = UserModel.__table__
         statement = insert(UserModel).values(
             user_id=user_id,
@@ -105,7 +119,9 @@ class UsersRepository(BaseRepository[UserModel, User]):
         return user
 
     # Обновление пользователя по ID или фильтру
-    async def update(self, where: Optional[Union[Dict[str, Any], int]] = None, **kwargs) -> int:
+    async def update(
+        self, where: Optional[Union[Dict[str, Any], int]] = None, **kwargs
+    ) -> int:
         if isinstance(where, int):
             where = {"user_id": where}