import pickle

def menu():
	print"To enter money, press 1"
	print"To take out money, press 2"
	print"To view your statistics, press 3"
	print"To exit, press 4"
	choice = raw_input("")
	while choice != 1 or 2 or 3 or 4:
		print"Invalid entry"
		print"Enter either 1 or 2 or 3 or 4"
		choice = raw_input("")
	return choice

def fileopen()
	file = open("C:\moneymanagerinfo", "r+")
	try:
		dict = pickle.load(file)
	except:
		dict = {}
		pickle.dump(dict, file)              
	return file

def main()
	if choice != 4:
		
		if choice == 1:
			for x in range(1, 5):
				print"To enter money into account number %d, press %d" % (x)
			secondchoice = input("")
			amount = input("Amount to enter: ")
			for x in range(1, 5):
				if secondchoice == x:
					try:
						dict["accountx"] = amount + dict["accountx"]
					except:
						dict["accountx"] = amount
						
			try:
				dict["total"] = amount + dict["total"]
			except:
				dict["total"] = amount
				
			try:
				dict["totalentered"] = secondchoice + dict["totalentered"]
			except:
				dict["totalentered"] = secondchoice
			del secondchoice
			del amount
			menu()
		
		if choice == 2:
			for x in range(1, 5):
				print"To take money from account number %d, press %d" % (x)
			secondchoice = input("")
			amount = input("Amount to take out: ")
			for x in range(1, 5):
				if secondchoice == x:
					try:
						dict["accountx"] = dict["accountx"] - amount
					except:
						print"Error, account does not exist"
			
			try:
				dict["total"] = dict["total"] - amount
			except:
				dict["total"] = - amount
						
			try:
				dict["totaltaken"] = dict["totaltaken"] + amount
			except:
				dict["totaltaken"] = amount
			del secondchoice
			del amount
			menu()
		
		if choice == 3:
			print"Total amount entered to date is %d" % (dict["totalentered"])
			print"Total amount taken to date is %d" % (dict["totaltaken"])
			print"Total money in accounts is %d" % (dict["total"])
			for x in range(1, 5):
				print"Total money in acount %d is %d" % (x, dict["accountx"])
			menu()
	
	pickle.dump(dict, file)

fileopen()
menu()
main()