From: Yen Chi Hsuan <yan12125@gmail.com>
Date: Sat, 5 Mar 2016 10:18:28 +0000 (+0800)
Subject: [utils] Add codec2ext
X-Git-Url: https://git.cielonegro.org/gitweb.cgi?a=commitdiff_plain;h=19a17d462388b5d10196352e5f9d1491d608783a;p=youtube-dl.git

[utils] Add codec2ext
---

diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py
index 22a39a0ab..dc0bf5627 100644
--- a/youtube_dl/utils.py
+++ b/youtube_dl/utils.py
@@ -1893,6 +1893,19 @@ def mimetype2ext(mt):
     }.get(res, res)
 
 
+def codec2ext(codec):
+    codec_type = codec.split('.')[0]
+
+    # Leave the return value None for unknown values as codec_type
+    # is not a good fallback for file extensions
+    return {
+        'avc1': 'mp4',
+        'mp4a': 'm4a',
+        'vorbis': 'webm',
+        'vp9': 'webm',
+    }.get(codec_type)
+
+
 def urlhandle_detect_ext(url_handle):
     try:
         url_handle.headers