DDownloader 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

DDownloader/main.py

@@ -1,31 +1,67 @@
-import argparse
+import os
+import re
 import logging
 import coloredlogs
+from colorama import Fore, Style
+import platform, time # Added to detect platform
+from DDownloader.modules.args_parser import parse_arguments
 from DDownloader.modules.banners import banners
 from DDownloader.modules.dash_downloader import DASH
 from DDownloader.modules.hls_downloader import HLS
-import os, re
 
+# Setup logger
 logger = logging.getLogger("+ MAIN + ")
 coloredlogs.install(level='DEBUG', logger=logger)
 
 def validate_directories():
-    if not os.path.exists('downloads'):
-        os.makedirs('downloads')
-        logger.debug("Created 'downloads' directory.")
+    """Ensure necessary directories exist."""
+    downloads_dir = 'downloads'
+    if not os.path.exists(downloads_dir):
+        os.makedirs(downloads_dir)
+        logger.debug(f"Created '{downloads_dir}' directory.")
 
-def parse_arguments():
-    parser = argparse.ArgumentParser(description="Download DASH or HLS streams with decryption keys.")
-    parser.add_argument("-u", "--url", required=True, help="Manifest URL pointing to the stream (.mpd or .m3u8).")
-    parser.add_argument("-k", "--key", action="append", help="Decryption keys in the format KID:KEY. Use multiple -k options for multiple keys.")
-    parser.add_argument("-o", "--output", required=True, help="Output file name.")
-    return parser.parse_args()
+def detect_platform():
+    """Detect the platform the script is running on."""
+    system_platform = platform.system().lower()
+    if system_platform == 'windows':
+        return 'Windows'
+    elif system_platform == 'linux':
+        return 'Linux'
+    elif system_platform == 'darwin':
+        return 'MacOS'
+    else:
+        return 'Unknown'
+
+def display_help():
+    """Display custom help message with emoji."""
+    print(
+        f"{Fore.WHITE}+" + "=" * 80 + f"+{Style.RESET_ALL}\n"
+        f"{Fore.CYAN}{'Option':<40}{'Description':<90}{Style.RESET_ALL}\n"
+        f"{Fore.WHITE}+" + "=" * 80 + f"+{Style.RESET_ALL}\n"
+        f"  {Fore.GREEN}-u, --url{' ' * 22}{Style.RESET_ALL}URL of the manifest (mpd/m3u8) 🌐\n"
+        f"  {Fore.GREEN}-p, --proxy{' ' * 20}{Style.RESET_ALL}A proxy with protocol (http://ip:port) 🌍\n"
+        f"  {Fore.GREEN}-o, --output{' ' * 19}{Style.RESET_ALL}Name of the output file 💾\n"
+        f"  {Fore.GREEN}-k, --key{' ' * 22}{Style.RESET_ALL}Decryption key in KID:KEY format 🔑\n"
+        f"  {Fore.GREEN}-h, --help{' ' * 21}{Style.RESET_ALL}Show this help message and exit ❓\n"
+        f"{Fore.WHITE}+" + "=" * 80 + f"+{Style.RESET_ALL}\n"
+    )
 
 def main():
     banners()
+    time.sleep(1)
+    platform_name = detect_platform()
+    logger.info(f"Running on platform: {platform_name}\n")
+    time.sleep(1)
+    
     validate_directories()
-    args = parse_arguments()
+    try:
+        args = parse_arguments()
+    except SystemExit:
+        display_help()
+        exit(1)
 
+    # Detect and initialize appropriate downloader
+    downloader = None
     if re.search(r"\.mpd\b", args.url, re.IGNORECASE):
         logger.info("DASH stream detected. Initializing DASH downloader...")
         downloader = DASH()
@@ -34,24 +70,30 @@ def main():
         downloader = HLS()
     else:
         logger.error("Unsupported URL format. Please provide a valid DASH (.mpd) or HLS (.m3u8) URL.")
-        return
+        exit(1)
 
+    # Configure downloader
     downloader.manifest_url = args.url
     downloader.output_name = args.output
     downloader.decryption_keys = args.key or []
+    downloader.proxy = args.proxy  # Add proxy if provided
 
+    # Log provided decryption keys
     if downloader.decryption_keys:
+        logger.info("Decryption keys provided:")
         for key in downloader.decryption_keys:
