Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

allow Request objects to specify their timeout

Browse source
This commit is contained in:
jesopo 2019-12-11 11:18:19 +00:00
parent 6f8338b7b5
commit 5a2905de01
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/utils/http.py
View file

@ -76,6 +76,8 @@ class Request(object):
proxy: typing.Optional[str] = None proxy: typing.Optional[str] = None
useragent: typing.Optional[str] = None useragent: typing.Optional[str] = None
timeout: int=5
def validate(self): def validate(self):
self.id = self.id or str(uuid.uuid4()) self.id = self.id or str(uuid.uuid4())
self.set_url(self.url) self.set_url(self.url)
@ -186,7 +188,7 @@ def _request(request_obj: Request) -> Response:
return our_response return our_response
try: try:
response = utils.deadline_process(_wrap, seconds=5) response = utils.deadline_process(_wrap, seconds=request_obj.timeout)
except utils.DeadlineExceededException: except utils.DeadlineExceededException:
raise HTTPTimeoutException() raise HTTPTimeoutException()