spalencsar/bearhub
1
0
Fork 0
mirror of https://github.com/spalencsar/bearhub.git synced 2026-07-07 01:14:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

[appimage] fix -> crashing when trying to upgrade and the internet connection is off

Browse Source
This commit is contained in:
Vinicius Moreira
2021-02-12 14:46:21 -03:00
parent 9af8dd4f1f
commit 8c0c4d4546
3 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
bauh/__init__.py
View File

@@ -1,4 +1,4 @@
__version__ = '0.9.14'
__version__ = '0.9.15'
__app_name__ = 'bauh'
import os

18
bauh/api/http.py
View File

@@ -68,12 +68,17 @@ class HttpClient:
res = self.get(url, params=params, headers=headers, allow_redirects=allow_redirects, session=session)
return yaml.safe_load(res.text) if res else None
def get_content_length_in_bytes(self, url: str, session: bool = True) -> int:
def get_content_length_in_bytes(self, url: str, session: bool = True) -> Optional[int]:
params = {'url': url, 'allow_redirects': True, 'stream': True}
if session:
res = self.session.get(**params)
else:
res = requests.get(**params)
try:
if session:
res = self.session.get(**params)
else:
res = requests.get(**params)
except requests.exceptions.ConnectionError:
self.logger.info("Internet seems to be off. Could not reach '{}'".format(url))
return
if res.status_code == 200:
size = res.headers.get('Content-Length')
@@ -84,7 +89,7 @@ class HttpClient:
except:
pass
def get_content_length(self, url: str, session: bool = True) -> str:
def get_content_length(self, url: str, session: bool = True) -> Optional[str]:
size = self.get_content_length_in_bytes(url, session)
if size:
@@ -98,4 +103,3 @@ class HttpClient:
res = self.session.get(**params)
return res.status_code in (200, 403)
return False