summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRoman Haritonov <reclosedev@gmail.com>2014-12-16 22:28:49 +0400
committerRoman Haritonov <reclosedev@gmail.com>2014-12-16 22:28:49 +0400
commita54813bfab6f257eb4a4c951e895ce39983d6cc3 (patch)
tree6c7546188ec8587faad7c6afd0b9fcf584bda874
parenteae0d2c23084990493516960db55d49f880bb460 (diff)
parent1fa71cdb0eef027be845bcf0d9ac45a1efa5362e (diff)
downloadrequests-cache-a54813bfab6f257eb4a4c951e895ce39983d6cc3.tar.gz
Merge pull request #36 from femtotrader/patch-1
Update core.py to support expire_after parameter with datetime.timedelta
-rw-r--r--requests_cache/core.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/requests_cache/core.py b/requests_cache/core.py
index a5f227b..de49f22 100644
--- a/requests_cache/core.py
+++ b/requests_cache/core.py
@@ -64,7 +64,10 @@ class CachedSession(OriginalSession):
else:
self.cache = backend
+ if expire_after is not None and not isinstance(expire_after, timedelta):
+ expire_after = timedelta(seconds=expire_after)
self._cache_expire_after = expire_after
+
self._cache_allowable_codes = allowable_codes
self._cache_allowable_methods = allowable_methods
self._is_cache_disabled = False
@@ -92,7 +95,7 @@ class CachedSession(OriginalSession):
if self._cache_expire_after is not None:
difference = datetime.utcnow() - timestamp
- if difference > timedelta(seconds=self._cache_expire_after):
+ if difference > self._cache_expire_after:
self.cache.delete(cache_key)
return send_request_and_cache_response()
# dispatch hook here, because we've removed it before pickling