Rule 34

youtube.py

@@ -1,7 +1,6 @@
 import os
 import requests
 from datetime import datetime, timedelta
-import urllib.request
 from pytube import YouTube
 from moviepy.editor import VideoFileClip
 from tqdm import tqdm
@@ -55,10 +54,8 @@ def youtube(link, resolusi_input):
     thumbnail_url = yt.thumbnail_url
 
     # Menentukan nama file thumbnail
-    thumbnail_file = f"{thumbnail_dir}/{judul_video}.jpg"
+    thumbnail_file = download_file(thumbnail_url, judul_video, thumbnail_dir)
 
-    # Mendownload thumbnail
-    urllib.request.urlretrieve(thumbnail_url, thumbnail_file)
     resolusi_tersedia = [stream.resolution for stream in yt.streams.filter(progressive=True)]
     video_info += f"Resolusi yang tersedia: {', '.join(resolusi_tersedia)}\n"
 
@@ -67,8 +64,8 @@ def youtube(link, resolusi_input):
 
     if stream is None:
         stream = yt.streams.filter(progressive=True, resolution='360p').first()
-        filename = download_youtube(stream.url, nama_channel, judul_video)
-        return filename, judul_video, video_info, thumbnail_file
+        video_file = download_youtube(stream.url, nama_channel, judul_video)
+        return video_file, judul_video, video_info, thumbnail_file
     else:
-        filename = download_youtube(stream.url, nama_channel, judul_video)
-        return filename, judul_video, video_info, thumbnail_file
+        video_file = download_youtube(stream.url, nama_channel, judul_video)
+        return video_file, judul_video, video_info, thumbnail_file