def display_client_menu():
'''
Displays a program menu for user to choose an option
'''
try:
# Clears the screen
secure_utils.clear_screen()
while True:
# Display My Logo
secure_utils.draw_client_logo()
print '----------------------------------\n'
print'1. Set Host Name/IP To Use\n2. Set Port Number To Use\n3. Connect To Secure Chat Session\n4. Program Information\n99. Exit Program\n'
print '----------------------------------\n'
user_chosen_option = raw_input('Enter an option >> ')
secure_utils.clear_screen()
if(user_chosen_option == '1'):
set_host_name()
elif(user_chosen_option == '2'):
set_port_number()
elif(user_chosen_option == '3'):
connect_to_server()
elif(user_chosen_option == '4'):
secure_utils.program_info()
elif(user_chosen_option == '5'):
print "Coming Soon"
elif(user_chosen_option == '6'):
print "Coming Soon"
elif(user_chosen_option == '98'):
print "Coming Soon"
elif(user_chosen_option == '99'):
break
else:
print('[--] ERROR: Invalid Option.')
raw_input("Press [ENTER] To Return To The Main Menu...\n")
secure_utils.clear_screen()
except KeyboardInterrupt:
print'[--] CTRL+C Pressed.'
except Exception, e:
print'[--] ERROR: %s' % str(e)
def display_server_menu():
'''
Displays a program menu for user to choose an option
'''
try:
# Clears the screen
secure_utils.clear_screen()
while True:
# Display My Logo
secure_utils.draw_server_logo()
print '----------------------------------\n'
print'1. Change Port To Use\n2. Run Server\n3. View/Add Blocked IPs\n4. Remove Blocked IPC\n5. Program Information\n99. Exit Program\n'
print '----------------------------------\n'
user_chosen_option = raw_input('Enter an option>> ')
secure_utils.clear_screen()
if(user_chosen_option == '1'):
set_port_number()
elif(user_chosen_option == '2'):
start_server_listener()
elif(user_chosen_option == '3'):
secure_utils.add_ip_to_ban_list()
elif(user_chosen_option == '4'):
secure_utils.remove_ip_from_ban_list()
elif(user_chosen_option == '5'):
secure_utils.program_info()
elif(user_chosen_option == '99'):
break
else:
print('[--] ERROR: Invalid Option.')
raw_input("Press [ENTER] To Return To The Main Menu...\n")
secure_utils.clear_screen()
except KeyboardInterrupt:
print'[--] CTRL+C Pressed.'
except Exception, e:
print'[--] ERROR: %s' % str(e)
