Lumi/services/SlotsStats.py

import json

from db import database


class SlotsStats:
    """
    Handles statistics for the /slots command
    """
    def __init__(self, user_id, is_won, bet, payout, spin_type, icons):
        self.user_id = user_id
        self.is_won = is_won
        self.bet = bet
        self.payout = payout
        self.spin_type = spin_type
        self.icons = json.dumps(icons)

    def push(self):
        """
        Insert the services from any given slots game into the database
        """
        query = """
        INSERT INTO slots (user_id, is_won, bet, payout, spin_type, icons)
        VALUES (%s, %s, %s, %s, %s, %s)
        """

        values = (self.user_id, self.is_won, self.bet, self.payout, self.spin_type, self.icons)

        database.execute_query(query, values)

    @staticmethod
    def get_user_stats(user_id):
        """
        Retrieve the Slots stats for a given user from the database.
        """
        query = """
        SELECT
            COUNT(*) AS amount_of_games,
            SUM(bet) AS total_bet,
            SUM(payout) AS total_payout,
            SUM(CASE WHEN spin_type = 'pair' AND is_won = 1 THEN 1 ELSE 0 END) AS games_won_pair,
            SUM(CASE WHEN spin_type = 'three_of_a_kind' AND is_won = 1 THEN 1 ELSE 0 END) AS games_won_three_of_a_kind,
            SUM(CASE WHEN spin_type = 'three_diamonds' AND is_won = 1 THEN 1 ELSE 0 END) AS games_won_three_diamonds,
            SUM(CASE WHEN spin_type = 'jackpot' AND is_won = 1 THEN 1 ELSE 0 END) AS games_won_jackpot
        FROM slots
        WHERE user_id = %s
        """

        (amount_of_games, total_bet,
         total_payout, games_won_pair, games_won_three_of_a_kind,
         games_won_three_diamonds, games_won_jackpot) = database.select_query(query, (user_id,))[0]

        return {
            "amount_of_games": amount_of_games,
            "total_bet": total_bet,
            "total_payout": total_payout,
            "games_won_pair": games_won_pair,
            "games_won_three_of_a_kind": games_won_three_of_a_kind,
            "games_won_three_diamonds": games_won_three_diamonds,
            "games_won_jackpot": games_won_jackpot
        }