1. class TokenCache:
  2. def __init__(self):
  3. self.cache = {} # Dictionary to store tokens and their associated data
  5. def store_token(self, token, data):
  6. """Stores a token and its associated data in the cache."""
  7. self.cache[token] = data
  9. def get_token(self, token):
  10. """Retrieves data associated with a token from the cache."""
  11. return self.cache.get(token)
  13. def remove_token(self, token):
  14. """Removes a token from the cache."""
  15. if token in self.cache:
  16. del self.cache[token]
  18. def clear_cache(self):
  19. """Clears all tokens from the cache."""
  20. self.cache = {}
  22. if __name__ == '__main__':
  23. # Example Usage
  24. cache = TokenCache()
  26. # Simulate obtaining a token
  27. token = "example_token"
  28. data = {"user_id": 123, "expiry": "2024-12-31"}
  30. # Store the token and data
  31. cache.store_token(token, data)
  32. print(f"Token stored: {token}")
  34. # Retrieve the token's data
  35. retrieved_data = cache.get_token(token)
  36. print(f"Retrieved data: {retrieved_data}")
  38. # Remove the token
  39. cache.remove_token(token)
  40. print(f"Token removed: {token}")
  42. # Attempt to retrieve the token again (should return None)
  43. retrieved_data = cache.get_token(token)
  44. print(f"Retrieved data after removal: {retrieved_data}")
  46. #Clear the cache
  47. cache.clear_cache()
  48. print("Cache cleared")

Add your comment