diff options
| author | Sergey M․ <dstftw@gmail.com> | 2015-05-02 23:06:01 +0600 | 
|---|---|---|
| committer | Sergey M․ <dstftw@gmail.com> | 2015-05-02 23:06:01 +0600 | 
| commit | e65e4c88748f2d245aa683116a2c6de907186751 (patch) | |
| tree | e811f6731bc4a505e8279afda4b67d31123f687e | |
| parent | 21f6330274c6f87c796bd9248ed82d2bc73de969 (diff) | |
[utils] Improve prepend_extension
Now `ext` is appended to filename if real extension != expected extension.
| -rw-r--r-- | youtube_dl/utils.py | 7 | 
1 files changed, 5 insertions, 2 deletions
diff --git a/youtube_dl/utils.py b/youtube_dl/utils.py index f07679c76..b3abfbc11 100644 --- 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=[]):  | 