-            logger.info(f"Decryption key(s) provided: --key {key}\n")
-            print(Fore.MAGENTA + "=" * 80 + Fore.RESET)
+            logger.info(f"  --key {key}")
+        print(Fore.MAGENTA + "=" * 80 + Fore.RESET)
 
+    # Execute downloader
     try:
         if isinstance(downloader, DASH):
             downloader.dash_downloader()
-        else:
+        elif isinstance(downloader, HLS):
             downloader.hls_downloader()
     except Exception as e:
-        logger.error(f"An error occurred: {e}")
+        logger.error(f"An error occurred during the download process: {e}")
+        exit(1)
 
 if __name__ == "__main__":
-    main()
+    main()

DDownloader/modules/args_parser.py

@@ -0,0 +1,24 @@
+import argparse
+from colorama import Fore, Style
+
+def parse_arguments():
+    """Parse and return command-line arguments."""
+    # Create the ArgumentParser with no default help and no description
+    parser = argparse.ArgumentParser(
+        add_help=False,  # Disable default help
+        usage="",  # Suppress the default usage message
+    )
+
+    # Add arguments (these will not include the default descriptions)
+    parser.add_argument("-u", "--url", required=True, help=argparse.SUPPRESS)
+    parser.add_argument("-p", "--proxy", help=argparse.SUPPRESS)
+    parser.add_argument("-o", "--output", required=True, help=argparse.SUPPRESS)
+    parser.add_argument("-k", "--key", action="append", help=argparse.SUPPRESS)
+    parser.add_argument(
+        "-h", "--help",
+        action="help",
+        default=argparse.SUPPRESS,
+        help=argparse.SUPPRESS
+    )
+
+    return parser.parse_args()

DDownloader/modules/banners.py

@@ -20,7 +20,7 @@ def banners():
     stdout.write(""+Fore.YELLOW   +"╔════════════════════════════════════════════════════════════════════════════╝\n")
     stdout.write(""+Fore.YELLOW   +"║ \x1b[38;2;255;20;147m• "+Fore.GREEN+"GITHUB             "+Fore.RED+"    |"+Fore.LIGHTWHITE_EX+"   GITHUB.COM/THATNOTEASY                        "+Fore.YELLOW+"║\n")
     stdout.write(""+Fore.YELLOW   +"╚════════════════════════════════════════════════════════════════════════════╝\n") 
-    print(f"{Fore.YELLOW}[DDownloader] - {Fore.GREEN}Download DASH or HLS streams with decryption keys. - {Fore.RED}[V0.2.0] \n{Fore.RESET}")
+    print(f"{Fore.YELLOW}[DDownloader] - {Fore.GREEN}Download DASH or HLS streams with decryption keys. - {Fore.RED}[V0.2.1] \n{Fore.RESET}")
 
 
 

DDownloader/modules/dash_downloader.py

@@ -12,6 +12,7 @@ class DASH:
     def __init__(self):
         self.manifest_url = None
         self.output_name = None
+        self.proxy = None
         self.decryption_keys = []
         self.binary_path = self._get_binary_path()
 
@@ -53,8 +54,14 @@ class DASH:
             '--del-after-done',
             '--save-name', self.output_name
         ]
+
         for key in self.decryption_keys:
             command.extend(['--key', key])
+
+        if self.proxy:
+            if not self.proxy.startswith("http://"):
+                self.proxy = f"http://{self.proxy}"
+            command.extend(['--custom-proxy', self.proxy])
         # logger.debug(f"Built command: {' '.join(command)}")
         return command
 

DDownloader/modules/hls_downloader.py

@@ -12,6 +12,7 @@ class HLS:
     def __init__(self):
         self.manifest_url = None
         self.output_name = None
+        self.proxy = None
         self.decryption_keys = []
         self.binary_path = self._get_binary_path()
 
@@ -44,7 +45,7 @@ class HLS:
     def _build_command(self):
         command = [
             self.binary_path,
-            f'"{self.manifest_url}"',  # Properly quote the URL
+            f'"{self.manifest_url}"',
             '--select-video', 'BEST',
             '--select-audio', 'BEST',
             '-mt',
@@ -54,8 +55,14 @@ class HLS:
             '--del-after-done',
             '--save-name', self.output_name
         ]
+
         for key in self.decryption_keys:
             command.extend(['--key', key])
