def extraMenu(self): ''' Menu com opções extra, tais como imprimir a informação em PDF, CSV entre outros. Este menu será apresentado no fim de cada acção ''' while True: os.system("clear") extraMenu = open("menus/extraOptions.txt", "r") print bcolors.VERDE + extraMenu.read() + bcolors.ENDC resposta = raw_input(bcolors.AZUL + "Faça a sua escolha: " + bcolors.ENDC) if resposta == "1": if len(self.valuesToSavePDF) > 3: pdf = PDF() pdf.setTitle("Active Services") pdf.alias_nb_pages() pdf.add_page() pdf.set_font('Times','',12) for i in self.valuesToSavePDF: pdf.cell(0,5, i ,0,1) pdf.output('scanningConnections.pdf','F') print bcolors.AMARELO + "PDF gerado com sucesso" + bcolors.ENDC raw_input("Prima enter para continuar...") break else: print bcolors.VERMELHO + "Não é possivel gerar qualquer resultado" + bcolors.ENDC raw_input("Prima enter para continuar...") break pass elif resposta == "2": if len(self.valuesToSaveCSV) > 3: with open('scanningConnections.csv', 'wb') as csvfile: spamwriter = csv.writer(csvfile, delimiter=',') for i in self.valuesToSaveCSV: #print len(i) spamwriter.writerow(i) print bcolors.AMARELO + "CSV gerado com sucesso" + bcolors.ENDC raw_input("Prima enter para continuar...") break else: print bcolors.VERMELHO + "Não é possivel gerar qualquer resultado" + bcolors.ENDC raw_input("Prima enter para continuar...") break pass elif resposta == "0": break pass pass
def extraMenu(self): ''' Menu com opções extra, tais como imprimir a informação em PDF, CSV entre outros. Este menu será apresentado no fim de cada acção ''' while True: os.system("clear") extraMenu = open("menus/extraOptionsLog.txt", "r") print bcolors.VERDE + extraMenu.read() + bcolors.ENDC resposta = raw_input(bcolors.AZUL + "Faça a sua escolha: " + bcolors.ENDC) if resposta == "1": if len(self.lista) > 1: pdf = PDF() pdf.setTitle("FileLog") pdf.alias_nb_pages() pdf.add_page() pdf.set_font('Times','',12) for i in self.lista: pdf.cell(0,5, str(i[0] + " - "+ i[1] + " - " + i[2] + " - " + i[3] + " - " + i[4]) ,0,1) pdf.output('fileLog.pdf','F') print bcolors.AMARELO + "PDF gerado com sucesso" + bcolors.ENDC raw_input("Prima enter para continuar...") break else: print bcolors.VERMELHO + "Não é possivel gerar qualquer resultado" + bcolors.ENDC raw_input("Prima enter para continuar...") break pass elif resposta == "2": if len(self.lista) > 1: with open('fileLog.csv', 'wb') as csvfile: spamwriter = csv.writer(csvfile, delimiter=',') for i in self.lista: #print len(i) spamwriter.writerow(i) print bcolors.AMARELO + "CSV gerado com sucesso" + bcolors.ENDC raw_input("Prima enter para continuar...") break else: print bcolors.VERMELHO + "Não é possivel gerar qualquer resultado" + bcolors.ENDC raw_input("Prima enter para continuar...") break pass elif resposta == "3": self.infoForGraph() #Construção do grafico #figure(figsize=(4, 2)) # image dimensions title("Try Connections - FileLog", size='large') ylabel("How many access") xlabel("Countries") # add bars for i, key in zip(range(len(self.codeFreq)), self.codeFreq.keys()): bar(i + 0.25 , self.codeFreq[key], color='red') # axis setup xticks(arange(0.65, len(self.codeFreq)), [('%s: %d' % (name, value)) for name, value in zip(self.codeFreq.keys(), self.codeFreq.values())], size='large') max_value = max(self.codeFreq.values()) tick_range = arange(0, max_value, (max_value / 7)) yticks(tick_range, size='large') formatter = FixedFormatter([str(x) for x in tick_range]) gca().yaxis.set_major_formatter(formatter) gca().yaxis.grid(which='major') show() pass elif resposta == "0": break pass pass