1 files changed, 31 insertions, 0 deletions
diff --git a/src/youload_playlist.py b/src/youload_playlist.py
new file mode 100644
index 0000000..4121d85
--- /dev/null
+++ b/src/youload_playlist.py
@@ -0,0 +1,31 @@
+from pytube import Playlist
+import os
+
+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"""
+
+    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)
+
+    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 prepare_for_download(self) -> None:
+        os.mkdir(self.folder_name)
+
+    def download_video(self, video_num: int) -> str:
+        video = self.yt_playlist.videos[video_num]
+        video_size = video.streams.get_highest_resolution().filesize
+        video.streams.get_highest_resolution().download(output_path=self.folder_name)
+
+        return f"Title: {video.title}, Size: {video_size // (1024 ** 2)} MB"
+