Save last request time after request itself.

This commit is contained in:
Bas Stottelaar
2014-09-21 13:52:06 +02:00
parent 2c72fb8e8e
commit f6346f3c41

View File

@@ -66,17 +66,19 @@ def request_response(url, method="get", auto_raise=True,
"req/sec.", request_limit - delta,
int(1.0 / request_limit))
# Sleep the remaining time and update time
# Sleep the remaining time
time.sleep(request_limit - delta)
# Set last access time
last_requests[lock] = time.time()
try:
# Request the URL
logger.debug("Requesting URL via %s method: %s", method.upper(), url)
response = request_method(url, **kwargs)
# Update rate limit last access time here, because a request will also
# take time.
if rate_limit:
last_requests[lock] = time.time()
# If status code != OK, then raise exception, except if the status code
# is white listed.
if whitelist_status_code and auto_raise: