def createMenu(moneyManager):
menu = Menu()
menu.addOption('Deposit', moneyManager.deposit)
menu.addOption('Withdraw', moneyManager.withdraw)
menu.addOption('Status', moneyManager.status)
menu.addOption('Update', moneyManager.update)
menu.addOption('Calculate Savings', moneyManager.calculateSavings)
menu.addOption('Create a sandbox', MoneySandbox, moneyManager)
return menu
def _withdrawAlloc(self, amount):
# Check if we even have suballocations to worry about
if len(self.subAllocs) > 0:
menu = Menu()
for subAlloc in self.subAllocs:
menu.addOption(subAlloc.product, subAlloc.withdraw, amount)
menu.addOption("Extra", self._withdrawFromExtra, amount)
print "What suballocation did you spend the money on?"
choice = menu.run()
else:
amount = self._withdrawFromExtra(amount)
return amount
class MoneySandbox:
def __init__(self, moneyManager):
# Keep around a copy of the real reference if we decide to commit the changes
self.realMoneyManager = copy.deepcopy(moneyManager)
# Keep a backup that will never be erased until we are done with the sandbox
self.backupMoneyManager = copy.deepcopy(moneyManager)
# Copy the money manager
self.sbMoneyManager = copy.deepcopy(moneyManager)
self.sbMenu = Menu()
self._createSBMenu()
self.run()
def run(self):
'''
Run sandbox
'''
while self.sbMenu.run() != 'q': pass
def commitChanges(self):
'''
Commits the changes made in the sandbox to the money and allocations
'''
choice = 'q'
while choice.lower != 'y' and choice.lower() != 'n':
choice = raw_input("Are you sure you want to commit these changes? (y/n): ")
if choice == 'y':
# Commit changes to the real money manager
self.realMoneyManager = copy.deepcopy(self.sbMoneyManager)
def discardChanges(self):
'''
Discard all changes and start over
'''
# copy back the original
self.sbMoneyManager = copy.deepcopy(self.realMoneyManager)
def undoCommit(self):
'''
Undoes a commit action and starts over
'''
self.realMoneyManager = copy.deepcopy(self.backupMoneyManager)
self.sbMoneyManager = copy.deepcopy(self.backupMoneyManager)
def _createSBMenu(self):
'''
Create a sandbox menu
'''
self.sbMenu.addOption('Deposit', self.sbMoneyManager.deposit)
self.sbMenu.addOption('Withdraw', self.sbMoneyManager.withdraw)
self.sbMenu.addOption('Status', self.sbMoneyManager.status)
self.sbMenu.addOption('Update', self.sbMoneyManager.update)
self.sbMenu.addOption('Commit Changes', self.commitChanges)
self.sbMenu.addOption('Discard Changes', self.discardChanges)
self.sbMenu.addOption('Undo Last Commit', self.undoCommit)
def createMenu(inventoryTable):
menu = Menu(writeInventory, inventoryTable)
menu.addOption("Status", status, inventoryTable)
menu.addOption("Daily Deduction", dailyUsage, inventoryTable)
return menu
class AllocationManager:
def __init__(self, allocationPath):
setContext()
self._allocationPath = allocationPath
self.allocationMap = dict()
self.withdrawMenu = Menu()
self.withdrawMenuList = []
self._readAllocations()
self._createWithdrawMenu()
def addSubAllocation(self, cat, product, percent, priority, savings, totalCost):
self.allocationMap[cat].addSubAllocation(cat, product, percent, priority, savings, totalCost)
def deposit(self, amount):
if amount != None and amount <= Decimal("0.00"):
return
# Handle debt first
amount = self._handleDebt(amount)
currentAmount = amount
for cat in self.allocationMap:
deduction = amount * self.allocationMap[cat].percent
retAmount = Decimal("0.00")
if deduction >= currentAmount:
retAmount = self.allocationMap[cat].deposit(currentAmount)
currrentAmount = ("0.00")
# stop here can't deposit anymore
return
else:
retAmount = self.allocationMap[cat].deposit(deduction)
currentAmount -= deduction
if retAmount > Decimal("0.00"):
ErrorMsg("Did not deposit the full amount into category {0}, you have ${0} left!".format(cat, retAmount))
elif retAmount < Decimal("0.00"):
ErrorMsg("Over deposited amount into category, you are ${0} in debt!".format(cat, retAmount))
if currentAmount > Decimal("0.00"):\
self.allocationMap[EXTRA_KEY].extraMoney += currentAmount
elif currentAmount < Decimal("0.00"):
ErrorMsg("Over deposited amount, you are ${0} in debt!".format(currentAmount))
def withdraw(self):
print "What did you spend money on?"
choice = self.withdrawMenu.run()
if choice != 'q':
cat = self.withdrawMenuList[choice - 1]
if cat == DEBT_KEY:
WarningMsg("You can not withdraw money from your debt!")
return
amount = Decimal(raw_input("How much money did you spend? "))
amount = self.allocationMap[cat].withdraw(amount)
if amount < Decimal("0.00"):
# We are in debt here so recalculate the total debt
self.calculateDebt()
WarningMsg("You have accumulated a debt of ${0}!".format(self.allocationMap[DEBT_KEY].debt.quantize(TWOPLACES)))
def status(self):
totalPercent = Decimal("0.00")
print "{0} Status {1}".format('=' * 10, '=' * 10)
for cat in self.allocationMap:
self.allocationMap[cat].status()
totalPercent += self.allocationMap[cat].percent
print "{0}/100% has been allocated.".format(totalPercent.quantize(TWOPLACES) * 100)
def update(self):
pass
def calculateSavings(self):
savings = Decimal("0.00")
for cat in self.allocationMap:
savings += self.allocationMap[cat].calculateSavings()
return savings
def finalize(self):
self._writeAllocations()
def calculateDebt(self):
'''
This functino is needed because the debt might occur in the same allocation twice and this will
resolve that issue by calculating the debt from 0.
'''
self.allocationMap[DEBT_KEY].debt = self.allocationMap[DEBT_KEY].debtReg
for cat in self.allocationMap:
if cat != DEBT_KEY:
self.allocationMap[DEBT_KEY].debt += self.allocationMap[cat].debt
def _handleDebt(self, amount):
if self.allocationMap[DEBT_KEY].debt < Decimal("0.00"):
posDebt = self.allocationMap[DEBT_KEY].debt * -1
if posDebt > amount:
self.allocationMap[DEBT_KEY].debt += amount
self.allocationMap[DEBT_KEY].debtReg += amount
amount = Decimal("0.00")
WarningMsg("The deposited amount did not resolve your debt! You still have a debt of ${0}!".format(self.allocationMap[DEBT_KEY].debt.quantize(TWOPLACES)))
else:
amount += self.allocationMap[DEBT_KEY].debt
self.allocationMap[DEBT_KEY].debt = Decimal("0.00")
self.allocationMap[DEBT_KEY].debtReg = Decimal("0.00")
InfoMsg("Debt resolved. Depositing ${0}.".format(amount.quantize(TWOPLACES)))
return amount
def _createWithdrawMenu(self):
for cat in self.allocationMap:
self.withdrawMenu.addOption(cat)
self.withdrawMenuList.append(cat)
def _readAllocations(self):
with open(self._allocationPath, 'r') as aFile:
alines = aFile.readlines()
for i in xrange(len(alines)):
if i == 0:
continue
category, percent, priority = alines[i].strip().split(',')
self.allocationMap.update({category : Allocation(cat=category, percent=percent, priority=priority)})
self.allocationMap.update({EXTRA_KEY: Allocation(cat=EXTRA_KEY, percent=Decimal("0.00"), priority=(len(self.allocationMap) + 1))})
self.allocationMap.update({DEBT_KEY: Allocation(cat=DEBT_KEY, percent=Decimal("0.00"), priority=(len(self.allocationMap) + 1))})
def _writeAllocations(self):
with open(self._allocationPath, 'w') as aFile:
aFile.write("Category,Percentage,Priority\n")
for cat in self.allocationMap:
if cat != EXTRA_KEY and cat != DEBT_KEY:
aFile.write(str(self.allocationMap[cat]))