1. import os
  2. import json
  4. def is_allowed_to_diagnose(user_id):
  5. """
  6. Checks if a user is allowed to have their data used for diagnostics.
  7. This is a placeholder - replace with your actual authorization logic.
  8. """
  9. # Example: Allow only specific user IDs or based on a flag in a config file.
  10. allowed_ids = ["user123", "user456"]
  11. if user_id in allowed_ids:
  12. return True
  13. else:
  14. return False
  16. def safeguard_data_for_diagnostics(user_data, user_id, diagnostics_file="diagnostics.json"):
  17. """
  18. Safeguards user data for diagnostics, only if authorized.
  19. """
  20. if is_allowed_to_diagnose(user_id):
  21. try:
  22. with open(diagnostics_file, "r") as f:
  23. diagnostics = json.load(f)
  24. except FileNotFoundError:
  25. diagnostics = {}
  27. if user_id not in diagnostics:
  28. diagnostics[user_id] = {}
  30. # Copy user data to diagnostics, excluding sensitive fields.
  31. data_to_diagnose = {k: v for k, v in user_data.items() if not k.startswith("_")} #example: exclude fields starting with underscore
  33. diagnostics[user_id].update(data_to_diagnose)
  35. with open(diagnostics_file, "w") as f:
  36. json.dump(diagnostics, f, indent=4)
  37. print(f"Data for user {user_id} saved for diagnostics.")
  38. else:
  39. print(f"Diagnostics not enabled for user {user_id}.")
  42. if __name__ == '__main__':
  43. # Example Usage
  44. user_data = {
  45. "name": "Alice",
  46. "email": "alice@example.com",
  47. "age": 30,
  48. "_password": "secret", #sensitive data
  49. "location": "New York"
  50. }
  51. user_id = "user123" # Replace with actual user ID
  52. safeguard_data_for_diagnostics(user_data, user_id)
  54. user_id = "user789"
  55. safeguard_data_for_diagnostics(user_data, user_id)

Add your comment