sdbs/sdbs-infra
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add timeout to index_request

Browse source
This commit is contained in:
Tomáš Mládek 2020-06-14 19:59:44 +02:00
parent ab510fac1e
commit 979bbe97e1
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
sdbs_infra/dashboard/models.py
View file

@ -1,5 +1,6 @@
import urllib.request import urllib.request
from enum import Enum from enum import Enum
from socket import timeout
from urllib.error import URLError from urllib.error import URLError
from urllib.request import Request from urllib.request import Request
@ -28,8 +29,8 @@ class Service(OrderedModel):
def index_request(self): def index_request(self):
try: try:
request = Request(self.url, headers={'User-Agent': 'its just me humble status page'}) request = Request(self.url, headers={'User-Agent': 'its just me humble status page'})
return urllib.request.urlopen(request) return urllib.request.urlopen(request, timeout=1)
except URLError: except (URLError, timeout):
return None return None
def get_status(self): def get_status(self):