From d93dcc219db9008f171f6bf98bb03e1d6d2a29ab Mon Sep 17 00:00:00 2001
From: Subhamoy Biswas <subhamoybiswas636@gmail.com>
Date: Tue, 21 Jan 2025 13:46:21 +0530
Subject: [PATCH] (refactor): added few loading indication messages

---
 pytubepp/main.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pytubepp/main.py b/pytubepp/main.py
index fcab7ca..331387f 100644
--- a/pytubepp/main.py
+++ b/pytubepp/main.py
@@ -391,8 +391,10 @@ def main():
 
         # Handle info display flags
         if args.show_info:
+            print('Loading...')
             downloader.show_video_info(args.url)
         if args.list_stream:
+            print('Loading...')
             downloader.show_all_streams(args.url)
         if args.raw_info:
             downloader.show_raw_info(args.url, args.json_prettify)
@@ -401,6 +403,7 @@ def main():
         
         # Handle download cases
         if hasattr(args, 'stream') and hasattr(args, 'caption'):
+            print('Loading...')
             if downloader.set_video_info(args.url):
                 if (args.caption not in downloader.captions.keys()) and (args.caption != 'none'):
                     print('\nInvalid caption code or caption not available! Please choose a different caption...!! (use -i to see available captions)')
@@ -420,6 +423,7 @@ def main():
                 else:
                     downloader.download_stream(args.url, args.stream, args.caption)
         elif hasattr(args, 'stream'):
+            print('Loading...')
             if downloader.set_video_info(args.url):
                 if downloader.default_caption == 'none':
                     downloader.download_stream(args.url, args.stream)
@@ -446,6 +450,7 @@ def main():
                     else:
                         print('Download cancelled! exiting...!!')
         elif hasattr(args, 'caption'):
+            print('Loading...')
             if downloader.set_video_info(args.url):
                 if (args.caption not in downloader.captions.keys()) and (args.caption != 'none'):
                     print('\nInvalid caption code or caption not available! Please choose a different caption...!! (use -i to see available captions)')
@@ -498,6 +503,7 @@ def main():
                     else:
                         print('Sorry, No downloadable video stream found....!!!')
         elif not any([args.show_info, args.raw_info, args.json_prettify, args.list_stream]):  # If no info flags are set
+            print('Loading...')
             if downloader.set_video_info(args.url):
                 if downloader.default_stream == 'max' and downloader.maxres:
                     if downloader.default_caption == 'none':