def test_parameter_value_stays_hidden_in_task_run_info_object(self):
my_task = t_f.t("test_string")
task_run = my_task.dbnd_run().root_task_run
task_run_info = build_task_run_info(task_run)
input_task_param_info = [
task_param_info
for task_param_info in task_run_info.task_run_params
if task_param_info.parameter_name == "input_"
]
assert len(input_task_param_info) == 1
assert input_task_param_info[0].value == "***"
def dump_task_run_info(self, task_run):
info = {
"task_run":
TaskRunInfoSchema().dump(build_task_run_info(task_run)).data,
"task_def": (TaskDefinitionInfoSchema().dump(
task_to_task_def(task_run.run.context, task_run.task)).data),
"run":
RunInfoSchema().dump(task_run.run).data,
}
# old implementation, replace with "dbnd_tracking" request
meta_data_file = task_run.meta_files.get_meta_data_file()
with meta_data_file.open("w") as yaml_file:
yaml.dump(info, yaml_file, default_flow_style=False)
def set_af_operator_doc_md(task_run, airflow_op):
op = airflow_op
op.doc_md = ("### Databand Info\n"
"* **Tracker**: [{0}]({0})\n"
"* **TaskRun UID**: {1}\n"
"* **Run Name**: {2}\n"
"* **Run UID**: {3}\n".format(
task_run.task_tracker_url,
task_run.task_run_uid,
task_run.run.name,
task_run.run.run_uid,
))
op.dbnd_task_run_uid = task_run.task_run_uid
op.dbnd_run_uid = task_run.run.run_uid
op.dbnd_task_tracker_url = task_run.task_tracker_url
op.dbnd_task_run_info = _task_run_info_schema.dump(
build_task_run_info(task_run)).data