Skip to content
This repository was archived by the owner on Dec 17, 2022. It is now read-only.

Commit b44261a

Browse files
committed
Prevent potential ZeroDivisionError in rate_per_second
1 parent 0219317 commit b44261a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bitlyshortener/shortener.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ def usage(self) -> float:
236236
usages = list(mapper(self._usage, tokens))
237237
time_used = time.monotonic() - start_time
238238
usage = sum(u["used"] for u in usages) / sum(u["limit"] for u in usages)
239-
rate_per_second = num_tokens / time_used
239+
rate_per_second = (num_tokens / time_used) if (time_used != 0) else float("inf")
240240
log.info(
241241
"%s retrieved usage of %s for %s tokens%s in %.1fs at a rate of %s/s.",
242242
strategy_desc,
@@ -267,7 +267,7 @@ def shorten_urls(self, long_urls: List[str]) -> List[str]:
267267
time_used = time.monotonic() - start_time
268268
num_short_urls = len(short_urls)
269269
assert num_long_urls == num_short_urls
270-
rate_per_second = num_short_urls / time_used
270+
rate_per_second = (num_short_urls / time_used) if (time_used != 0) else float("inf")
271271
log.info(
272272
"%s retrieved %s short URLs%s in %.1fs at a rate of %s/s. %s",
273273
strategy_desc,

0 commit comments

Comments
 (0)