github/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-14 20:38:11 -05:00
Code Issues Projects Releases Packages Wiki Activity

[postprocessor/embedthumbnail] Close file before possible renaming

Browse source
This commit is contained in:
Alex Merkel 2020-06-21 12:06:01 +02:00
parent e987deb504
commit ac0ad4f91d
No known key found for this signature in database
GPG key ID: B8F579CC2A6662A2
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
youtube_dl/postprocessor/embedthumbnail.py
View file

@ -44,12 +44,12 @@ def run(self, info):
#Check for mislabeled webp file
with open(encodeFilename(thumbnail_filename), "rb") as f:
b = f.read(16)
if b'\x57\x45\x42\x50' in b: #Binary for WEBP
[thumbnail_filename_path, thumbnail_filename_extension] = os.path.splitext(thumbnail_filename)
if not thumbnail_filename_extension == ".webp":
webp_thumbnail_filename = thumbnail_filename_path + ".webp"
os.rename(thumbnail_filename, webp_thumbnail_filename)
thumbnail_filename = webp_thumbnail_filename
if b'\x57\x45\x42\x50' in b: #Binary for WEBP
[thumbnail_filename_path, thumbnail_filename_extension] = os.path.splitext(thumbnail_filename)
if not thumbnail_filename_extension == ".webp":
webp_thumbnail_filename = thumbnail_filename_path + ".webp"
os.rename(thumbnail_filename, webp_thumbnail_filename)
thumbnail_filename = webp_thumbnail_filename
#If not a jpg or png thumbnail, convert it to jpg using ffmpeg
if not os.path.splitext(thumbnail_filename)[1].lower() in ['.jpg', '.png']: