try: from pytube import Playlist, YouTube from pytube.exceptions import PytubeError, VideoUnavailable from tqdm import tqdm from colorama import init, Fore, Style init(autoreset=True) except ImportError as e: print("Missing required libraries. Install with: pip install pytube tqdm colorama") raise e
def print_summary(self): """Print download summary""" print(f"\n'='*60") print(f"Fore.CYAN📊 DOWNLOAD SUMMARY") print(f"'='*60") print(f"Fore.GREEN✅ Successful: self.stats['successful']") print(f"Fore.YELLOW⏭️ Skipped: self.stats['skipped']") print(f"Fore.RED❌ Failed: self.stats['failed']") print(f"Fore.CYAN📁 Output directory: self.output_dir.absolute()") if self.stats["failed_videos"]: print(f"\nFore.REDFailed videos:") for failed in self.stats["failed_videos"]: print(f" - failed['url'] (failed['error'])") # Save failed URLs to file for retry if self.stats["failed_videos"]: failed_log = self.output_dir / "failed_downloads.txt" with open(failed_log, 'w') as f: for failed in self.stats["failed_videos"]: f.write(f"failed['url']\n") print(f"Fore.YELLOW💾 Failed URLs saved to: failed_log") def main(): """Main function with command-line interface""" parser = argparse.ArgumentParser( description="Download YouTube playlists with quality selection", formatter_class=argparse.RawDescriptionHelpFormatter, epilog=""" Examples: python playlist_downloader.py "https://youtube.com/playlist?list=..." Download only audio (MP3) python playlist_downloader.py "PLAYLIST_URL" --audio-only Download in 720p maximum resolution python playlist_downloader.py "PLAYLIST_URL" --max-res 720p Start from video 5, download only 10 videos python playlist_downloader.py "PLAYLIST_URL" --start 5 --max-videos 10 Download to custom directory python playlist_downloader.py "PLAYLIST_URL" --output "~/Music/Playlists" """ ) youtube playlist downloader python script
def sanitize_filename(self, filename: str) -> str: """Remove invalid characters from filename""" # Remove invalid characters for Windows/Linux/Mac invalid_chars = r'[<>:"/\\|?*]' filename = re.sub(invalid_chars, '_', filename) # Limit filename length if len(filename) > 200: filename = filename[:200] return filename.strip() try: from pytube import Playlist, YouTube from pytube
def get_video_stream(self, video: YouTube): """Get appropriate video stream based on quality settings""" try: if self.download_audio_only: # Get audio stream (highest bitrate) return video.streams.filter(only_audio=True).first() # Get video streams if self.quality == "lowest": stream = video.streams.get_lowest_resolution() else: # highest # Filter by max resolution if specified streams = video.streams.filter(progressive=True, file_extension='mp4') if self.max_resolution != "highest": # Convert '1080p' to '1080' for comparison max_res = int(self.max_resolution.replace('p', '')) streams = streams.filter(res=f"max_resp") if streams: stream = streams.last() # Highest resolution else: # Fallback to non-progressive (video only + audio) stream = video.streams.get_highest_resolution() return stream except Exception as e: print(f"Fore.YELLOW⚠️ Error getting stream: e") return None try: from pytube import Playlist
parser.add_argument("playlist_url", help="YouTube playlist URL") parser.add_argument("-o", "--output", default="downloads", help="Output directory (default: downloads)") parser.add_argument("-q", "--quality", choices=["highest", "lowest"], default="highest", help="Video quality (default: highest)") parser.add_argument("--max-res", default="1080p", help="Maximum resolution (e.g., 720p, 1080p) (default: 1080p)") parser.add_argument("--audio-only", action="store_true", help="Download only audio as MP3") parser.add_argument("--start", type=int, default=1, help="Start downloading from this video index (1-based)") parser.add_argument("--max-videos", type=int, default=None, help="Maximum number of videos to download")