forked from rDrama/rDrama
113 lines
3.3 KiB
Python
113 lines
3.3 KiB
Python
from json.encoder import INFINITY
|
|
import random
|
|
from files.helpers.const import *
|
|
|
|
def shuffle(stuff):
|
|
random.shuffle(stuff)
|
|
return stuff
|
|
|
|
class Slots:
|
|
command_word = "!slots"
|
|
casino_word = "!casino"
|
|
if SITE == 'rdrama.net': minimum_bet = 100
|
|
else: minimum_bet = 10
|
|
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}' |