Better mp4 bar

Author: Arrowar

StreamingCommunity/Api/Site/animeunity/site.py

@@ -9,7 +9,7 @@
 
 # Internal utilities
 from StreamingCommunity.Util.headers import get_userAgent
-from StreamingCommunity.Util.http_client import create_client, create_client_curl
+from StreamingCommunity.Util.http_client import create_client_curl
 from StreamingCommunity.Util.table import TVShowManager
 from StreamingCommunity.TelegramHelp.telegram_bot import get_bot_instance
 
@@ -77,7 +77,7 @@ def title_search(query: str) -> int:
 
     # First call: /livesearch
     try:
-        response1 = create_client(headers=headers).post(f'{site_constant.FULL_URL}/livesearch', cookies=cookies, json={'title': query})
+        response1 = create_client_curl(headers=headers).post(f'{site_constant.FULL_URL}/livesearch', cookies=cookies, data={'title': query})
         response1.raise_for_status()
         process_results(response1.json().get('records', []), seen_titles, media_search_manager, choices)
 
@@ -98,7 +98,7 @@ def title_search(query: str) -> int:
             'dubbed': False,
             'season': False
         }
-        response2 = create_client(headers=headers).post(f'{site_constant.FULL_URL}/archivio/get-animes', cookies=cookies, json=json_data)
+        response2 = create_client_curl(headers=headers).post(f'{site_constant.FULL_URL}/archivio/get-animes', cookies=cookies, data=json_data)
         response2.raise_for_status()
         process_results(response2.json().get('records', []), seen_titles, media_search_manager, choices)
 

StreamingCommunity/Api/Site/animeunity/util/ScrapeSerie.py

@@ -5,7 +5,7 @@
 
 # Internal utilities
 from StreamingCommunity.Util.headers import get_headers
-from StreamingCommunity.Util.http_client import create_client
+from StreamingCommunity.Util.http_client import create_client_curl
 from StreamingCommunity.Api.Player.Helper.Vixcloud.util import EpisodeManager, Episode
 
 
@@ -52,7 +52,7 @@ def _fetch_all_episodes(self):
         """
         try:
             # Get initial episode count
-            response = create_client(headers=self.headers).get(f"{self.url}/info_api/{self.media_id}/")
+            response = create_client_curl(headers=self.headers).get(f"{self.url}/info_api/{self.media_id}/")
             response.raise_for_status()
             initial_count = response.json()["episodes_count"]
 
@@ -68,7 +68,7 @@ def _fetch_all_episodes(self):
                     "end_range": end_range
                 }
 
-                response = create_client(headers=self.headers).get(f"{self.url}/info_api/{self.media_id}/1", params=params)
+                response = create_client_curl(headers=self.headers).get(f"{self.url}/info_api/{self.media_id}/1", params=params)
                 response.raise_for_status()
 
                 chunk_episodes = response.json().get("episodes", [])

StreamingCommunity/Lib/Downloader/MP4/downloader.py

@@ -76,6 +76,7 @@ def MP4_downloader(url: str, path: str, referer: str = None, headers_: dict = No
     - Single Ctrl+C: Completes download gracefully
     - Triple Ctrl+C: Saves partial download and exits
     """
+    url = url.strip()
     if TELEGRAM_BOT:
         bot = get_bot_instance()
         console.log("####")
@@ -134,20 +135,23 @@ def MP4_downloader(url: str, path: str, referer: str = None, headers_: dict = No
 
                 # Create progress bar with percentage instead of n_fmt/total_fmt
                 console.print("[cyan]You can safely stop the download with [bold]Ctrl+c[bold] [cyan]")
+                
                 progress_bar = tqdm(
                     total=total,
                     ascii='░▒█',
                     bar_format=f"{Colors.YELLOW}MP4{Colors.CYAN} Downloading{Colors.WHITE}: "
-                               f"{Colors.RED}{{percentage:.1f}}% {Colors.MAGENTA}{{bar:40}} {Colors.WHITE}"
-                               f"{Colors.DARK_GRAY}[{Colors.YELLOW}{{elapsed}}{Colors.WHITE} < {Colors.CYAN}{{remaining}}{Colors.DARK_GRAY}] "
-                               f"{Colors.LIGHT_CYAN}{{rate_fmt}}",
+                               f"{Colors.MAGENTA}{{bar:40}} "
+                               f"{Colors.LIGHT_GREEN}{{n_fmt}}{Colors.WHITE}/{Colors.CYAN}{{total_fmt}}"
+                               f" {Colors.DARK_GRAY}[{Colors.YELLOW}{{elapsed}}{Colors.WHITE} < {Colors.CYAN}{{remaining}}{Colors.DARK_GRAY}]"
+                               f"{Colors.WHITE}{{postfix}} ",
                     unit='B',
                     unit_scale=True,
                     unit_divisor=1024,
                     mininterval=0.05,
                     file=sys.stdout
                 )
-
+                
+                start_time = time.time()
                 downloaded = 0
                 with open(temp_path, 'wb') as file, progress_bar as bar:
                     try:
@@ -160,6 +164,14 @@ def MP4_downloader(url: str, path: str, referer: str = None, headers_: dict = No
                                 size = file.write(chunk)
                                 downloaded += size
                                 bar.update(size)
+                                
+                                # Update postfix with speed and final size
+                                elapsed = time.time() - start_time
+                                if elapsed > 0:
+                                    speed = downloaded / elapsed
+                                    speed_str = internet_manager.format_transfer_speed(speed)
+                                    postfix_str = f"{Colors.LIGHT_MAGENTA}@ {Colors.LIGHT_CYAN}{speed_str}"
+                                    bar.set_postfix_str(postfix_str)
 
                     except KeyboardInterrupt:
                         if not interrupt_handler.force_quit: