From 32175cd211f02c49bce6e1402d0f8e2f57eeaa1c Mon Sep 17 00:00:00 2001
From: Christoph Alt <christoph.alt@fau.de>
Date: Mon, 28 Aug 2023 13:17:49 +0200
Subject: [PATCH] the apply_update function allows now also to insert values
---
cbutil/__init__.py | 1 +
cbutil/update_data.py | 8 ++++----
2 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/cbutil/__init__.py b/cbutil/__init__.py
index 6bd35ff..91c8e0a 100644
--- a/cbutil/__init__.py
+++ b/cbutil/__init__.py
@@ -5,3 +5,4 @@ from .util import read_file_line_wise, time_conversion, file_time_to_sec, get_fr
from .data_points import DataPoint, data_point_factory
from .get_job_info import get_url_from_env, get_job_datapoints
from .gitlab_api import get_git_infos_from_api
+from .update_data import get_updated_data
diff --git a/cbutil/update_data.py b/cbutil/update_data.py
index 3275191..753245e 100644
--- a/cbutil/update_data.py
+++ b/cbutil/update_data.py
@@ -37,10 +37,10 @@ def apply_updates(old_data, update_mapping):
"""
for data_point in old_data:
for update_key, mapping in update_mapping.items():
- if (current_value := data_point.get(update_key)):
- for old_value, new_value in mapping.items():
- if current_value == old_value:
- data_point[update_key] = new_value
+ current_value = data_point.get(update_key, None)
+ for old_value, new_value in mapping.items():
+ if current_value == old_value:
+ data_point[update_key] = new_value
yield data_point
--
GitLab