forked from MarseyWorld/MarseyWorld
35 lines
833 B
Python
35 lines
833 B
Python
import random
|
|
|
|
class Treasure:
|
|
special_min = 100
|
|
special_max = 1000
|
|
standard_min = 10
|
|
standard_max = 100
|
|
|
|
def __init__(self, g):
|
|
self.db = g.db
|
|
|
|
def check_for_treasure(self, in_text, from_comment):
|
|
if '!slots' not in in_text and '!blackjack' not in in_text:
|
|
seed = random.randint(1, 1000)
|
|
is_special = seed == 1000
|
|
is_standard = seed >= 990
|
|
amount = 0
|
|
|
|
if is_special:
|
|
amount = random.randint(self.special_min, self.special_max)
|
|
elif is_standard:
|
|
amount = random.randint(self.standard_min, self.standard_max)
|
|
if random.randint(1, 100) > 75: amount = -amount
|
|
|
|
if amount != 0:
|
|
user = from_comment.author
|
|
user.coins += amount
|
|
|
|
if user.coins < 0: user.coins = 0
|
|
|
|
from_comment.treasure_amount = str(amount)
|
|
|
|
self.db.add(user)
|
|
self.db.add(from_comment)
|
|
self.db.commit() |