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)