fixed tokenbucket a bit

This commit is contained in:
Jimmy Cao 2011-07-10 15:19:13 -05:00
parent 9ea736f6e5
commit c76f9215c4

View File

@ -51,11 +51,11 @@ class TokenBucket(object):
@property
def tokens(self):
now = time()
if self._tokens < self.capacity:
now = time.time()
delta = self.fill_rate * (now - self.timestamp)
self._tokens = min(self.capacity, self._tokens + delta)
self.timestamp = now
self.timestamp = now
return self._tokens