def area_chart(): ips = get_ips() crits = get_crits() high = get_high() medium = get_medium() low = get_low() stacked_area_chart(ips, crits, high, medium, low)
def basic_scatter_chart(data_option): ips = get_ips() if data_option == '1': score = get_score() scatter_plot(ips, score) elif data_option == '2': total_vuln = get_total_vulnerabilities() scatter_plot(ips, total_vuln) elif data_option == '3': crits = get_crits() scatter_plot(ips, crits) elif data_option == '4': high_vuln = get_high() scatter_plot(ips, high_vuln) elif data_option == '5': medium_vuln = get_medium() scatter_plot(ips, medium_vuln) elif data_option == '6': low_vuln = get_low() scatter_plot(ips, low_vuln) elif data_option == '7': info_vuln == get_info_vulns() scatter_plot(ips, info_vuln)
def the_equalizer():
os.system("clear")
print("Enter information for Nessus Server:")
username = raw_input("\nUsername: "******"VisProj"
password = getpass.getpass() #"Password1234"
port = raw_input("Port the server is running on: ") #"8834"
create_token(username, password, port)
token = get_token()
os.system("clear")
printer.banner()
while 1:
printer.print_main_menu()
option = raw_input(printer.header)
if option == 'c' or option == 'C':
os.system("clear")
printer.banner()
elif option == 'e' or option == 'E':
os.system("rm -rf token.txt")
os.system("rm -rf report.txt")
sys.exit()
elif option == '1':
get_info(token, port, '2')
option = raw_input("\n(press q to quit): ")
while option != 'q':
option = raw_input("\n(press q to quit): ")
the_equalizer()
elif option == '2':
report_id = raw_input(printer.report_id)
get_report(token, port, report_id, '2')
option = raw_input("\n(press q to quit): ")
while option != 'q':
option = raw_input("\n(press q to quit): ")
the_equalizer()
elif option == '3':
report_id = raw_input(printer.report_id)
get_report(token, port, report_id, '3')
ips = get_ips()
vulns = get_total_vulnerabilities()
table = BeautifulTable()
table.column_headers = ("IP", "# of Vulnerabilities")
print "\n"
x = 0
while x < len(ips):
table.append_row([ips[x], vulns[x]])
x += 1
print table
option = raw_input("\n(press q to quit): ")
while option != 'q':
option = raw_input("\n(press q to quit): ")
elif option == '4':
chart_menu(token, port)
os.system("clear")
printer.banner()
os.system("rm -rf token.txt")
os.system("rm -rf report.txt")