src/video_downloader.py

import subprocess
from pathlib import Path

import msg
from utils import KeepDir

HOME = Path.cwd()


def download_video(url: str, folder: str = 'background'):
    """
    Downloads a video from the given URL and saves it to the specified folder.

    Args:
        url (str): The URL of the video to download.
        folder (str, optional): The name of the folder to save the video in. Defaults to 'background'.
    """
    directory = HOME / folder
    if not directory.exists():
        directory.mkdir()

    with KeepDir() as keep_dir:
        keep_dir.chdir(folder)
        subprocess.run(['yt-dlp', '-f bestvideo[ext=mp4]',
                       '--restrict-filenames', url], check=True)