[EmbedThumbnail] Do not obey -k
This commit is contained in:
parent
94aa064497
commit
43d7f5a5d0
5 changed files with 29 additions and 21 deletions
|
@ -3307,6 +3307,17 @@ class YoutubeDL:
|
|||
''' Alias of sanitize_info for backward compatibility '''
|
||||
return YoutubeDL.sanitize_info(info_dict, actually_filter)
|
||||
|
||||
def _delete_downloaded_files(self, *files_to_delete, info={}, msg=None):
|
||||
for filename in set(filter(None, files_to_delete)):
|
||||
if msg:
|
||||
self.to_screen(msg % filename)
|
||||
try:
|
||||
os.remove(filename)
|
||||
except OSError:
|
||||
self.report_warning(f'Unable to delete file {filename}')
|
||||
if filename in info.get('__files_to_move', []): # NB: Delete even if None
|
||||
del info['__files_to_move'][filename]
|
||||
|
||||
@staticmethod
|
||||
def post_extract(info_dict):
|
||||
def actual_post_extract(info_dict):
|
||||
|
@ -3339,14 +3350,8 @@ class YoutubeDL:
|
|||
for f in files_to_delete:
|
||||
infodict['__files_to_move'].setdefault(f, '')
|
||||
else:
|
||||
for old_filename in set(files_to_delete):
|
||||
self.to_screen('Deleting original file %s (pass -k to keep)' % old_filename)
|
||||
try:
|
||||
os.remove(encodeFilename(old_filename))
|
||||
except OSError:
|
||||
self.report_warning('Unable to remove downloaded original file')
|
||||
if old_filename in infodict['__files_to_move']:
|
||||
del infodict['__files_to_move'][old_filename]
|
||||
self._delete_downloaded_files(
|
||||
*files_to_delete, info=infodict, msg='Deleting original file %s (pass -k to keep)')
|
||||
return infodict
|
||||
|
||||
def run_all_pps(self, key, info, *, additional_pps=None):
|
||||
|
|
|
@ -92,6 +92,12 @@ class PostProcessor(metaclass=PostProcessorMetaClass):
|
|||
if self._downloader:
|
||||
return self._downloader.write_debug(text, *args, **kwargs)
|
||||
|
||||
def _delete_downloaded_files(self, *files_to_delete, **kwargs):
|
||||
if not self._downloader:
|
||||
for filename in set(filter(None, files_to_delete)):
|
||||
os.remove(filename)
|
||||
return self._downloader._delete_downloaded_files(*files_to_delete, **kwargs)
|
||||
|
||||
def get_param(self, name, default=None, *args, **kwargs):
|
||||
if self._downloader:
|
||||
return self._downloader.params.get(name, default, *args, **kwargs)
|
||||
|
|
|
@ -220,11 +220,9 @@ class EmbedThumbnailPP(FFmpegPostProcessor):
|
|||
os.replace(temp_filename, filename)
|
||||
|
||||
self.try_utime(filename, mtime, mtime)
|
||||
|
||||
files_to_delete = [thumbnail_filename]
|
||||
if self._already_have_thumbnail:
|
||||
if original_thumbnail == thumbnail_filename:
|
||||
files_to_delete = []
|
||||
elif original_thumbnail != thumbnail_filename:
|
||||
files_to_delete.append(original_thumbnail)
|
||||
return files_to_delete, info
|
||||
converted = original_thumbnail != thumbnail_filename
|
||||
self._delete_downloaded_files(
|
||||
thumbnail_filename if converted or not self._already_have_thumbnail else None,
|
||||
original_thumbnail if converted and not self._already_have_thumbnail else None,
|
||||
info=info)
|
||||
return [], info
|
||||
|
|
|
@ -374,7 +374,7 @@ class FFmpegPostProcessor(PostProcessor):
|
|||
self.real_run_ffmpeg(
|
||||
[(concat_file, ['-hide_banner', '-nostdin', '-f', 'concat', '-safe', '0'])],
|
||||
[(out_file, out_flags)])
|
||||
os.remove(concat_file)
|
||||
self._delete_downloaded_files(concat_file)
|
||||
|
||||
@classmethod
|
||||
def _concat_spec(cls, in_files, concat_opts=None):
|
||||
|
@ -701,8 +701,7 @@ class FFmpegMetadataPP(FFmpegPostProcessor):
|
|||
self.run_ffmpeg_multiple_files(
|
||||
(filename, metadata_filename), temp_filename,
|
||||
itertools.chain(self._options(info['ext']), *options))
|
||||
for file in filter(None, files_to_delete):
|
||||
os.remove(file) # Don't obey --keep-files
|
||||
self._delete_downloaded_files(*files_to_delete)
|
||||
os.replace(temp_filename, filename)
|
||||
return [], info
|
||||
|
||||
|
@ -1049,7 +1048,7 @@ class FFmpegSplitChaptersPP(FFmpegPostProcessor):
|
|||
destination, opts = self._ffmpeg_args_for_chapter(idx + 1, chapter, info)
|
||||
self.real_run_ffmpeg([(in_file, opts)], [(destination, self.stream_copy_opts())])
|
||||
if in_file != info['filepath']:
|
||||
os.remove(in_file)
|
||||
self._delete_downloaded_files(in_file, msg=None)
|
||||
return [], info
|
||||
|
||||
|
||||
|
|
|
@ -314,7 +314,7 @@ class ModifyChaptersPP(FFmpegPostProcessor):
|
|||
self.to_screen(f'Removing chapters from {filename}')
|
||||
self.concat_files([in_file] * len(concat_opts), out_file, concat_opts)
|
||||
if in_file != filename:
|
||||
os.remove(in_file)
|
||||
self._delete_downloaded_files(in_file, msg=None)
|
||||
return out_file
|
||||
|
||||
@staticmethod
|
||||
|
|
Loading…
Reference in a new issue