fix leaky bucket use

datawhores · Aug 20, 2024 · c9f1164 · c9f1164
1 parent e012f34
commit c9f1164
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ofscraper/classes/sessionmanager/download.py b/ofscraper/classes/sessionmanager/download.py
@@ -1,6 +1,5 @@
 import contextlib
 import time
-from aiolimiter import AsyncLimiter
 import asyncio
 import ofscraper.classes.sessionmanager.ofsession as ofsessionmanager
 import ofscraper.classes.sessionmanager.sessionmanager as sessionManager
@@ -15,6 +14,7 @@
     SIGN,
     TOO_MANY,
 )
+from ofscraper.classes.sessionmanager.leaky import LeakyBucket
 import ofscraper.utils.settings as settings
 
 
@@ -51,7 +51,7 @@ def __init__(
         wait_min = wait_min or constants.getattr("OF_MIN_WAIT_API")
         wait_max = wait_max or constants.getattr("OF_MAX_WAIT_API")
         log = log or common_globals.log
-        self.leaky_bucket = AsyncLimiter(settings.get_download_limit(), 1)
+        self.leaky_bucket = LeakyBucket(settings.get_download_limit(), 1)
         super().__init__(
             sem_count=sem_count,
             retries=retries,