def receive(self, conn, addr):
msg = None
try:
msg = conn.recv(MESSAGE_LENGTH)
log.debug(msg)
msg, params = msg.split("~")
parameters = self.extractParams(params)
except Exception as e:
log.error('Error while receiving message from %s' % addr)
log.error(e)
raise e
return msg, parameters
def receive(self, conn, addr):
msg = None
try:
msg = conn.recv(MESSAGE_LENGTH)
log.debug(msg)
msg, params = msg.split("~")
parameters = self.extractParams(params)
except Exception as e:
log.error('Error while receiving message from %s' % addr)
log.error(e)
raise e
return msg, parameters
def start(self, conn, addr):
self.respond(conn, "connecting...", str("type=valid"))
log.info('Session started with %s' % addr)
login_params = None
debug = None
try:
while True:
request_message, request_params = self.receive(conn, addr)
# Get response message and parameters
response_params = None
response_msg = None
debug = False
log.info('Request from %s - %s' % (addr, request_message))
if request_message == "authenticate":
login_params = request_params
response_msg, account_type = Bank().login(
request_params['email'], request_params['password'])
response_params = str("type=" + account_type)
elif request_message == "logout":
response_msg = "Logout Successful"
del Bank().logged_ins[login_params['email']]
else:
response_msg, response_params, debug = self.bank_operation(
request_message, request_params)
# Respond to client
self.respond(conn, response_msg, response_params)
if debug:
log.debug('Response to %s - %s' % (addr, response_msg))
log.info('Passbook sent to %s' % (addr))
else:
log.info('Response to %s - %s' % (addr, response_msg))
# Close connection if authentication failed or logout
if ("Login Unsuccessful" in response_msg
or response_msg == "Logout Successful"):
conn.close()
break
except Exception as e:
if login_params['email'] in Bank().logged_ins:
del Bank().logged_ins[login_params['email']]
log.error(e)
log.error('Error after menu ' + str(addr))
finally:
self.count -= 1
conn.close()
