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

fix network errors

Browse source
This commit is contained in:
Tomáš Mládek 2020-07-15 03:28:03 +02:00
parent 33a33daa83
commit f19e04c962
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
sdbs_infra/dashboard/views.py
View file

@ -6,6 +6,7 @@ from urllib.parse import urlparse
import aiohttp import aiohttp
import psutil import psutil
from aiohttp import ClientConnectorError
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
from django.views.generic import TemplateView from django.views.generic import TemplateView
from humanize import naturalsize from humanize import naturalsize
@ -30,12 +31,12 @@ class IndexView(TemplateView):
session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=5, sock_connect=1)) session = aiohttp.ClientSession(timeout=aiohttp.ClientTimeout(total=5, sock_connect=1))
for link in links: for link in links:
index_text = None, None index_text = None
if not link.image: if not link.image:
try: try:
async with session.get(link.url) as response: async with session.get(link.url) as response:
index_status, index_text = response.status, await response.text() index_status, index_text = response.status, await response.text()
except asyncio.TimeoutError: except (asyncio.TimeoutError, ClientConnectorError):
pass pass
image = link.image.url if link.image else self.extract_favicon(link.url, index_text) image = link.image.url if link.image else self.extract_favicon(link.url, index_text)
@ -59,7 +60,7 @@ class IndexView(TemplateView):
try: try:
async with session.get(service.url) as response: async with session.get(service.url) as response:
index_status, index_text = response.status, await response.text() index_status, index_text = response.status, await response.text()
except asyncio.TimeoutError: except (asyncio.TimeoutError, ClientConnectorError):
pass pass
if service.port: if service.port: