瀏覽代碼

[utils] Refactor handle_youtubedl_headers for future extension

Yen Chi Hsuan 9 年之前
父節點
當前提交
992fc9d6e1
共有 1 個文件被更改,包括 5 次插入4 次删除
  1. 5 4
      youtube_dl/utils.py

+ 5 - 4
youtube_dl/utils.py

@@ -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):