feat(casino): Казино для групп

AuthFailed · AuthFailed · commit 3cd631ff9f56 · 2025-11-16T21:49:10.000+05:00
Closes #154
diff --git a/tgbot/filters/group_casino.py b/tgbot/filters/group_casino.py
@@ -0,0 +1,50 @@
+"""Фильтры для проверки доступа к казино в группах."""
+
+from aiogram.dispatcher.event.bases import CancelHandler
+from aiogram.filters import BaseFilter
+from aiogram.types import Message
+from stp_database import Employee, MainRequestsRepo
+
+
+class IsGroupCasinoAllowed(BaseFilter):
+    """Фильтр проверки доступа к казино в группе.
+
+    Проверяет:
+    1. Зарегистрирована ли группа в базе данных
+    2. Разрешен ли казино в группе (is_casino_allowed=True)
+    3. Разрешен ли доступ к казино пользователю (is_casino_allowed=True)
+
+    При отсутствии любого из условий прерывает обработку без сообщения.
+    """
+
+    async def __call__(
+        self, message: Message, user: Employee, stp_repo: MainRequestsRepo, **kwargs
+    ) -> bool:
+        """Проверяет разрешен ли доступ к казино для пользователя и группы.
+
+        Args:
+            message: Входящее сообщение.
+            user: Экземпляр пользователя с моделью Employee
+            stp_repo: Репозиторий операций с базой STP
+            **kwargs: Дополнительные аргументы.
+
+        Returns:
+            True если доступ разрешен.
+
+        Raises:
+            CancelHandler: Если доступ запрещен (без отправки сообщения).
+        """
+        # Проверяем пользователя
+        if user is None or not user.is_casino_allowed:
+            raise CancelHandler()
+
+        # Проверяем группу
+        try:
+            group = await stp_repo.group.get_groups(message.chat.id)
+            if not group or not group.is_casino_allowed:
+                raise CancelHandler()
+        except Exception:
+            # Если ошибка при получении группы - запрещаем доступ
+            raise CancelHandler()
+
+        return True
diff --git a/tgbot/handlers/__init__.py b/tgbot/handlers/__init__.py
@@ -8,6 +8,7 @@
 from tgbot.handlers.groups.admin.admin import group_admin_router
 from tgbot.handlers.groups.admin.settings import group_settings_router
 from tgbot.handlers.groups.join import groups_router
+from tgbot.handlers.groups.user.casino import group_casino_router
 from tgbot.handlers.groups.user.main import group_user_router
 from tgbot.handlers.groups.user.whois import group_whois_router
 from tgbot.handlers.inline.inline import user_inline_router
@@ -23,6 +24,7 @@
     group_whois_router,
     group_admin_router,
     group_user_router,
+    group_casino_router,
     group_settings_router,
     channels_router,
     groups_router,
