yt_dlp_downloader.py

#!/usr/bin/env python
# encoding=utf-8
import sys
import os
import subprocess
import logging
import argparse

logging.basicConfig(level=logging.INFO)

def parse_arguments():
    parser = argparse.ArgumentParser(description='Download YouTube videos.')
    parser.add_argument('url', help='The URL of the video to download')
    parser.add_argument('-o', '--output', help='The output folder for the downloaded video', default=None)
    args = parser.parse_args()
    return args

def create_output_folder(folder_path):
    if not os.path.exists(folder_path):
        try:
            os.makedirs(folder_path)
            logging.info('Successfully created folder %s' % folder_path)
        except OSError:
            logging.error("Could not create directory")
            sys.exit(1)

def check_yt_dlp():
    try:
        subprocess.run(["yt-dlp", "--version"], check=True)
    except FileNotFoundError:
        logging.error("yt-dlp command not found. Please make sure it is installed and added to the system PATH.")
        sys.exit(1)
    except subprocess.CalledProcessError:
        logging.error("yt-dlp command is not working correctly. Please check if it is installed correctly.")
        sys.exit(1)

def download_video(url, output_folder=None):
    if output_folder and not os.path.exists(output_folder):
        os.makedirs(output_folder)
        commands = f"yt-dlp -civ -f 'bv*+ba' -R infinite {url} -o {output_folder}/%(title)s.%(ext)s"
        
    # if sys.platform == 'win32':
    #     commands = f"yt-dlp -civ -f 'bv*+ba' -R infinite {url}"
    # else:   
    #     commands = f"yt-dlp -civ -f 'bv*+ba/b' -R infinite --exec 'mv {{}} {output_folder}' {url}"
        
    # if output_folder:
    #     commands += f" -o \"{output_folder}/{%(title)s}.%(ext)s\""

    if subprocess.call(commands, shell=True) == 0:
        logging.info('Successfully downloaded the video')
        logging.info(f'Video has been downloaded to: {output_folder if output_folder else os.getcwd()}')
    else:
        logging.error('Failed to download the video')

def main():
    args = parse_arguments()
    if args.output:
        create_output_folder(args.output)
    check_yt_dlp()
    download_video(args.url, args.output)

if __name__ == '__main__':
    main()