Ejemplo n.º 1
0
    def run_OnButtonClick(self):
        try:
            if len(self.md5s_dict.keys()) > 0:
                self.results = run_vt_analyse(self.md5s_dict, retrieve_apikey(self.config), self.language)

                output_file = get_output_file(self.config, self.input_file_string.get())

                # Create the output log
                save_results(output_file, self.input_file_string.get(), self.file_type, self.md5s_dict, self.results, self.language)

                # Open the log
                self.console.insert(tk.END, get_string(VariousCodes.scan_complete, self.language))
                self.console.see(tk.END)
                webopen(output_file)
            else:
                self.file_dialog_button.focus_set()
                raise ScriptWarning(ErrorsCodes.input_file_no_md5)

        except ScriptWarning as e:
            self.console.insert(tk.END, get_string(VariousCodes.warning, self.language).format(message=e.message(self.language)))
            self.console.see(tk.END)

        except ScriptError as e:
            self.console.insert(tk.END, get_string(VariousCodes.error, self.language).format(message=e.message(self.language)))
            self.console.see(tk.END)
Ejemplo n.º 2
0
 def apikey_save_OnButtonClick(self):
     try:
         self.config["apikey"] = retrieve_apikey({"apikey": self.apikey_entry.get()})
         self.apikey_string.set(self.config["apikey"])
         save_config(self.config, self.config_file)
         self.console.insert(tk.END, get_string(VariousCodes.config_save, self.language).format(property="apikey"))
         self.console.see(tk.END)
     except ScriptWarning as e:
         self.console.insert(tk.END, get_string(VariousCodes.warning, self.language).format(message=e.message(self.language)))
         self.console.see(tk.END)
         self.apikey_entry.focus_set()
Ejemplo n.º 3
0
 def apikey_save_OnButtonClick(self):
     try:
         self.config["apikey"] = retrieve_apikey(
             {"apikey": self.apikey_entry.get()})
         self.apikey_string.set(self.config["apikey"])
         save_config(self.config, self.config_file)
         self.console.insert(
             tk.END,
             get_string(VariousCodes.config_save,
                        self.language).format(property="apikey"))
         self.console.see(tk.END)
     except ScriptWarning as e:
         self.console.insert(
             tk.END,
             get_string(
                 VariousCodes.warning,
                 self.language).format(message=e.message(self.language)))
         self.console.see(tk.END)
         self.apikey_entry.focus_set()
Ejemplo n.º 4
0
    def run_OnButtonClick(self):
        try:
            if len(self.md5s_dict.keys()) > 0:
                self.results = run_vt_analyse(self.md5s_dict,
                                              retrieve_apikey(self.config),
                                              self.language)

                output_file = get_output_file(self.config,
                                              self.input_file_string.get())

                # Create the output log
                save_results(output_file, self.input_file_string.get(),
                             self.file_type, self.md5s_dict, self.results,
                             self.language)

                # Open the log
                self.console.insert(
                    tk.END,
                    get_string(VariousCodes.scan_complete, self.language))
                self.console.see(tk.END)
                webopen(output_file)
            else:
                self.file_dialog_button.focus_set()
                raise ScriptWarning(ErrorsCodes.input_file_no_md5)

        except ScriptWarning as e:
            self.console.insert(
                tk.END,
                get_string(
                    VariousCodes.warning,
                    self.language).format(message=e.message(self.language)))
            self.console.see(tk.END)

        except ScriptError as e:
            self.console.insert(
                tk.END,
                get_string(
                    VariousCodes.error,
                    self.language).format(message=e.message(self.language)))
            self.console.see(tk.END)
Ejemplo n.º 5
0
 def test_bad_lenght(self):
     with self.assertRaises(ScriptWarning) as error:
         retrieve_apikey({"apikey": "A" * 63})
     self.assertEqual(error.exception.code,
                      ErrorsCodes.apikey_invalid_lenght)
Ejemplo n.º 6
0
 def test_default(self):
     with self.assertRaises(ScriptWarning) as error:
         retrieve_apikey(default_config)
     self.assertEqual(error.exception.code,
                      ErrorsCodes.apikey_invalid_default)
Ejemplo n.º 7
0
 def test_none(self):
     with self.assertRaises(ScriptWarning) as error:
         retrieve_apikey({})
     self.assertEqual(error.exception.code, ErrorsCodes.apikey_invalid_none)
Ejemplo n.º 8
0
 def test_valid(self):
     apikey = retrieve_apikey({"apikey": "A" * 64})
     self.assertEqual(apikey, "A" * 64)