mario_vs_wario_v2/src/get_videos.py

# script used for downloading the source videos from YouTube
# details about which videos to download are in the config.py file

from pytube import YouTube
import os 
from config import VIDEOS_DIR, YT_LINKS_DIR, CLASSES

if __name__ == "__main__":
    
    os.makedirs(os.path.dirname(VIDEOS_DIR), exist_ok=True)

    for class_name in CLASSES:

        print(f"Downloading the {class_name} video...")
    
        yt = YouTube(YT_LINKS_DIR[class_name])

        (
            yt
            .streams.filter(progressive=True, file_extension="mp4")
            .order_by("resolution")
            .desc()
            .first()
            .download(VIDEOS_DIR)
        )

        print("done.")