1. import json
  2. import logging
  4. # Configure logging
  5. logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
  8. def load_user_data(filepath="user_data.json"):
  9. """Loads user data from a JSON file. Falls back to an empty dictionary if the file doesn't exist or is invalid."""
  10. try:
  11. with open(filepath, 'r') as f:
  12. user_data = json.load(f)
  13. logging.info(f"Loaded user data from {filepath}")
  14. return user_data
  15. except FileNotFoundError:
  16. logging.warning(f"User data file not found: {filepath}. Using empty data.")
  17. return {}
  18. except json.JSONDecodeError:
  19. logging.error(f"Error decoding JSON from {filepath}. Using empty data.")
  20. return {}
  21. except Exception as e:
  22. logging.exception(f"An unexpected error occurred while loading user data: {e}. Using empty data.")
  23. return {}
  26. def save_user_data(data, filepath="user_data.json"):
  27. """Saves user data to a JSON file."""
  28. try:
  29. with open(filepath, 'w') as f:
  30. json.dump(data, f, indent=4) # Use indent for readability
  31. logging.info(f"Saved user data to {filepath}")
  32. except Exception as e:
  33. logging.error(f"Error saving user data to {filepath}: {e}")
  36. def get_user(user_id, user_data):
  37. """Retrieves user data by ID. Returns None if not found."""
  38. if user_id in user_data:
  39. return user_data[user_id]
  40. else:
  41. logging.warning(f"User with ID {user_id} not found.")
  42. return None
  44. def set_user(user_id, user_data):
  45. """Sets user data for a given ID."""
  46. user_data[user_id] = user_data
  47. return user_data
  49. if __name__ == '__main__':
  50. # Example Usage
  51. user_data = load_user_data() # Load data from file
  53. if user_data:
  54. user_data = set_user("user123", {"name": "Alice", "email": "alice@example.com"})
  55. print(f"User data for user123: {user_data}")
  56. retrieved_user = get_user("user123", user_data)
  57. print(f"Retrieved user: {retrieved_user}")
  59. save_user_data(user_data) # Save updated data
  60. else:
  61. print("No user data available.")

Add your comment