voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Replace else-if with elif.

Browse source
This commit is contained in:
Ethan Dalool 2020-02-17 23:04:39 -08:00
parent 1d0f96a966
commit 9b8b0b6339
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
voussoirkit/ratelimiter.py
View file

@ -52,15 +52,16 @@ class Ratelimiter:
if self.balance >= cost: if self.balance >= cost:
self.balance -= cost self.balance -= cost
successful = True successful = True
elif self.mode == 'reject':
successful = False
else: else:
if self.mode == 'reject': deficit = cost - self.balance
successful = False time_needed = deficit / self.gain_rate
else: time.sleep(time_needed)
deficit = cost - self.balance self.balance = 0
time_needed = deficit / self.gain_rate successful = True
time.sleep(time_needed)
self.balance = 0
successful = True
self.last_operation = time.time() self.last_operation = time.time()
return successful return successful