youtubeDL.py

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

if len(sys.argv) < 2 or len(sys.argv) > 5:
    print('Usage: <script> <URL> or <script> <URL> <New Folder Created>')
    sys.exit()

script = sys.argv[0]
url = sys.argv[1]

if len(sys.argv) < 4:
    local_path = os.getcwd() if len(sys.argv) == 2 else sys.argv[2]
    if not os.path.exists(local_path):
        try:
            os.makedirs(local_path)
            print('Successfully created folder %s' % local_path)
        except OSError:
            print("Could not create directory")
            sys.exit(1)
else:
    print('You must provide a path to download the video')

if len(sys.argv) == 2:
    commands = f"yt-dlp -civ -f bestvideo+bestaudio -R infinite  {url}"
else:
    if sys.platform == 'win32':
        commands = f"yt-dlp -civ -f bestvideo+bestaudio -R infinite {url}"
    else:
        commands = f"yt-dlp -civ -f 'bv*+ba/b' -R infinite  --exec 'mv {{}} {local_path}' {url}"

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

if subprocess.call(commands, shell=True) == 0:
    print('Successfully downloaded the video')
    print(f'Video has been downloaded to: {local_path}')
else:
    print('Failed to download the video')