diff --git a/tgbot/handlers/groups/user/casino.py b/tgbot/handlers/groups/user/casino.py
@@ -0,0 +1,211 @@
+"""Обработчики команд казино для групп."""
+
+import asyncio
+import logging
+import re
+
+from aiogram import F, Router
+from aiogram.enums import DiceEmoji
+from aiogram.filters import Command
+from aiogram.types import Message
+from stp_database import Employee, MainRequestsRepo
+
+from tgbot.dialogs.events.common.game.casino import (
+    calculate_simple_multiplier,
+    calculate_slots_multiplier,
+    format_result,
+)
+from tgbot.filters.group_casino import IsGroupCasinoAllowed
+
+logger = logging.getLogger(__name__)
+
+group_casino_router = Router()
+group_casino_router.message.filter(F.chat.type.in_(("group", "supergroup")))
+
+
+def parse_casino_command(message_text: str) -> int:
+    """Извлечь ставку из команды казино.
+
+    Args:
+        message_text: Текст команды (например, "/slots 50" или "/dice")
+
+    Returns:
+        Размер ставки (минимум 10)
+    """
+    # Паттерн для извлечения числа из команды
+    match = re.search(r"/(?:slots|dice|darts|bowling)\s+(\d+)", message_text)
+    if match:
+        bet_amount = int(match.group(1))
+        return max(10, bet_amount)  # Минимальная ставка 10
+    return 10  # Ставка по умолчанию
+
+
+async def process_casino_game(
+    message: Message,
+    user: Employee,
+    stp_repo: MainRequestsRepo,
+    game_type: str,
+    dice_emoji: DiceEmoji,
+    bet_amount: int,
+) -> None:
+    """Обработать игру в казино.
+
+    Args:
+        message: Сообщение от пользователя
+        user: Экземпляр пользователя с моделью Employee
+        stp_repo: Репозиторий операций с базой STP
+        game_type: Тип игры (slots, dice, darts, bowling)
+        dice_emoji: Emoji для dice API
+        bet_amount: Размер ставки
+    """
+    try:
+        # Проверяем баланс
+        user_balance = await stp_repo.transaction.get_user_balance(user.user_id)
+
+        if user_balance < bet_amount:
+            await message.reply(
+                f"❌ Недостаточно баллов для игры!\n"
+                f"💰 Твой баланс: {user_balance} баллов\n"
+                f"🎲 Нужно для ставки: {bet_amount} баллов"
+            )
+            return
+
+        # Отправляем dice и ждем результата
+        dice_message = await message.reply_dice(emoji=dice_emoji)
+        dice_value = dice_message.dice.value
+
+        # Ждем анимацию (3 секунды)
+        await asyncio.sleep(3)
+
+        # Рассчитываем выигрыш
+        if game_type == "slots":
+            multiplier = calculate_slots_multiplier(dice_value)
+        else:
+            multiplier = calculate_simple_multiplier(dice_value)
+
+        # Вычисляем чистый выигрыш/проигрыш
+        if multiplier > 0:
+            gross_win = int(bet_amount * multiplier)
+            net_win = gross_win - bet_amount
+        else:
+            net_win = -bet_amount
+
+        # Обновляем баланс
+        if net_win > 0:
+            await stp_repo.transaction.add_transaction(
+                user_id=user.user_id,
+                transaction_type="earn",
+                source_type="casino",
+                amount=net_win,
+                comment=f"Выигрыш в {game_type}: {dice_value} (ставка {bet_amount})",
+            )
+        elif net_win < 0:
+            await stp_repo.transaction.add_transaction(
+                user_id=user.user_id,
+                transaction_type="spend",
+                source_type="casino",
+                amount=abs(net_win),
+                comment=f"Проигрыш в {game_type}: {dice_value} (ставка {bet_amount})",
+            )
+
+        # Получаем новый баланс
+        new_balance = await stp_repo.transaction.get_user_balance(user.user_id)
+
+        # Формируем сообщение о результате используя общую функцию
+        result_data = format_result(game_type, dice_value, multiplier, net_win)
+
+        # Дополняем информацией о ставке и балансе для групповых команд
+        message_parts = [
+            f"{result_data['result_icon']} <b>{result_data['result_title']}</b>",
+            result_data["result_message"],
+        ]
+
+        # Информация о ставке и выигрыше
+        if net_win > 0:
+            message_parts.append(f"\n💰 <b>Выиграно:</b> +{net_win} баллов")
+        elif net_win < 0:
+            message_parts.append(f"\n💸 <b>Проиграно:</b> {abs(net_win)} баллов")
+
+        message_parts.append(f"\n🎯 <b>Ставка:</b> {bet_amount} баллов")
+        message_parts.append(f"\n\n💳 <b>Баланс:</b> {new_balance} баллов")
+
+        result_message = "\n".join(message_parts)
+
+        await dice_message.reply(result_message)
+
+        # Логируем игру
+        user_name = (
+            user.fullname
+            if user
+            else f"@{message.from_user.username}"
+            if message.from_user.username
+            else message.from_user.full_name
+        )
+        logger.info(
+            f"[Casino/{game_type}] {user_name} ({user.user_id}) играл с ставкой {bet_amount}, "
+            f"результат {dice_value}, выигрыш {net_win}"
+        )
+
+    except Exception as e:
+        logger.error(f"Ошибка в казино {game_type}: {e}")
+        await message.reply("🚨 Произошла ошибка при игре в казино. Попробуй позже.")
+
+
+@group_casino_router.message(Command("slots"), IsGroupCasinoAllowed())
+async def slots_cmd(message: Message, user: Employee, stp_repo: MainRequestsRepo):
+    """Обработчик команды /slots для групп.
+
+    Args:
+        message: Сообщение от пользователя
+        user: Экземпляр пользователя с моделью Employee
+        stp_repo: Репозиторий операций с базой STP
+    """
+    bet_amount = parse_casino_command(message.text)
+    await process_casino_game(
+        message, user, stp_repo, "slots", DiceEmoji.SLOT_MACHINE, bet_amount
+    )
+
+
+@group_casino_router.message(Command("dice"), IsGroupCasinoAllowed())
+async def dice_cmd(message: Message, user: Employee, stp_repo: MainRequestsRepo):
+    """Обработчик команды /dice для групп.
+
+    Args:
+        message: Сообщение от пользователя
+        user: Экземпляр пользователя с моделью Employee
+        stp_repo: Репозиторий операций с базой STP
+    """
+    bet_amount = parse_casino_command(message.text)
+    await process_casino_game(
+        message, user, stp_repo, "dice", DiceEmoji.DICE, bet_amount
+    )
+
+
+@group_casino_router.message(Command("darts"), IsGroupCasinoAllowed())
+async def darts_cmd(message: Message, user: Employee, stp_repo: MainRequestsRepo):
+    """Обработчик команды /darts для групп.
+
+    Args:
+        message: Сообщение от пользователя
+        user: Экземпляр пользователя с моделью Employee
+        stp_repo: Репозиторий операций с базой STP
+    """
+    bet_amount = parse_casino_command(message.text)
+    await process_casino_game(
+        message, user, stp_repo, "darts", DiceEmoji.DART, bet_amount
+    )
+
+
+@group_casino_router.message(Command("bowling"), IsGroupCasinoAllowed())
+async def bowling_cmd(message: Message, user: Employee, stp_repo: MainRequestsRepo):
+    """Обработчик команды /bowling для групп.
+
+    Args:
+        message: Сообщение от пользователя
+        user: Экземпляр пользователя с моделью Employee
+        stp_repo: Репозиторий операций с базой STP
+    """
+    bet_amount = parse_casino_command(message.text)
+    await process_casino_game(
+        message, user, stp_repo, "bowling", DiceEmoji.BOWLING, bet_amount
+    )
diff --git a/tgbot/services/mailing.py b/tgbot/services/mailing.py
@@ -93,7 +93,7 @@ async def send_activation_product_email(
 
 <a href="https://t.me/{user.username}"<b>{user.fullname}</b></a> из {user.division} просит активировать предмет <b>{product.name}</b><br><br>
 
-<b>О предмете</b>
+<b>О предмете</b><br>
 💵 Стоимость: {product.cost} баллов<br>
 📝 Описание: {product.description}<br>
 📍 Всего активаций: {product.count} (Осталось у специалиста - {product.count - (purchase.usage_count + 1)})<br>
