BankController.py

from __future__ import division  # to allow fraction division
from simpy import *
from Customer import *
from Bank import *
from Player import *
from GlobalDeclaration import *
from Tools import *

class BankController:
    """This bank simulates a bank. Customers can go into the bank and carry out transactions"""
    def __init__(self, env, player):
        self.env = env
        self.player = player
        self.bank = 0


        #todo : add in save money part

    def setup(self):
        checking_account = CheckingAccount(G.player_initial_money,self.env) # a checking acount 
        saving_account = SavingAccount(0,self.env) # a saving account 
        self.player.set_bank_account(checking_account,saving_account) #give the above accounts to the player
        self.bank = Bank(self.player) #set up the bank

    


    def checking_to_saving(self):
        time = 0
        print "You're transfering money from your checking account to your saving account. "
        while True: 
            time = check_positive_valid_input("Please enter the time you keep money in your saving account? Units are in months."+'\n'\
                +"Please note you can't save money for less than a month")
            if time/1 >=1 :
                break
            print "Please enter a valid input."   
        interest_rate = 0.05#self.bank.interest_rate(time,False)
        print "The interest rate for your saving is %.5f " %interest_rate  #print out 5 effective digits
        wanna_save = check_confirm("Print Y to continue your transfering money to your saving account with above interest rate. Press N to exit.") 
        if wanna_save : 
            amount = self.bank.money  #initialize amount to enter with a large number to make sure we'll be in while loop
            while amount > self.player.checking_account.balance : 
                amount = check_positive_valid_input("Please enter the amount you want to keep in saving account : ")
                if amount < self.player.checking_account.balance:
                    break
                print "Please enter the right input."
            self.player.checking_account.withdraw(amount) #first, withdraw the money from checking account
            self.player.saving_account.save(amount) #then put it in saving account 
            G.tillWithdraw = time*4-1   #set the countdown, unit in weeks
            self.player.saving_account.interest_rate = interest_rate
            self.player.saving_account.interest= amount * interest_rate
        else : 
            return

    def saving_to_checking (self):
        if self.player.saving_account.balance == 0:
            print "You don't have any money in your saving account."
            return
        print "You have %d dollars on your saving account." %self.player.saving_account.balance
        amount = 0
        while True:
            amount = check_positive_valid_input("Please enter the amount you want to transfer from saving account to checking account.")
            if amount <= self.player.saving_account.balance:
                break
            print "Not enough money in saving account to do the transfer. Please enter a smaller amount."
        if G.tillWithdraw != 0:
            print "You have to wait %d months to get interest from your saving. " %G.tillWithdraw
            go_on = check_confirm("Enter Y to continue transferming money to checking without any interest. Enter N to stop.")
            if not go_on :
                return

            self.player.saving_account.interest = (self.player.saving_account_balance-amount)*self.player.saving_account.interest_rate 
        to_checking = amount + self.player.saving_account.interest   #remaining principal plus remaining interest 
        self.player.saving_account.withdraw(amount)    #first, withdraw the money from saving account
        self.player.checking_account.save(to_checking) #then put it in checking account 


    def system_saving_withdraw(self,amount):
        if G.tillWithdraw != 0:
            self.player.saving_account.interest = (self.player.saving_account_balance-amount)*self.bank.interest_rate  
            print "You lose your interest on your %d dollars in your saving account. " %amount
        self.player.saving_account.withdraw(amount)



    def loan(self):
        if self.player.loan != 0:
            print "You can't loan more money because you still need to pay the bank back %d dollars." %self.player.loan
            return
        time = 0
        while True: 
            time = check_positive_valid_input("Please enter the time you want keep this loan? Units are in months."+'\n'\
                +"Please note you can't loan money for less than a month")
            if time/1 >= 1 :
                break
            print "Please enter a valid input."   
        interest_rate = self.bank.interest_rate(time,True)
        print "Your personal credit is %d" %self.player.credit
        print "The interest rate for your loan is %.5f " %interest_rate  #print out 5 effective digits
        wanna_loan = check_confirm("Print Y to continue your loan application with above interest rate. Press N to exit.")
        if wanna_loan : 
            amount = check_positive_valid_input("Please enter the amount you want to loan from the bank ? ")
            G.tillPay = time*4 -1 #set the countdown to pay the money, unit in weeks 
            self.player.loan = amount*(1 + interest_rate)  #record loan amount and interest together 
            self.player.checking_account.save(amount) #transfer the money to player's checking account
            print "Sucessfully borrow %d dollars from the bank. You have to pay %d dollars including interest back %d months later." %(amount,self.player.loan,time)
        else : 
            return


    def pay_loan(self):
        if  self.player.checking_account.balance >= self.player.loan  :
            print "Paying back all the %d dollars loan and interest using your checking account " %self.player.loan  
            self.player.checking_account.withdraw(self.player.loan )  #pay back the loan by withdrawing money from checking account
            self.player.loan = 0
            return
        if  (self.player.checking_account.balance + self.player.saving_account.balance) >= self.player.loan :
            print "Paying back the %d dollars of your totoal %d loan and interest using your checking account. " %(self.player.checking_account.balance,self.player.loan)
            self.player.checking_account.withdraw(self.player.checking_account.balance) #first, use all money in checking account 
            print "Paying back the remaining %d dollars of loan and interest using your saving account. " %(self.player.loan) 
            self.player.loan  -= self.player.checking_account.balance  #deduct the paid part from self.player.loan 
            self.system_saving_withdraw(self.player.loan) # next, pay the rest of the money using the money in saving account
            return
        #has not enough money to pay.
        else :
            original_loan = self.player.loan
            print "Paying back the %d dollars of your totoal %d loan using your checking account. " %(self.player.checking_account.balance,self.player.loan) 
            self.player.loan  -= self.player.checking_account.balance #deduct the paid part from self.player.loan 
            self.player.checking_account.withdraw(self.player.checking_account.balance) #first, use all money in checking account 
            print "Paying back the %d dollars of the remaining %d loan using your saving account. " %(self.player.saving_account.balance,self.player.loan) 
            self.player.loan -= self.player.saving_account.balance  #deduct the paid part from self.player.loan 
            self.player.system_saving_withdraw(self.player.saving_account_balance)# next, pay the rest of the money using the money in saving account
            deduct = 100*(self.player.loan/orginial_loan)
            print "You don't have enough money to pay back your loan. You still have %d dollars to be paid back. As a result, we deduct %d points from your credit points" %(self.loan,deduct)
            self.player.credit -= deduct   #deduct credit based on how much you couldn't repaly

        
    def check_checking(self):
        print "Your checking account balance is %d dollars. " %self.player.checking_account.balance 


    def check_saving(self):
        print "Your saving account balance is %d dollars. " %self.player.saving_account.balance   
        if self.player.saving_account.balance  != 0:
            print "You have %d weeks till withdraw the money." %G.tillWithdraw  
        






if __name__ == '__main__':
    env = Environment()
    player = Player('Neil')
    bc = BankController(env,player)
    bc.setup()
    bc.loan()