PublicMirrors/you-get
1
0
Fork 0
mirror of https://github.com/soimort/you-get.git synced 2025-02-03 00:33:58 +03:00
Code Issues Actions Packages Projects Releases 1 Wiki Activity

[common] fix urls_size (support faker and headers)

Browse Source
This commit is contained in:
Mort Yao 2015-12-13 04:55:50 +01:00
parent f1e7885140
commit 944a166c02
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/you_get/common.py
View File

@ -322,11 +322,8 @@ def url_size(url, faker = False, headers = {}):
size = response.headers['content-length']
return int(size) if size!=None else float('inf')
# TO BE DEPRECATED
# urls_size() does not have a faker
# also it takes too long time
def urls_size(urls):
return sum(map(url_size, urls))
def urls_size(urls, faker = False, headers = {}):
return sum([url_size(url, faker=faker, headers=headers) for url in urls])
def get_head(url, headers = {}):
if headers:
@ -665,7 +662,7 @@ def download_urls(urls, title, ext, total_size, output_dir='.', refer=None, merg
if not total_size:
try:
total_size = urls_size(urls)
total_size = urls_size(urls, faker=faker, headers=headers)
except:
import traceback
traceback.print_exc(file=sys.stdout)