1. import datetime
  2. import time
  3. import random
  5. class RateLimiter:
  6. def __init__(self, rate=5, period=60):
  7. self.rate = rate # requests per period
  8. self.period = period # period in seconds
  9. self.tokens = rate
  10. self.last_reset = time.time()
  12. def is_allowed(self):
  13. now = time.time()
  14. if now - self.last_reset > self.period:
  15. self.tokens = self.rate
  16. self.last_reset = now
  17. if self.tokens > 0:
  18. self.tokens -= 1
  19. return True
  20. else:
  21. return False
  23. def wait(self):
  24. now = time.time()
  25. if now - self.last_reset < self.period:
  26. time.sleep(self.period - (now - self.last_reset))
  27. self.last_reset = now
  30. def encode_date(date_value, rate_limiter):
  31. """Encodes a date value with rate limiting."""
  32. if rate_limiter.is_allowed():
  33. # Simulate some processing
  34. time.sleep(random.uniform(0.01, 0.1))
  35. encoded_date = datetime.datetime.now().isoformat() # Example encoding
  36. rate_limiter.tokens -=1
  37. return encoded_date
  38. else:
  39. rate_limiter.wait() # Wait before retrying
  40. return encode_date(date_value, rate_limiter) # Retry the encoding
  43. if __name__ == '__main__':
  44. rate_limiter = RateLimiter(rate=2, period=10) # Allow 2 requests per 10 seconds
  45. date_values = [datetime.datetime.now() for _ in range(10)] #Example date values
  47. for date_value in date_values:
  48. encoded = encode_date(date_value, rate_limiter)
  49. print(f"Encoded date: {encoded}")

Add your comment