youtube-dl

commit e65e4c88748f2d245aa683116a2c6de907186751
parent 21f6330274c6f87c796bd9248ed82d2bc73de969
Author: Sergey M․ <dstftw@gmail.com>
Date:   Sat,  2 May 2015 23:06:01 +0600

[utils] Improve prepend_extension

Now `ext` is appended to filename if real extension != expected extension.

Diffstat:
M
youtube_dl/utils.py
 | 
7
+++++--

1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py
@@ -1349,9 +1349,12 @@ def parse_duration(s):
     return res
 
 
-def prepend_extension(filename, ext):
+def prepend_extension(filename, ext, expected_real_ext=None):
     name, real_ext = os.path.splitext(filename)
-    return '{0}.{1}{2}'.format(name, ext, real_ext)
+    return (
+        '{0}.{1}{2}'.format(name, ext, real_ext)
+        if not expected_real_ext or real_ext[1:] == expected_real_ext
+        else '{0}.{1}'.format(filename, ext))
 
 
 def check_executable(exe, args=[]):