1. import json
  2. import datetime
  4. def attach_metadata(entry, metadata_source, override_values=None):
  5. """
  6. Attaches metadata to an entry, allowing for manual overrides.
  8. Args:
  9. entry (dict): The entry dictionary to which metadata will be attached.
  10. metadata_source (dict): A dictionary containing the metadata to be attached.
  11. override_values (dict, optional): A dictionary of metadata keys and their
  12. manual override values. Defaults to None.
  14. Returns:
  15. dict: The entry dictionary with attached metadata. Returns the original entry if metadata_source is empty.
  16. """
  18. if not metadata_source:
  19. return entry
  21. # Merge metadata, prioritizing overrides
  22. merged_metadata = metadata_source.copy()
  24. if override_values:
  25. merged_metadata.update(override_values) # Overwrite with provided values
  27. # Add timestamp
  28. merged_metadata["timestamp"] = datetime.datetime.now().isoformat()
  30. # Attach metadata to the entry
  31. entry.update(merged_metadata)
  33. return entry
  35. if __name__ == '__main__':
  36. # Example Usage
  37. entry = {"id": 123, "data": "some data"}
  38. metadata_source = {
  39. "source": "system",
  40. "user": "john.doe",
  41. "environment": "production"
  42. }
  43. override_values = {
  44. "user": "jane.doe", # Override user
  45. "environment": "staging" # Override environment
  46. }
  48. updated_entry = attach_metadata(entry, metadata_source, override_values)
  49. print(json.dumps(updated_entry, indent=4))
  51. # Example with no overrides
  52. entry2 = {"id": 456, "data": "another data"}
  53. metadata_source2 = {
  54. "source": "system",
  55. "user": "john.doe",
  56. "environment": "production"
  57. }
  58. updated_entry2 = attach_metadata(entry2, metadata_source2)
  59. print(json.dumps(updated_entry2, indent=4))
  61. #Example with empty metadata source
  62. entry3 = {"id": 789, "data": "yet another data"}
  63. metadata_source3 = {}
  64. updated_entry3 = attach_metadata(entry3, metadata_source3)
  65. print(json.dumps(updated_entry3, indent=4))

Add your comment