add timeout to index_request

This commit is contained in:
Tomáš Mládek 2020-06-14 19:59:44 +02:00
parent ab510fac1e
commit 979bbe97e1

View file

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