[utils] Refactor handle_youtubedl_headers for future extension
authorYen Chi Hsuan <yan12125@gmail.com>
Sun, 29 Nov 2015 04:58:29 +0000 (12:58 +0800)
committerYen Chi Hsuan <yan12125@gmail.com>
Sun, 29 Nov 2015 04:58:29 +0000 (12:58 +0800)
youtube_dl/utils.py

index c43e9e3a139a7222b3f0ffc53217a9bcdf3f0a02..d0606b4bcd3d4706912f753441608dff721d7699 100644 (file)
@@ -664,12 +664,13 @@ def _create_http_connection(ydl_handler, http_class, is_https, *args, **kwargs):
 
 
 def handle_youtubedl_headers(headers):
-    if 'Youtubedl-no-compression' in headers:
-        filtered_headers = dict((k, v) for k, v in headers.items() if k.lower() != 'accept-encoding')
+    filtered_headers = headers
+
+    if 'Youtubedl-no-compression' in filtered_headers:
+        filtered_headers = dict((k, v) for k, v in filtered_headers.items() if k.lower() != 'accept-encoding')
         del filtered_headers['Youtubedl-no-compression']
-        return filtered_headers
 
-    return headers
+    return filtered_headers
 
 
 class YoutubeDLHandler(compat_urllib_request.HTTPHandler):