feat: Add user prompt for managing partial downloads

Adds a new function to prompt users on whether to keep or delete partial download files. This improves user experience by simplifying the management of temporary files during the download process. The previous inline code has been refactored into a dedicated function for better readability and maintainability.

Author: MPZ-00

dl.py

@@ -132,6 +132,25 @@ def generate_custom_filename(base: str, episode_tag: str, ext: str = ".mp4") ->
     safe_base = re.sub(r'[\\/*?:"<>|]', "_", base)
     return f"{safe_base}_{episode_tag}{ext}"
 
+def prompt_partial_file_cleanup():
+    """Prompt user to keep or delete partial download files."""
+    print("\n[?] What would you like to do with partial downloads?")
+    print("  [K]eep - Keep .part files to resume later")
+    print("  [D]elete - Remove all .part files and start fresh next time")
+    
+    try:
+        choice = input("Your choice (K/D): ").strip().upper()
+        if choice == 'D':
+            print("[*] Cleaning up temporary files...")
+            delpartfiles()
+            print("[*] All .part files removed.")
+        elif choice == 'K':
+            print("[*] Keeping .part files for resume.")
+        else:
+            print("[*] Invalid choice, keeping .part files by default.")
+    except (EOFError, KeyboardInterrupt):
+        print("\n[*] Keeping .part files by default.")
+
 def parse_arguments():
     parser = argparse.ArgumentParser(
         description="Multi-threaded downloader for video sources with advanced detection methods.",
@@ -175,22 +194,7 @@ def main():
                 sys.stderr.flush()
 
                 # Ask user what to do with partial downloads
-                print("\n[?] What would you like to do with partial downloads?")
-                print("  [K]eep - Keep .part files to resume later")
-                print("  [D]elete - Remove all .part files and start fresh next time")
-                
-                try:
-                    choice = input("Your choice (K/D): ").strip().upper()
-                    if choice == 'D':
-                        print("[*] Cleaning up temporary files...")
-                        delpartfiles()
-                        print("[*] All .part files removed.")
-                    elif choice == 'K':
-                        print("[*] Keeping .part files for resume.")
-                    else:
-                        print("[*] Invalid choice, keeping .part files by default.")
-                except (EOFError, KeyboardInterrupt):
-                    print("\n[*] Keeping .part files by default.")
+                prompt_partial_file_cleanup()
             else:
                 # Normal completion - clean up
                 print("[*] Cleaning up temporary files...")
@@ -311,22 +315,7 @@ def list_dl(doc, args):
                 sys.stderr.flush()
 
                 # Ask user what to do with partial downloads
-                print("\n[?] What would you like to do with partial downloads?")
-                print("  [K]eep - Keep .part files to resume later")
-                print("  [D]elete - Remove all .part files and start fresh next time")
-                
-                try:
-                    choice = input("Your choice (K/D): ").strip().upper()
-                    if choice == 'D':
-                        print("[*] Cleaning up temporary files...")
-                        delpartfiles()
-                        print("[*] All .part files removed.")
-                    elif choice == 'K':
-                        print("[*] Keeping .part files for resume.")
-                    else:
-                        print("[*] Invalid choice, keeping .part files by default.")
-                except (EOFError, KeyboardInterrupt):
-                    print("\n[*] Keeping .part files by default.")
+                prompt_partial_file_cleanup()
             else:
                 # Normal shutdown - wait for completion
                 executor.shutdown(wait=True)