def send_mail(request, compare_key): def after_check(): result_data = fileManager.load_compare_result_file(compare_key) list_data = result_data[key.COMPARE_RESULT_LIST_DATA] list_data[key.KEY_BASE_ENV] = result_data[key.KEY_BASE_ENV] list_data[key.KEY_COMPARE_ENV] = result_data[key.KEY_COMPARE_ENV] html_content = render_to_string('compare_info_mail_content.html', list_data) mail_sender = MailSender(config.SEND_COMPARE_RESULT_MAIL) mail_sender.compose_msg(None, None, html_content) file_path = fileManager.get_compare_result_full_file_name( "_html", compare_key) file_name = "test_report.html" application = "text" mail_sender.add_attachment(file_path, file_name, application) mail_sender.send() mail_sender.quit() response = HttpResponse() return response return action_error_check(after_check)
def get_ruleset(request): def after_check(): request_json = get_post_request_json(request) ruleset = log.get_ruleset(request_json) result = {KEY_RULESET_DATA: ruleset} return render(request, "ruleset_content_view.html", result) return action_error_check(after_check)
def get_module_list(request): def after_check(): modules = Module.objects.all() data = ModulesBuilder(modules).get_data() result = ResponseBuilder(data=data).get_data() return JsonResponse(result.get_data()) return action_error_check(after_check)
def get_rulesets_report_job(request, scheduler_id): def after_check(): scheduler_info = ReportSchedulerInfo.objects.get(id=scheduler_id) scheduler_data = RulesetReportSchedulerBuilder( request.user, scheduler_info).get_data() result = ResponseBuilder(data=scheduler_data).get_data() response = JsonResponse(data=result) return response return action_error_check(after_check)
def get_rulesets_sync_jobs(request): def after_check(): schedulers = sync_scheduler.get_schedulers() data_list = list() for scheduler in schedulers: data_builder = RulesetSyncSchedulerBuilder(request.user, scheduler) data_list.append(data_builder.get_data()) result = ResponseBuilder(data=data_list).get_data() return JsonResponse(data=result) return action_error_check(after_check)
def download_rulesets(request): def after_check(): request_json = get_post_request_json(request) zip_file_path = services.download_rulesets(request_json) if os.path.exists(zip_file_path): with open(zip_file_path, 'rb') as fh: response = HttpResponse(fh.read(), content_type="application/zip") return response raise Http404 return action_error_check(after_check)
def download_compare_report(request, compare_key): def after_check(): file_path = fileManager.get_compare_result_full_file_name( "_html", compare_key) data = fileManager.load_compare_result_file(compare_key) file_name = data[key.COMPARE_RESULT_LIST_DATA][ key.COMPARE_RESULT_DATE_TIME] + "_report" if os.path.exists(file_path): with open(file_path, 'rb') as fh: response = HttpResponse(fh.read(), content_type="text/html") response[ 'Content-Disposition'] = 'attachment; filename="' + file_name + '.html"' return response raise Http404 return action_error_check(after_check)
def update_ruleset(request): def after_check(): sync.update_ruleset_test() return action_error_check(after_check)