tools.py

import os
import sys

from moviepy.video.io.VideoFileClip import VideoFileClip
from pytube import YouTube, Playlist
from youtube_search import YoutubeSearch as yts

SUFFIX = "audio"
Download_loc = sys.argv[0].strip("main.py")


def get_urls(titles: list, suffix="audio"):
    """
    :param titles: List of music titles to download
    :param suffix: Optional, word added at the end of the query (default: audio)
    :return: Dictionary linking music titles (input) to their youtube urls
    """
    urls = {}
    for j in titles:
        search = yts(j + suffix, max_results=1).to_dict()
        name = search[0]['title']
        url = "https://www.youtube.com" + search[0]['url_suffix']
        urls[name] = url
    return urls


def download_from_url(url: str, download_path="Downloads", quick: bool = False):
    """
    :param quick: Boolean, choose if the download will be  long or quick. Quick will result in a quick convert, and an
    unreadable file by Itunes and most programs
    :param url: Youtube urls of musics to download
    :param download_path: Optional path where to download the music file
    :return: Downloads the music, and returns absolute path of the downloaded music
    """
    vid = YouTube(url)
    if quick:
        stream = vid.streams.filter(only_audio=True).first()
    else:
        stream = vid.streams.filter().first()
    outputPath = stream.download(output_path=download_path)
    return outputPath


def download_from_name(name, suffix="audio", download_path="Downloads", quick_mode=False):
    """
    :param quick_mode: Whether or not to convert it quickly
    :param download_path: path to where the file will be downloaded
    :param suffix: suffix for the search on youtube ("audio" by default)
    :param name: Name of the music to download
    :return: Downloads the music, and returns absolute path of the downloaded music
    """
    search = yts(name + suffix, max_results=1).to_dict()
    link = "https://www.youtube.com" + search[0]['url_suffix']
    return download_from_url(link, download_path, quick_mode)


def download_from_yt_playlist(playlist, download_path="Downloads", quick_mode=False):
    """
    :param playlist: Youtube playlist link
    :param download_path: Path where to download the songs
    :param quick_mode: Whether or not to convert it quickly and badly
    :return: True when done
    """
    try:
        p = Playlist(playlist)
        for song_url in p.video_urls:
            file = download_from_url(song_url, download_path, quick_mode)
            if quick_mode:
                quick_convert(file)
            else:
                long_convert(file)
            print("Downloaded", os.path.splitext(os.path.basename(file))[0])
    except KeyError:
        return False
    return True


def quick_convert(file):
    """
    :param file: Absolute path to the file to convert
    :return: None, but converts the file to mp3
    """
    name, ext = os.path.splitext(file)
    os.rename(file, name + ".mp3")


def long_convert(file):
    """
    :param file: Absolute path to the file to convert
    :return: None, but converts the file to mp3
    """
    mp3_file = os.path.splitext(file)[0] + ".mp3"
    video_clip = VideoFileClip(file)
    audio_clip = video_clip.audio
    audio_clip.write_audiofile(mp3_file, verbose=False, logger=None)
    audio_clip.close()
    video_clip.close()
    os.remove(file)


def download(name, suffix="audio", download_location="Downloads", quick=False) -> str:
    """
    :param name: Name of the song to download
    :param suffix: suffix for the search on youtube ("audio" by default)
    :param download_location: path to downloaded song
    :param quick: whether to quickly convert the song, or keep the metadata and convert it the right way
    :return: Downloads the music as mp4, converts it to mp3, and returns absolute path of the downloaded music
    """
    file = download_from_name(name, suffix, download_location, quick)
    song_name = os.path.splitext(os.path.basename(file))[0]
    if quick:
        quick_convert(file)
    else:
        long_convert(file)
    return song_name