def pause_dag(self): r_args = request.args report_name = r_args.get("report_name") r = Report(report_name) if r_args.get("is_paused") == "true": r.activate_dag() else: r.pause_dag() return "OK"
def to_report(name, v): """ Generates report objects from variable data """ r = Report(name) r.report_id = v["report_id"] r.report_title = v["report_title"] r.report_title_id = v["report_title_id"] r.description = v["description"] r.owner_name = v["owner_name"] r.owner_email = v["owner_email"] r.subscribers = v["subscribers"] r.tests = v["tests"] r.schedule_type = v["schedule_type"] if "daily" in r.schedule_type: r.schedule_time = v["schedule_time"] if "weekly" in r.schedule_type: r.schedule_time = v["schedule_time"] r.schedule_week_day = v["schedule_week_day"] r.schedule = v["schedule"] return r
def delete(self, report_name): r = Report(report_name) r.delete_report_variable(VariablesReportRepo.report_prefix) r.delete_dag() flash(f"Deleted report: {report_name}", "info") return redirect(url_for("ReportsView.list"))
def trigger(self, report_name): r = Report(report_name) r.trigger_dag() flash(f"Triggered report: {report_name}", "info") return redirect(url_for("ReportsView.list"))
def test_task_id_naming(self): r = Report("dummy name") assert r.dag_id == "rb_status_dummy_name"
def test_task_id_naming(self): r = Report("dummy name") self.assertEqual(r.dag_id, "rb_status_dummy_name")