#! /usr/bin/env python

#
#	Bank client.
#
#	The client searches the two banks and performs a set of operations.
#	(the banks are searched simply by listing the :banks namespace!)
#

import sys
import Pyro.naming, Pyro.core

from banks import BankError

group = ':banks'  # the default namespace group

# initialize the client and set the default namespace group
Pyro.core.initClient()
Pyro.config.PYRO_NS_DEFAULTGROUP=group

# locate the NS
locator = Pyro.naming.NameServerLocator()
print 'Searching Naming Service...',
ns = locator.getNS()

print 'Naming Service found at',ns.URI.address,'('+(Pyro.protocol.getHostname(ns.URI.address) or '??')+') port',ns.URI.port

# List the banks.
# This is done by simply looking in the :banks namespace, to see what
# banks have registered. The filter is for removing any groups that could
# be in the namespace (the type of real names is 1).
banknames = filter(lambda x: x[1]==1, ns.list(group))
banknames = map(lambda (x,y): x, banknames) # keep only the object name
if not banknames:
	raise RuntimeError('There are no banks to do business with!')

banks={}	# banks (proxies)


print
for name in banknames:
	print 'Found a bank: ',name
	try:
		URI=ns.resolve(name)
	except Pyro.core.PyroError,x:
		print 'Bank can\'t be found:',x
		raise SystemExit

	# create a proxy for the bank object
	banks[name] = Pyro.core.getAttrProxyForURI(URI)

def selectBank():
	i = 1
	banknames=banks.keys()
	for b in banknames:
		print i," ",b
		i=i+1
	b = input("Select a bank: ")
	return banks[banknames[b-1]]
	
def createAccount():
	print "\nCreate Account."
	bank = selectBank()
	name = raw_input("Enter name: ")
	a = bank.createAccount(name)
	amount = input("Initial deposit: ")
	a.deposit(amount)
	print "Balance:", a.balance
	

def removeAccount():
	print "\nRemove Account."
	bank = selectBank()
	name = raw_input("Enter name: ")
	bank.deleteAccount(name)

def viewBalance():
	print "\nView Balance."
	bank = selectBank()
	name = raw_input("Enter name: ")
	ac = bank.findAccount(name)
	print ac.balance

def deposit():
	print "\nDeposit."
	bank = selectBank()
	name = raw_input("Enter name: ")
	ac = bank.findAccount(name)
	amount = input("Amount: ")
	ac.deposit(amount)
	print "New balance:", ac.balance

def withdraw():
	print "\nWithdraw."
	bank = selectBank()
	name = raw_input("Enter name: ")
	ac = bank.findAccount(name)
	amount = input("Amount: ")
	ac.withdraw(amount)
	print "New balance:", ac.balance

def listAll():
	print "\nList all accounts."
	for (bankname,bank) in banks.items():
		print bank.name
		accs = bank.allAccounts()
		if not accs:
			print "   No accounts."
		for a in accs:
			print "  ",a.name, a.balance

going = 1

while going:
	print "\n---- menu ----"
	print "1: create account"
	print "2: remove account"
	print "3: view balance"
	print "4: list all accounts"
	print "5: deposit money"
	print "6: withdraw money"
	print "0: exit"
	print
	try:
		choice = input("Choice: ")
		if choice==0:	going=0
		elif choice==1:	createAccount()
		elif choice==2:	removeAccount()
		elif choice==3:	viewBalance()
		elif choice==4:	listAll()
		elif choice==5:	deposit()
		elif choice==6:	withdraw()
	except BankError,x:
		print 'Problem:',x
	except StandardError,x:
		print 'Try again (input incorrect?)'
		raise

		
