rDrama/files/classes/slots.py

from json.encoder import INFINITY
import random

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

class Slots:
	command_word = "!slots"
	casino_word = "!casino"
	minimum_bet = 100
	maximum_bet = INFINITY
	payout_to_symbols = {
		2: ["👣", "🍀", "🌈", "⭐️"],
		3: ["🍎", "🔞", "⚛️", "☢️"],
		5: ["✡️", "⚔️", "🍆", "🍒"],
		12: ["🐱"]
	}

	def __init__(self, g):
		self.db = g.db
			
	def check_for_slots_command(self, in_text, from_user, from_comment):
		if self.command_word in in_text:
			for word in in_text.split():
				if self.command_word in word:
					try:
						wager = word[len(self.command_word):]
						wager_value = int(wager)
					except: break

					if (wager_value < self.minimum_bet): break
					elif (wager_value > self.maximum_bet): break
					elif (wager_value > from_user.coins): break

					from_user.coins -= wager_value

					payout = self.determine_payout()
					symbols = self.build_symbols(payout)
					text = self.build_text(wager_value, payout, from_user, "Coins")
					reward = wager_value * payout

					from_user.coins += reward
					self.db.add(from_user)

					from_comment.slots_result = f'{symbols} {text}'
					self.db.add(from_comment)

		if self.casino_word in in_text:
			for word in in_text.split():
				if self.casino_word in word:
					try:
						wager = word[len(self.casino_word):]
						wager_value = int(wager)
					except: break

					if (wager_value < self.minimum_bet): break
					elif (wager_value > self.maximum_bet): break
					elif (wager_value > from_user.procoins): break

					from_user.procoins -= wager_value

					payout = self.determine_payout()
					symbols = self.build_symbols(payout)
					text = self.build_text(wager_value, payout, from_user, "Marseybux")
					reward = wager_value * payout

					from_user.procoins += reward
					self.db.add(from_user)

					from_comment.slots_result = f'{symbols} {text}'
					self.db.add(from_comment)


	def determine_payout(self):
		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 build_symbols(self, for_payout):
		all_symbols = []
		
		for payout in self.payout_to_symbols:
			for symbol in self.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: return "".join([all_symbols[0], all_symbols[0], all_symbols[2]])
		else:
			relevantSymbols = shuffle(self.payout_to_symbols[for_payout])
			symbol = relevantSymbols[0]
			return "".join([symbol, symbol, symbol])
	
	def build_text(self, wager_value, result, user, currency):
		if result == 0:
			user.winnings -= wager_value
			return f'Lost {wager_value} {currency}'
		elif result == 1:
			return 'Broke Even'
		elif result == 12:
			user.winnings += wager_value * (result-1)
			return f'Jackpot! Won {wager_value * result} {currency}'
		else:
			user.winnings += wager_value * (result-1)
			return f'Won {wager_value * result} {currency}'