From 835e1f615548e2871f0755547c094af06f2e2113 Mon Sep 17 00:00:00 2001 From: Aevann Date: Sat, 18 Feb 2023 21:57:34 +0200 Subject: [PATCH] give gambling badges for roulette too (testing on devrama) --- files/classes/casino_game.py | 2 ++ files/helpers/roulette.py | 3 +++ 2 files changed, 5 insertions(+) diff --git a/files/classes/casino_game.py b/files/classes/casino_game.py index 0c6c6626f..307511517 100644 --- a/files/classes/casino_game.py +++ b/files/classes/casino_game.py @@ -22,6 +22,8 @@ class CasinoGame(Base): kind = Column(String) game_state = Column(JSON) + user = relationship("User") + def __init__(self, *args, **kwargs): if "created_utc" not in kwargs: kwargs["created_utc"] = int(time.time()) diff --git a/files/helpers/roulette.py b/files/helpers/roulette.py index de1355d5b..a6dcf057c 100644 --- a/files/helpers/roulette.py +++ b/files/helpers/roulette.py @@ -8,6 +8,7 @@ from flask import g from files.classes.casino_game import CasinoGame from files.helpers.alerts import * from files.helpers.get import get_account +from files.helpers.casino import distribute_wager_badges class RouletteAction(str, Enum): STRAIGHT_UP_BET = "STRAIGHT_UP_BET", @@ -209,6 +210,8 @@ def spin_roulette_wheel(): else: game.winnings = -game.wager + distribute_wager_badges(game.user, game.wager, won=(game.winnings > 0)) + game.active = False g.db.add(game)