]> gitweb @ CieloNegro.org - youtube-dl.git/commitdiff
[YoutubeDL] Ensure dir existence for each requested format (closes #14116)
authorSergey M․ <dstftw@gmail.com>
Tue, 5 Sep 2017 16:31:34 +0000 (23:31 +0700)
committerSergey M․ <dstftw@gmail.com>
Tue, 5 Sep 2017 16:31:34 +0000 (23:31 +0700)
youtube_dl/YoutubeDL.py

index 5f4c93ea370d794e62f0e6eee4f1f81cecd1300b..4f208f1e1364f7ae2a4347798c265f991d48b66f 100755 (executable)
@@ -1710,12 +1710,17 @@ class YoutubeDL(object):
         if filename is None:
             return
 
-        try:
-            dn = os.path.dirname(sanitize_path(encodeFilename(filename)))
-            if dn and not os.path.exists(dn):
-                os.makedirs(dn)
-        except (OSError, IOError) as err:
-            self.report_error('unable to create directory ' + error_to_compat_str(err))
+        def ensure_dir_exists(path):
+            try:
+                dn = os.path.dirname(path)
+                if dn and not os.path.exists(dn):
+                    os.makedirs(dn)
+                return True
+            except (OSError, IOError) as err:
+                self.report_error('unable to create directory ' + error_to_compat_str(err))
+                return False
+
+        if not ensure_dir_exists(sanitize_path(encodeFilename(filename))):
             return
 
         if self.params.get('writedescription', False):
@@ -1853,8 +1858,11 @@ class YoutubeDL(object):
                         for f in requested_formats:
                             new_info = dict(info_dict)
                             new_info.update(f)
-                            fname = self.prepare_filename(new_info)
-                            fname = prepend_extension(fname, 'f%s' % f['format_id'], new_info['ext'])
+                            fname = prepend_extension(
+                                self.prepare_filename(new_info),
+                                'f%s' % f['format_id'], new_info['ext'])
+                            if not ensure_dir_exists(fname):
+                                return
                             downloaded.append(fname)
                             partial_success = dl(fname, new_info)
                             success = success and partial_success