From ef55effa58ee01089bffb441a65e932393d9facb Mon Sep 17 00:00:00 2001 From: nathansmith117 Date: Wed, 27 Mar 2024 16:11:09 -0600 Subject: Working on yt-dlp --- src/model/youload_playlist.py | 65 ------------------------------------------- 1 file changed, 65 deletions(-) delete mode 100644 src/model/youload_playlist.py (limited to 'src/model/youload_playlist.py') diff --git a/src/model/youload_playlist.py b/src/model/youload_playlist.py deleted file mode 100644 index 9189522..0000000 --- a/src/model/youload_playlist.py +++ /dev/null @@ -1,65 +0,0 @@ -from pytube import Playlist -import os -import util - -def make_alpha_numeric(string: str) -> str: - return "".join(char for char in string if char.isalnum()) - -class YouLoadPlayList: - """A class for download and handling youtube playlists""" - - DOWNLOAD_TYPES = ["default", "highest", "lowest", "mp3", "720p", "480p", "360p", "240p", "144p"] - DEFAULT_DOWNLOAD_TYPE = "default" - - def __init__(self, link: str): - self.yt_playlist = Playlist(link) - self.video_count = len(self.yt_playlist.videos) - self.folder_name = make_alpha_numeric(self.yt_playlist.title) - self.download_type = self.DEFAULT_DOWNLOAD_TYPE - - def generate_video_info(self, video_num: int) -> str: - """Returns information on video in playlist at 'video_num'""" - - video = self.yt_playlist.videos[video_num] - video_size = video.streams.get_highest_resolution().filesize // 1048576 - return f"Title: {video.title}, Size: {video_size} MB" - - def set_download_directory(self, directory: str) -> None: - """Sets where the playlist folder will be downloaded""" - self.folder_name = os.path.join(directory, make_alpha_numeric(self.yt_playlist.title)) - - def prepare_for_download(self) -> None: - """Gets the playlist ready for download. Creates the output folder and that stuff.""" - os.mkdir(self.folder_name) - - def download_video(self, video_num: int, use_prefix: bool) -> str: - """Download video at 'video_num'""" - video = self.yt_playlist.videos[video_num] - - # Create prefix. - filename_prefix = "" - - if use_prefix: - filename_prefix = str(video_num + 1) + " " - - # Download this fucker. - if self.download_type == "default": - video.streams.first().download(output_path=self.folder_name, filename_prefix=filename_prefix) - elif self.download_type == "highest": - video.streams.get_highest_resolution().download(output_path=self.folder_name, filename_prefix=filename_prefix) - elif self.download_type == "lowest": - video.streams.get_lowest_resolution().download(output_path=self.folder_name, filename_prefix=filename_prefix) - elif self.download_type == "mp3": - file_path = video.streams.first().download(output_path=self.folder_name, filename_prefix=filename_prefix) - - # To mp3. - file_path_mp3 = file_path[:-4] + ".mp3" - util.convert_mp4_to_mp3(file_path, file_path_mp3) - - # Remove mp4 file. - os.remove(file_path) - else: - video.streams.get_by_resolution(self.download_type).download(output_path=self.folder_name, filename_prefix=filename_prefix) - - return f"Title: {video.title}" - -- cgit v1.2.3