Mercurial > p > roundup > code
diff roundup/rate_limit.py @ 5739:e225f403cc35
Run pylint and clean up it's issues. Also fix comment.
| author | John Rouillard <rouilj@ieee.org> |
|---|---|
| date | Tue, 28 May 2019 18:10:30 -0400 |
| parents | 2f116ba7e7cf |
| children | 5d0873a4de4a |
line wrap: on
line diff
--- a/roundup/rate_limit.py Sun May 26 22:09:00 2019 -0400 +++ b/roundup/rate_limit.py Tue May 28 18:10:30 2019 -0400 @@ -6,7 +6,7 @@ from datetime import timedelta, datetime -class RateLimit: +class RateLimit: # pylint: disable=too-few-public-methods def __init__(self, count, period): self.count = count self.period = period @@ -17,8 +17,8 @@ class Gcra: - - memory = {} + def __init__(self): + self.memory = {} def get_tat(self, key): # This should return a previous tat for the key or the current time. @@ -47,7 +47,7 @@ self.memory[key] = datetime.strptime(tat,"%Y-%m-%dT%H:%M:%S.%f") def update(self, key, limit, testonly=False): - '''Determine if the item assocaited with the key should be + '''Determine if the item associated with the key should be rejected given the RateLimit limit. ''' now = datetime.utcnow() @@ -91,7 +91,7 @@ (limit.period - (tat - now)).total_seconds() \ /limit.inverse),ret['X-RateLimit-Limit']) - tat_epochsec = (tat - datetime(1970, 1, 1)).total_seconds() + # tat_in_epochsec = (tat - datetime(1970, 1, 1)).total_seconds() seconds_to_tat = (tat - now).total_seconds() ret['X-RateLimit-Reset'] = max(seconds_to_tat, 0) ret['X-RateLimit-Reset-date'] = "%s"%tat
