From 0f27bf854560fa37a5fa9f9d9645fe4c321fe487 Mon Sep 17 00:00:00 2001
From: 1RandomDev <84292805+1RandomDev@users.noreply.github.com>
Date: Wed, 24 Aug 2022 18:10:49 +0200
Subject: Set correct file extension for thumbnails

---
 app/ytdl.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'app/ytdl.py')

diff --git a/app/ytdl.py b/app/ytdl.py
index 38e29b8..8382184 100644
--- a/app/ytdl.py
+++ b/app/ytdl.py
@@ -6,6 +6,7 @@ import time
 import asyncio
 import multiprocessing
 import logging
+import re
 from dl_formats import get_format, get_opts
 
 log = logging.getLogger('ytdl')
@@ -124,6 +125,10 @@ class Download:
             self.tmpfilename = status.get('tmpfilename')
             if 'filename' in status:
                 self.info.filename = os.path.relpath(status.get('filename'), self.download_dir)
+
+                # Set correct file extension for thumbnails
+                if(self.info.format == 'thumbnail'):
+                    self.info.filename = re.sub(r'\.webm$', '.jpg', self.info.filename)
             self.info.status = status['status']
             self.info.msg = status.get('msg')
             if 'downloaded_bytes' in status:
-- 
cgit