добавил подсчёт количества рефералов

app/services/db_manager.py

@@ -61,6 +61,18 @@ class DatabaseManager:
         """
         return await self.postgres_repo.get_last_transactions(telegram_id, limit)
 
+    async def get_referrals_count(self,telegram_id: int) -> int:
+        """
+        Docstring for get_referrals_count
+        
+        :param self: Description
+        :param telegram_id: Description
+        :type telegram_id: int
+        :return: Description
+        :rtype: int
+        """
+        return await self.postgres_repo.get_referrals_count(telegram_id)
+
     # async def update_balance(self, telegram_id: int, amount: float):
     #     """
     #     Обновляет баланс пользователя и добавляет транзакцию.

app/services/postgres_rep.py

@@ -277,4 +277,24 @@ class PostgresRepository:
             except SQLAlchemyError as e:
                 self.logger.error(f"Ошибка при поиске плана: {plan_id}: {e}")
                 await session.rollback()
-                return None
+                return None
+
+    async def get_referrals_count(self, user_telegram_id: int) -> int:
+        """
+        Получить количество рефералов пользователя.
+        
+        :param user_telegram_id: Telegram ID пользователя-пригласителя
+        :return: Количество рефералов
+        """
+        async for session in self.session_generator():
+            try:
+                result = await session.execute(
+                    select(Referral)
+                    .where(Referral.inviter_id == user_telegram_id)
+                )
+                referrals = result.scalars().all()
+                return len(referrals)
+            except SQLAlchemyError as e:
+                self.logger.error(f"Ошибка при получении количества рефералов для пользователя {user_telegram_id}: {e}")
+                return 0
+