1. import json
  2. import pickle
  3. import datetime
  5. class Cookie:
  6. def __init__(self, name, value, expiry=None, domain=None, path=None, secure=False, httponly=False):
  7. self.name = name
  8. self.value = value
  9. self.expiry = expiry # datetime object
  10. self.domain = domain
  11. self.path = path
  12. self.secure = secure
  13. self.httponly = httponly
  15. def __repr__(self):
  16. return f"Cookie(name='{self.name}', value='{self.value}', expiry={self.expiry}, domain={self.domain}, path={self.path}, secure={self.secure}, httponly={self.httponly})"
  19. def serialize_cookies_json(cookies):
  20. """Serializes a list of Cookie objects to a JSON string."""
  21. cookie_data = []
  22. for cookie in cookies:
  23. cookie_data.append({
  24. "name": cookie.name,
  25. "value": cookie.value,
  26. "expiry": cookie.expiry.isoformat() if cookie.expiry else None, # Convert datetime to ISO format
  27. "domain": cookie.domain,
  28. "path": cookie.path,
  29. "secure": cookie.secure,
  30. "httponly": cookie.httponly
  31. })
  32. return json.dumps(cookie_data, indent=4)
  34. def deserialize_cookies_json(json_string):
  35. """Deserializes a JSON string of cookies back into a list of Cookie objects."""
  36. cookie_data = json.loads(json_string)
  37. cookies = []
  38. for data in cookie_data:
  39. expiry_str = data.get("expiry")
  40. expiry = datetime.datetime.fromisoformat(expiry_str) if expiry_str else None
  41. cookies.append(Cookie(
  42. name=data["name"],
  43. value=data["value"],
  44. expiry=expiry,
  45. domain=data.get("domain"),
  46. path=data.get("path"),
  47. secure=data["secure"],
  48. httponly=data["httponly"]
  49. ))
  50. return cookies
  52. def serialize_cookies_pickle(cookies, filename):
  53. """Serializes a list of Cookie objects to a pickle file."""
  54. with open(filename, 'wb') as f:
  55. pickle.dump(cookies, f)
  57. def deserialize_cookies_pickle(filename):
  58. """Deserializes a list of Cookie objects from a pickle file."""
  59. with open(filename, 'rb') as f:
  60. return pickle.load(f)
  63. if __name__ == '__main__':
  64. # Example Usage
  65. cookies = [
  66. Cookie("sessionid", "1234567890", datetime.datetime(2024, 12, 31, 23, 59, 59)),
  67. Cookie("user_id", "9876543210", domain="example.com", secure=True)
  68. ]
  70. # Serialize to JSON
  71. json_string = serialize_cookies_json(cookies)
  72. print("JSON Serialization:\n", json_string)
  74. # Deserialize from JSON
  75. deserialized_cookies = deserialize_cookies_json(json_string)
  76. print("\nDeserialized Cookies:\n", deserialized_cookies)
  78. # Serialize to Pickle
  79. serialize_cookies_pickle(cookies, "cookies.pickle")
  81. # Deserialize from Pickle
  82. loaded_cookies = deserialize_cookies_pickle("cookies.pickle")
  83. print("\nDeserialized from Pickle:\n", loaded_cookies)

Add your comment