Skip to content

Commit efcdf85

Browse files
committed
Update the rate-limiting decorator
1 parent 2df7df9 commit efcdf85

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

seleniumbase/common/decorators.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import inspect
22
import logging
33
import math
4+
import sys
45
import threading
56
import time
67
import warnings
@@ -60,11 +61,18 @@ def decorate(func):
6061
def rate_limited_function(*args, **kargs):
6162
try:
6263
rate_lock.acquire(True)
63-
elapsed = time.clock() - last_time_called[0]
64+
elapsed = None
65+
if sys.version_info[0] >= 3:
66+
elapsed = time.process_time() - last_time_called[0]
67+
else:
68+
elapsed = time.clock() - last_time_called[0]
6469
wait_time_remaining = min_interval - elapsed
6570
if wait_time_remaining > 0:
6671
time.sleep(wait_time_remaining)
67-
last_time_called[0] = time.clock()
72+
if sys.version_info[0] >= 3:
73+
last_time_called[0] = time.process_time()
74+
else:
75+
last_time_called[0] = time.clock()
6876
finally:
6977
rate_lock.release()
7078
return func(*args, **kargs)

0 commit comments

Comments
 (0)