1. import json
  2. import os
  4. def load_cookies(filepath):
  5. """Loads cookies from a JSON file.
  7. Args:
  8. filepath (str): The path to the JSON file containing cookies.
  10. Returns:
  11. dict: A dictionary of cookies, or an empty dictionary if the file
  12. doesn't exist or is invalid.
  13. """
  14. try:
  15. with open(filepath, 'r') as f:
  16. cookies = json.load(f)
  17. return cookies
  18. except FileNotFoundError:
  19. print(f"Cookie file not found: {filepath}")
  20. return {}
  21. except json.JSONDecodeError:
  22. print(f"Invalid JSON format in cookie file: {filepath}")
  23. return {}
  25. def create_dummy_cookies(filepath):
  26. """Creates a dummy cookie file for testing.
  28. Args:
  29. filepath (str): The path to the file to create.
  30. """
  31. cookies = {
  32. "example_site": {"session_id": "1234567890", "user_id": "user123"},
  33. "another_site": {"auth_token": "abcdef123456"}
  34. }
  35. with open(filepath, 'w') as f:
  36. json.dump(cookies, f, indent=4)
  38. if __name__ == '__main__':
  39. cookie_file = "cookies.json" # Define the cookie file path
  41. # Create a dummy cookie file if it doesn't exist
  42. if not os.path.exists(cookie_file):
  43. create_dummy_cookies(cookie_file)
  45. cookies = load_cookies(cookie_file)
  46. print(cookies)

Add your comment