From c5ca06d44b4dfcf72dbc841a56ce0b59ddf5e36a Mon Sep 17 00:00:00 2001 From: MaxwellGoblin Date: Sat, 6 May 2017 22:42:15 +0800 Subject: [PATCH] url_info should not ignore refer when called in url_save --- src/you_get/common.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/you_get/common.py b/src/you_get/common.py index 2bdc3655..037787bd 100755 --- a/src/you_get/common.py +++ b/src/you_get/common.py @@ -479,7 +479,13 @@ def url_locations(urls, faker = False, headers = {}): return locations def url_save(url, filepath, bar, refer = None, is_part = False, faker = False, headers = {}, timeout = None, **kwargs): - file_size = url_size(url, faker = faker, headers = headers) + new_headers = headers.copy() + if refer is not None: + for k in new_headers: + if k.lower() == 'referer': + del new_headers[k] + new_headers['referer'] = refer + file_size = url_size(url, faker = faker, headers = new_headers) if os.path.exists(filepath): if not force and file_size == os.path.getsize(filepath):