commit f19b0c68e2f1022187e8d3406f2e33fe3b42b77b Author: juga0 <juga@riseup.net> Date: Thu Feb 21 13:51:04 2019 +0000 requests: Add class to override Session methods Override get and head methods to always have a timeout. --- sbws/util/requests.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/sbws/util/requests.py b/sbws/util/requests.py index 0b1ece2..e0a6864 100644 --- a/sbws/util/requests.py +++ b/sbws/util/requests.py @@ -1,7 +1,20 @@ import requests from sbws import settings -import sbws.util.stem as stem_utils +from sbws.util import stem as stem_utils + + +class TimedSession(requests.Session): + """Requests Session that sends timeout in the head and get methods. + """ + + def get(self, url, **kwargs): + return super().get(url, timeout=getattr(self, "_timeout", None), + **kwargs) + + def head(self, url, **kwargs): + return super().head(url, timeout=getattr(self, "_timeout", None), + **kwargs) def make_session(controller, timeout):