def _process_compare(self, compare_id): result = self.compare_module.compare( convert_compare_id_to_list(compare_id)) if isinstance(result, dict): self.db_interface.add_compare_result(result) else: logging.error(result)
def check_objects_exist(self, compare_id, raise_exc=True): for uid in convert_compare_id_to_list(compare_id): if not self.existence_quick_check(uid): if raise_exc: raise FactCompareException('{} not found in database'.format(uid)) return True return False
def _app_show_compare_result(self, compare_id): compare_id = normalize_compare_id(compare_id) try: with ConnectTo(CompareDbInterface, self._config) as sc: result = sc.get_compare_result(compare_id) except FactCompareException as exception: return render_template('compare/error.html', error=exception.get_message()) if not result: return render_template('compare/wait.html', compare_id=compare_id) download_link = self._create_ida_download_if_existing(result, compare_id) uid_list = convert_compare_id_to_list(compare_id) plugin_views, plugins_without_view = self._get_compare_plugin_views(result) compare_view = self._get_compare_view(plugin_views) self._fill_in_empty_fields(result, compare_id) return render_template_string( compare_view, result=result, uid_list=uid_list, download_link=download_link, plugins_without_view=plugins_without_view )
def check_objects_exist(self, compare_id): uids = convert_compare_id_to_list(compare_id) for uid in uids: if not self.existence_quick_check(uid): raise FactCompareException( '{} not found in database'.format(uid))
def test_convert_compare_id_to_list(input_data, expected): assert convert_compare_id_to_list(input_data) == expected