StreamingCommunity 1.7.6__py3-none-any.whl → 1.8.0__py3-none-any.whl

StreamingCommunity/Src/Lib/Downloader/MP4/downloader.py

@@ -1,156 +1,156 @@
-# 09.06.24
-
-import os
-import sys
-import ssl
-import certifi
-import logging
-
-
-# External libraries
-import httpx
-from tqdm import tqdm
-
-
-# Internal utilities
-from StreamingCommunity.Src.Util.headers import get_headers
-from StreamingCommunity.Src.Util.color import Colors
-from StreamingCommunity.Src.Util.console import console, Panel
-from StreamingCommunity.Src.Util._jsonConfig import config_manager
-from StreamingCommunity.Src.Util.os import internet_manager
-
-
-# Logic class
-from ...FFmpeg import print_duration_table
-
-
-# Suppress SSL warnings
-import urllib3
-urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
-
-
-# Config
-GET_ONLY_LINK = config_manager.get_bool('M3U8_PARSER', 'get_only_link')
-TQDM_USE_LARGE_BAR = config_manager.get_int('M3U8_DOWNLOAD', 'tqdm_use_large_bar')
-REQUEST_VERIFY = config_manager.get_float('REQUESTS', 'verify_ssl')
-REQUEST_TIMEOUT = config_manager.get_float('REQUESTS', 'timeout')
-
-
-
-def MP4_downloader(url: str, path: str, referer: str = None, headers_: dict = None):
-    """
-    Downloads an MP4 video from a given URL with robust error handling and SSL bypass.
-
-    Parameters:
-        - url (str): The URL of the MP4 video to download.
-        - path (str): The local path where the downloaded MP4 file will be saved.
-        - referer (str, optional): The referer header value.
-        - headers_ (dict, optional): Custom headers for the request.
-    """
-    # Early return for link-only mode
-    if GET_ONLY_LINK:
-        return {'path': path, 'url': url}
-
-    # Validate URL
-    if not (url.lower().startswith('http://') or url.lower().startswith('https://')):
-        logging.error(f"Invalid URL: {url}")
-        console.print(f"[bold red]Invalid URL: {url}[/bold red]")
-        return None
-
-    # Prepare headers
-    try:
-        headers = {}
-        if referer:
-            headers['Referer'] = referer
-        
-        # Use custom headers if provided, otherwise use default user agent
-        if headers_:
-            headers.update(headers_)
-        else:
-            headers['User-Agent'] = get_headers()
-
-    except Exception as header_err:
-        logging.error(f"Error preparing headers: {header_err}")
-        console.print(f"[bold red]Error preparing headers: {header_err}[/bold red]")
-        return None
-
-    try:
-        # Create a custom transport that bypasses SSL verification
-        transport = httpx.HTTPTransport(
-            verify=False,  # Disable SSL certificate verification
-            http2=True     # Optional: enable HTTP/2 support
-        )
-        
-        # Download with streaming and progress tracking
-        with httpx.Client(transport=transport, timeout=httpx.Timeout(60.0)) as client:
-            with client.stream("GET", url, headers=headers, timeout=REQUEST_TIMEOUT) as response:
-                response.raise_for_status()
-                
-                # Get total file size
-                total = int(response.headers.get('content-length', 0))
-                
-                # Handle empty streams
-                if total == 0:
-                    console.print("[bold red]No video stream found.[/bold red]")
-                    return None
-
-                # Create progress bar
-                progress_bar = tqdm(
-                    total=total,
-                    ascii='░▒█',
-                    bar_format=f"{Colors.YELLOW}[MP4] {Colors.WHITE}({Colors.CYAN}video{Colors.WHITE}): "
-                               f"{Colors.RED}{{percentage:.2f}}% {Colors.MAGENTA}{{bar}} {Colors.WHITE}[ "
-                               f"{Colors.YELLOW}{{n_fmt}}{Colors.WHITE} / {Colors.RED}{{total_fmt}} {Colors.WHITE}] "
-                               f"{Colors.YELLOW}{{elapsed}} {Colors.WHITE}< {Colors.CYAN}{{remaining}} {Colors.WHITE}| "
-                               f"{Colors.YELLOW}{{rate_fmt}}{{postfix}} {Colors.WHITE}]",
-                    unit='iB',
-                    unit_scale=True,
-                    desc='Downloading',
-                    mininterval=0.05
-                )
-
-                # Ensure directory exists
-                os.makedirs(os.path.dirname(path), exist_ok=True)
-
-                # Download file
-                with open(path, 'wb') as file, progress_bar as bar:
-                    downloaded = 0
-                    for chunk in response.iter_bytes(chunk_size=1024):
-                        if chunk:
-                            size = file.write(chunk)
-                            downloaded += size
-                            bar.update(size)
-
-                            # Optional: Add a check to stop download if needed
-                            # if downloaded > MAX_DOWNLOAD_SIZE:
-                            #     break
-
-        # Post-download processing
-        if os.path.exists(path) and os.path.getsize(path) > 0:
-            console.print(Panel(
-                f"[bold green]Download completed![/bold green]\n"
-                f"[cyan]File size: [bold red]{internet_manager.format_file_size(os.path.getsize(path))}[/bold red]\n"
-                f"[cyan]Duration: [bold]{print_duration_table(path, description=False, return_string=True)}[/bold]", 
-                title=f"{os.path.basename(path.replace('.mp4', ''))}", 
-                border_style="green"
-            ))
-            return path
-        
-        else:
-            console.print("[bold red]Download failed or file is empty.[/bold red]")
-            return None
-
-    except httpx.HTTPStatusError as http_err:
-        logging.error(f"HTTP error occurred: {http_err}")
-        console.print(f"[bold red]HTTP Error: {http_err}[/bold red]")
-        return None
-    
-    except httpx.RequestError as req_err:
-        logging.error(f"Request error: {req_err}")
-        console.print(f"[bold red]Request Error: {req_err}[/bold red]")
-        return None
-    
-    except Exception as e:
-        logging.error(f"Unexpected error during download: {e}")
-        console.print(f"[bold red]Unexpected Error: {e}[/bold red]")
-        return None
+# 09.06.24
+
+import os
+import sys
+import ssl
+import certifi
+import logging
+
+
+# External libraries
+import httpx
+from tqdm import tqdm
+
+
+# Internal utilities
+from StreamingCommunity.Src.Util.headers import get_headers
+from StreamingCommunity.Src.Util.color import Colors
+from StreamingCommunity.Src.Util.console import console, Panel
+from StreamingCommunity.Src.Util._jsonConfig import config_manager
+from StreamingCommunity.Src.Util.os import internet_manager
+
+
+# Logic class
+from ...FFmpeg import print_duration_table
+
+
+# Suppress SSL warnings
+import urllib3
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+
+# Config
+GET_ONLY_LINK = config_manager.get_bool('M3U8_PARSER', 'get_only_link')
+TQDM_USE_LARGE_BAR = config_manager.get_int('M3U8_DOWNLOAD', 'tqdm_use_large_bar')
+REQUEST_VERIFY = config_manager.get_float('REQUESTS', 'verify_ssl')
+REQUEST_TIMEOUT = config_manager.get_float('REQUESTS', 'timeout')
+
+
+
+def MP4_downloader(url: str, path: str, referer: str = None, headers_: dict = None):
+    """
+    Downloads an MP4 video from a given URL with robust error handling and SSL bypass.
+
+    Parameters:
+        - url (str): The URL of the MP4 video to download.
+        - path (str): The local path where the downloaded MP4 file will be saved.
+        - referer (str, optional): The referer header value.
+        - headers_ (dict, optional): Custom headers for the request.
+    """
+    # Early return for link-only mode
+    if GET_ONLY_LINK:
+        return {'path': path, 'url': url}
+
+    # Validate URL
+    if not (url.lower().startswith('http://') or url.lower().startswith('https://')):
+        logging.error(f"Invalid URL: {url}")
+        console.print(f"[bold red]Invalid URL: {url}[/bold red]")
+        return None
+
+    # Prepare headers
+    try:
+        headers = {}
+        if referer:
+            headers['Referer'] = referer
+        
+        # Use custom headers if provided, otherwise use default user agent
+        if headers_:
+            headers.update(headers_)
+        else:
+            headers['User-Agent'] = get_headers()
+
+    except Exception as header_err:
+        logging.error(f"Error preparing headers: {header_err}")
+        console.print(f"[bold red]Error preparing headers: {header_err}[/bold red]")
+        return None
+
+    try:
+        # Create a custom transport that bypasses SSL verification
+        transport = httpx.HTTPTransport(
+            verify=False,  # Disable SSL certificate verification
+            http2=True     # Optional: enable HTTP/2 support
+        )
+        
+        # Download with streaming and progress tracking
+        with httpx.Client(transport=transport, timeout=httpx.Timeout(60.0)) as client:
+            with client.stream("GET", url, headers=headers, timeout=REQUEST_TIMEOUT) as response:
+                response.raise_for_status()
+                
+                # Get total file size
+                total = int(response.headers.get('content-length', 0))
+                
+                # Handle empty streams
+                if total == 0:
+                    console.print("[bold red]No video stream found.[/bold red]")
+                    return None
+
+                # Create progress bar
+                progress_bar = tqdm(
+                    total=total,
+                    ascii='░▒█',
+                    bar_format=f"{Colors.YELLOW}[MP4] {Colors.WHITE}({Colors.CYAN}video{Colors.WHITE}): "
+                               f"{Colors.RED}{{percentage:.2f}}% {Colors.MAGENTA}{{bar}} {Colors.WHITE}[ "
+                               f"{Colors.YELLOW}{{n_fmt}}{Colors.WHITE} / {Colors.RED}{{total_fmt}} {Colors.WHITE}] "
+                               f"{Colors.YELLOW}{{elapsed}} {Colors.WHITE}< {Colors.CYAN}{{remaining}} {Colors.WHITE}| "
+                               f"{Colors.YELLOW}{{rate_fmt}}{{postfix}} {Colors.WHITE}]",
+                    unit='iB',
+                    unit_scale=True,
+                    desc='Downloading',
+                    mininterval=0.05
+                )
+
+                # Ensure directory exists
+                os.makedirs(os.path.dirname(path), exist_ok=True)
+
+                # Download file
+                with open(path, 'wb') as file, progress_bar as bar:
+                    downloaded = 0
+                    for chunk in response.iter_bytes(chunk_size=1024):
+                        if chunk:
+                            size = file.write(chunk)
+                            downloaded += size
+                            bar.update(size)
+
+                            # Optional: Add a check to stop download if needed
+                            # if downloaded > MAX_DOWNLOAD_SIZE:
+                            #     break
+
+        # Post-download processing
+        if os.path.exists(path) and os.path.getsize(path) > 0:
+            console.print(Panel(
+                f"[bold green]Download completed![/bold green]\n"
+                f"[cyan]File size: [bold red]{internet_manager.format_file_size(os.path.getsize(path))}[/bold red]\n"
+                f"[cyan]Duration: [bold]{print_duration_table(path, description=False, return_string=True)}[/bold]", 
+                title=f"{os.path.basename(path.replace('.mp4', ''))}", 
+                border_style="green"
+            ))
+            return path
+        
+        else:
+            console.print("[bold red]Download failed or file is empty.[/bold red]")
+            return None
+
+    except httpx.HTTPStatusError as http_err:
+        logging.error(f"HTTP error occurred: {http_err}")
+        console.print(f"[bold red]HTTP Error: {http_err}[/bold red]")
+        return None
+    
+    except httpx.RequestError as req_err:
+        logging.error(f"Request error: {req_err}")
+        console.print(f"[bold red]Request Error: {req_err}[/bold red]")
+        return None
+    
+    except Exception as e:
+        logging.error(f"Unexpected error during download: {e}")
+        console.print(f"[bold red]Unexpected Error: {e}[/bold red]")
+        return None