rDrama/files/helpers/slots.py

import json
from json.encoder import INFINITY
import random
from .const import *
from files.classes.casino_game import Casino_Game
from files.helpers.casino import distribute_wager_badges
from flask import g

minimum_bet = 5
maximum_bet = INFINITY
payout_to_symbols = {
	2: ["👣", "🍀", "🌈", "⭐️"],
	3: ["🍎", "🔞", "⚛️", "☢️"],
	5: ["✡️", "⚔️", "🍆", "🍒"],
	12: ["🐱"]
}


def casino_slot_pull(gambler, wager_value, currency):
	over_min = wager_value >= minimum_bet
	under_max = wager_value <= maximum_bet
	charged = gambler.charge_account(currency, wager_value)

	if (over_min and under_max and charged):
		payout = determine_payout()
		reward = wager_value * payout

		gambler.pay_account(currency, reward)

		if currency == 'coins':
			distribute_wager_badges(gambler, wager_value, won=(payout > 0))

		symbols = build_symbols(payout)
		text = build_text(wager_value, payout, currency)
		game_state = {
			"symbols": symbols,
			"text": text
		}
		casino_game = Casino_Game()
		casino_game.active = False
		casino_game.user_id = gambler.id
		casino_game.currency = currency
		casino_game.wager = wager_value
		casino_game.winnings = reward - wager_value
		casino_game.kind = 'slots'
		casino_game.game_state = json.dumps(game_state)
		g.db.add(casino_game)

		return casino_game.id, casino_game.game_state
	else:
		return 0, "{}", 


def build_symbols(for_payout):
	all_symbols = []

	for payout in payout_to_symbols:
		for symbol in payout_to_symbols[payout]:
			all_symbols.append(symbol)

	shuffle(all_symbols)

	if for_payout == 0:
		return "".join([all_symbols[0], ",", all_symbols[1], ",", all_symbols[2]])
	elif for_payout == 1:
		indices = shuffle([0, 1, 2])
		symbol_set = ["", "", ""]
		match_a = indices[0]
		match_b = indices[1]
		nonmatch = indices[2]
		matching_symbol = all_symbols[0]
		other_symbol = all_symbols[1]
		symbol_set[match_a] = matching_symbol
		symbol_set[match_b] = matching_symbol
		symbol_set[nonmatch] = other_symbol

		return "".join([symbol_set[0], ",", symbol_set[1], ",", symbol_set[2]])
	else:
		relevantSymbols = shuffle(payout_to_symbols[for_payout])
		symbol = relevantSymbols[0]

		return "".join([symbol, ",", symbol, ",", symbol])


def build_text(wager_value, result, currency):
	if result == 0:
		return f'Lost {wager_value} {currency}'
	elif result == 1:
		return 'Broke Even'
	elif result == 12:
		return f'Jackpot! Won {wager_value * (result-1)} {currency}'
	else:
		return f'Won {wager_value * (result-1)} {currency}'


def determine_payout():
	value = random.randint(1, 100)
	if value == 100:
		return 12
	elif value >= 96:
		return 5
	elif value >= 88:
		return 3
	elif value >= 72:
		return 2
	elif value >= 61:
		return 1
	else:
		return 0


def shuffle(stuff):
	random.shuffle(stuff)
	return stuff