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

0.9.0

Browse Source
This commit is contained in:
Vinícius Moreira
2020-04-13 11:49:28 -03:00
parent 83aab1ff55
commit 01a60ea686
165 changed files with 12778 additions and 6171 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
bauh/api/http.py
View File

@@ -67,7 +67,7 @@ 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(self, url: str, session: bool = True) -> str:
def get_content_length_in_bytes(self, url: str, session: bool = True) -> int:
params = {'url': url, 'allow_redirects': True, 'stream': True}
if session:
res = self.session.get(**params)
@@ -77,8 +77,17 @@ class HttpClient:
if res.status_code == 200:
size = res.headers.get('Content-Length')
if size is not None:
return system.get_human_size_str(size)
if size:
try:
return int(size)
except:
pass
def get_content_length(self, url: str, session: bool = True) -> str:
size = self.get_content_length_in_bytes(url, session)
if size:
return system.get_human_size_str(size)
def exists(self, url: str, session: bool = True) -> bool:
params = {'url': url, 'allow_redirects': True, 'verify': False, 'timeout': 5}