Numbers addedrelease1.1

1 files changed, 11 insertions, 5 deletions

diff --git a/src/model/youload_playlist.py b/src/model/youload_playlist.py
index f23533f..a021b83 100644
--- a/src/model/youload_playlist.py
+++ b/src/model/youload_playlist.py
@@ -31,20 +31,26 @@ class YouLoadPlayList:
         """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) -> str:
+    def download_video(self, video_num: int, use_prefix: bool) -> str:
         """Download video at 'video_num'"""
         video = self.yt_playlist.videos[video_num]
         video_size = video.streams.get_highest_resolution().filesize
 
+        # Create prefix.
+        filename_prefix = ""
+
+        if use_prefix:
+            filename_prefix = str(video_num + 1) + " "
+
         # Download this fucker.
         if self.download_type == "highest":
-            video.streams.get_highest_resolution().download(output_path=self.folder_name)
+            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)
+            video.streams.get_lowest_resolution().download(output_path=self.folder_name, filename_prefix=filename_prefix)
         elif self.download_type == "audio":
-            video.streams.get_audio_only().download(output_path=self.folder_name)
+            video.streams.get_audio_only().download(output_path=self.folder_name, filename_prefix=filename_prefix)
         else:
-            video.streams.get_by_resolution(self.download_type).download(output_path=self.folder_name)
+            video.streams.get_by_resolution(self.download_type).download(output_path=self.folder_name, filename_prefix=filename_prefix)
 
         return f"Title: {video.title}, Size: {video_size // (1024 ** 2)} MB"