5505-public/dima
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Implemented input argument to enable append information to exisintg attributes, which must take the values of either strings or lists.

Browse Source
This commit is contained in:
florez_j
2024-06-20 15:32:33 +02:00
parent 106795ae59
commit cedfe614e7
1 changed files with 33 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/metadata_review_lib.py
View File

@ -329,27 +329,48 @@ class MetadataHarvester:
"datasets": {} "datasets": {}
} }
def add_project_info(self, key_or_dict, value=None): def add_project_info(self, key_or_dict, value=None, append=False):
self._add_info("project", key_or_dict, value) self._add_info("project", key_or_dict, value, append)
def add_sample_info(self, key_or_dict, value=None): def add_sample_info(self, key_or_dict, value=None, append=False):
self._add_info("sample", key_or_dict, value) self._add_info("sample", key_or_dict, value, append)
def add_environment_info(self, key_or_dict, value=None): def add_environment_info(self, key_or_dict, value=None, append=False):
self._add_info("environment", key_or_dict, value) self._add_info("environment", key_or_dict, value, append)
def add_instrument_info(self, key_or_dict, value=None): def add_instrument_info(self, key_or_dict, value=None, append=False):
self._add_info("instruments", key_or_dict, value) self._add_info("instruments", key_or_dict, value, append)
def add_dataset_info(self, key_or_dict, value=None): def add_dataset_info(self, key_or_dict, value=None, append=False):
self._add_info("datasets", key_or_dict, value) self._add_info("datasets", key_or_dict, value, append)
def _add_info(self, category, key_or_dict, value): def _add_info(self, category, key_or_dict, value, append):
"""Internal helper method to add information to a category.""" """Internal helper method to add information to a category."""
if isinstance(key_or_dict, dict): if isinstance(key_or_dict, dict):
self.metadata[category].update(key_or_dict) self.metadata[category].update(key_or_dict)
else: else:
self.metadata[category][key_or_dict] = value if key_or_dict in self.metadata[category]:
if append:
current_value = self.metadata[category][key_or_dict]
if isinstance(current_value, list):
if not isinstance(value, list):
# Append the new value to the list
self.metadata[category][key_or_dict].append(value)
else:
self.metadata[category][key_or_dict] = current_value + value
elif isinstance(current_value, str):
# Append the new value as a comma-separated string
self.metadata[category][key_or_dict] = current_value + ',' + str(value)
else:
# Handle other types (for completeness, usually not required)
self.metadata[category][key_or_dict] = [current_value, value]
else:
self.metadata[category][key_or_dict] = value
else:
self.metadata[category][key_or_dict] = value
def get_metadata(self): def get_metadata(self):
return { return {