slots / machine.py

@ ca0aa0efce7d42603bbd58644254b69e28eb2f99 | history


import sys
import os
import random
import reels 

class Machine:
	def __init__(self,bal,coin):
		self.balance = bal
		self.bet = coin
	def printBalance(self):
		if self.balance > 0:
			print 'You have $' + str(self.balance) + ". Don't spend it all in one place."
	def addMoney(self, amt):
		self.balance += amt
	def changeBet(self, bet):
		if bet <= self.balance:
			self.bet = bet
			print 'Bet changed to $' + str(bet) + "."
		else:
			print "You can't bet more money than is in the machine!"
	def gamble(self):
		if self.balance <= 0:
			self.isBroke()
		else:
			reelA = int(random.random()*100)
			reelB = int(random.random()*100)
			reelC = int(random.random()*100)
		
			win = reels.spin(reelA, reelB, reelC)
		
			self.balance += self.bet * (win - 1)
			if win > 0:
				print "Woo! Woo! You won $" + str(win*self.bet) + "!\n"
				self.printBalance()
			else:
				print "Meh. Try again!\n"
				self.printBalance()
		if self.bet > self.balance > 0:
			print "Your wager now exceeds your balance. "
			self.changeBet(self.balance)
	def isBroke(self):
		print 'You have no more money! Get outta here!'
		sys.exit(0)