+
+        if self.proxy:
+            if not self.proxy.startswith("http://"):
+                self.proxy = f"http://{self.proxy}"
+            command.extend(['--custom-proxy', self.proxy])
         # logger.debug(f"Built command: {' '.join(command)}")
         return command
 

{DDownloader-0.2.0.dist-info → DDownloader-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: DDownloader
-Version: 0.2.0
+Version: 0.2.1
 Summary: A downloader for DRM-protected content.
 Home-page: https://github.com/ThatNotEasy/DDownloader
 Author: ThatNotEasy

DDownloader-0.2.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+DDownloader/__init__.py,sha256=lVZwmZNId0Dai7XBQpxglmJtIxAtZplRHDsvobL2UNo,33
+DDownloader/main.py,sha256=9R2OW58YMZ4ECXbxIHhaZCmTY7ZJ3TU7Lk_S3eVslOI,3650
+DDownloader/modules/__init__.py,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
+DDownloader/modules/args_parser.py,sha256=JsmemrQcxBSz8aWHWTpAO6I4BfGoK6GmJvJFuhRhZ7Y,919
+DDownloader/modules/banners.py,sha256=xgDGg8djkrbDBIuWefZPSEF_8iQ-WvX-vrZ2q43tZSs,3989
+DDownloader/modules/dash_downloader.py,sha256=iJWs7qo4qG8vu-7ROFlrurVSo2uvjlOTAOeb_SaXbqo,3230
+DDownloader/modules/hls_downloader.py,sha256=o7X6_w6gQNxot1NIHm7WYSfPwj7Q5Xbsvo1MCYzma1k,3284
+DDownloader/modules/streamlink.py,sha256=F8vneSkxgGgqxRBhCHxvID-KwltpDG2QerH6QsAHuxE,506
+DDownloader-0.2.1.dist-info/METADATA,sha256=fHkn3lPJDel63pvxyN9MGSdAO63STzGgPIlaC2Hj6xg,2016
+DDownloader-0.2.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+DDownloader-0.2.1.dist-info/entry_points.txt,sha256=tCZVr_SRONlWlMFsVKgcPj3lxe9gBtWD4GuWukMv75g,54
+DDownloader-0.2.1.dist-info/top_level.txt,sha256=INZYgY1vEHV1MIWTPXKJL8j8-ZXjWb8u4XLuU3S8umY,12
+DDownloader-0.2.1.dist-info/RECORD,,

DDownloader-0.2.0.dist-info/RECORD

@@ -1,12 +0,0 @@
-DDownloader/__init__.py,sha256=lVZwmZNId0Dai7XBQpxglmJtIxAtZplRHDsvobL2UNo,33
-DDownloader/main.py,sha256=Fr16CMszpIaQnAYWa2404Zkq7lpDST79AkPLY3FC78o,2151
-DDownloader/modules/__init__.py,sha256=frcCV1k9oG9oKj3dpUqdJg1PxRT2RSN_XKdLCPjaYaY,2
-DDownloader/modules/banners.py,sha256=qVx4D57E4bnHwu2t4eBgEN5Xnc5DsNn3CRtkipfjWBA,3989
-DDownloader/modules/dash_downloader.py,sha256=9eigcmQXY3kYggf6vN5mik9q8OqT_5dgyzaZ4CJ4H6E,3008
-DDownloader/modules/hls_downloader.py,sha256=0C3hyITZEbZOvGQhaWCZCguIIhThwlQYVduvxoHGCLA,3088
-DDownloader/modules/streamlink.py,sha256=F8vneSkxgGgqxRBhCHxvID-KwltpDG2QerH6QsAHuxE,506
-DDownloader-0.2.0.dist-info/METADATA,sha256=vLDqMrNYyzzgEZRYIWVbvA39s2QPGj2HTrDubZBSBO0,2016
-DDownloader-0.2.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-DDownloader-0.2.0.dist-info/entry_points.txt,sha256=tCZVr_SRONlWlMFsVKgcPj3lxe9gBtWD4GuWukMv75g,54
-DDownloader-0.2.0.dist-info/top_level.txt,sha256=INZYgY1vEHV1MIWTPXKJL8j8-ZXjWb8u4XLuU3S8umY,12
-DDownloader-0.2.0.dist-info/RECORD,,