Redis Leaky Bucket Rate Limiting Mock in Python

import time
from collections import deque


class LeakyBucket:
    def __init__(self, capacity, leak_rate):
        self.capacity = capacity
        self.leak_rate = leak_rate
        self.water = 0.0
        self.timestamp = time.time()
        self.history = deque()

    def allow(self):
        current = time.time()
        elapsed = current - self.timestamp
        self.water = max(0, self.water - elapsed * self.leak_rate)
        self.timestamp = current

        if self.water + 1 <= self.capacity:
            self.water += 1
            self.history.append(current)
            return True
        return False

    def water_level(self):
        current = time.time()
        elapsed = current - self.timestamp
        return max(0, self.water - elapsed * self.leak_rate)


if __name__ == "__main__":
    bucket = LeakyBucket(capacity=3, leak_rate=1)
    for _ in range(5):
        print(bucket.allow())
    time.sleep(2)
    print("Water level after 2s:", bucket.water_level())
    print(bucket.allow())