def main():
#Getting here implies all pre-requisites have been met (NOT FINISHED!)
accountName = options.a
ipAddress = options.i
port = int(options.p)
request = {}
response = {}
request["accountName"] = accountName
request["pin"] = getCardPin()
response["account"] = accountName
if options.n:
request["action"] = "new"
request["amount"] = float(options.n)
response["initial_balance"] = float(options.n)
elif options.d:
request["action"] = "deposit"
request["amount"] = float(options.d)
response["deposit"] = float(options.d)
elif options.w:
request["action"] = "withdraw"
request["amount"] = float(options.w)
response["withdraw"] = float(options.w)
elif options.g:
request["action"] = "get"
request = {"request" : request}
# print >> sys.stderr, request
sys.stdout.flush()
message = NetMsg(request)
encodedMessage = message.encryptedJson(getAuthKey(), message.getJson())
# Create socket and send data to bank
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
sock.connect((ipAddress, port))
sock.sendall(encodedMessage)
sock.settimeout(10.0)
received = sock.recv(1024)
sock.settimeout(None)
received = json.loads(NetMsg.decryptJson(getAuthKey(), received))
if received["msg"] == "transaction_completed":
if options.g:
response["balance"] = received["balance"]
print json.dumps(response)
sys.stdout.flush()
elif received["msg"] == "transaction_failed":
sys.exit(255)
elif received["msg"] == "protocol_error":
sys.exit(63)
except socket.error, e:
print >> sys.stderr, e
sys.stdout.flush()
sys.exit(63)
def parse_request(self, data):
try:
# decrypt message
net_msg = json.loads(NetMsg.decryptJson(auth_key, data))
msg = net_msg["msg"]
response = {}
response["msgID"] = net_msg["msgID"] + 1
response["code"] = 0
# perform atm request
request = msg["request"]
if request["action"] == "new":
if request["amount"] < 10:
self.print_to_stderr("Amount was less than 10\n")
response["msg"] = "transaction_failed"
self.send_ack(json.dumps(response))
# self.send_ack("transaction_failed")
return
if self.get_acct_w_acct_name_n_card_key(request["accountName"],
request["pin"]):
self.print_to_stderr("Account already exists\n")
response["msg"] = "transaction_failed"
# self.send_ack("transaction_failed")
return
new_account = Account(request["pin"],
request["accountName"], request["amount"])
self.accounts.append(new_account)
response["msg"] = "transaction_completed"
self.send_ack(json.dumps(response))
# self.send_ack("transaction_completed")
elif request["action"] == "deposit":
if request["amount"] <= 0:
self.print_to_stderr("Amount was less than or equal to 0\n")
response["msg"] = "transaction_failed"
self.send_ack(json.dumps(response))
# self.send_ack("transaction_failed")
return
account = self.get_acct_w_acct_name_n_card_key(request[
"accountName"],
request["pin"])
if account is None:
self.print_to_stderr("Could not verify user account\n")
response["msg"] = "transaction_failed"
self.send_ack(json.dumps(response))
# self.send_ack("transaction_failed")
return
else:
account.deposit(request["amount"])
response["msg"] = "transaction_completed"
response["balance"] = account.get_balance()
self.send_ack(json.dumps(response))
# self.send_ack("transaction_completed")
elif request["action"] == "withdraw":
# account = self.get_acct_w_acct_name_n_card_key(request["pin"])
account = self.get_acct_w_acct_name_n_card_key(request[
"accountName"],
request["pin"])
if account is None:
self.print_to_stderr("Could not verify user account\n")
response["msg"] = "transaction_failed"
self.send_ack(json.dumps(response))
# self.send_ack("transaction_failed")
return
else:
if not account.withdraw(request["amount"]):
response["msg"] = "transaction_failed"
response["code"] = 419
self.send_ack(json.dumps(response))
# self.request.sendall("419")
else:
# self.send_ack("transaction_completed")
response["msg"] = "transaction_completed"
response["balance"] = account.get_balance()
self.send_ack(json.dumps(response))
elif request["action"] == "get":
account = self.get_acct_w_acct_name_n_card_key(request[
"accountName"],
request["pin"])
if account is None:
self.print_to_stderr("Could not verify user account\n")
return
else:
account.current_balance()
response["msg"] = "transaction_completed"
response["balance"] = account.get_balance()
self.send_ack(json.dumps(response))
# self.send_ack(str(account.get_balance()))
except ValueError:
# self.request.sendall("protocol_error")
response["msg"] = "protocol_error"
self.send_ack(json.dumps(response))
print "protocol_error"