rDrama/files/classes/treasure.py

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()