1. import json
  2. import csv
  4. def import_user_data(data_source, data_format="json", encoding="utf-8"):
  5. """
  6. Imports user data from a specified source (JSON or CSV).
  7. Handles potential errors and edge cases.
  9. Args:
  10. data_source (str): Path to the data file.
  11. data_format (str, optional): Format of the data file ("json" or "csv"). Defaults to "json".
  12. encoding (str, optional): Encoding of the file. Defaults to "utf-8".
  14. Returns:
  15. list: A list of user data dictionaries. Returns an empty list if an error occurs.
  16. """
  17. user_data = []
  19. try:
  20. if data_format.lower() == "json":
  21. with open(data_source, 'r', encoding=encoding) as f:
  22. try:
  23. data = json.load(f)
  24. if not isinstance(data, list):
  25. print("Error: JSON data must be a list of objects.")
  26. return []
  27. for item in data:
  28. if not isinstance(item, dict):
  29. print("Error: Each element in JSON list must be a dictionary.")
  30. return []
  31. user_data.append(item)
  32. except json.JSONDecodeError as e:
  33. print(f"Error decoding JSON: {e}")
  34. return []
  36. elif data_format.lower() == "csv":
  37. with open(data_source, 'r', encoding=encoding) as f:
  38. reader = csv.DictReader(f)
  39. for row in reader:
  40. user_data.append(row)
  41. else:
  42. print("Error: Unsupported data format. Use 'json' or 'csv'.")
  43. return []
  44. except FileNotFoundError:
  45. print(f"Error: File not found at {data_source}")
  46. return []
  47. except Exception as e:
  48. print(f"An unexpected error occurred: {e}")
  49. return []
  51. return user_data

Add your comment