main.py

#!/bin/python3

from pytube import YouTube  # library to donwload songs and urls
from pytube import Playlist  # library do donwload playlists
# library to gather ytmusic search results with urls, artworks and other metadatas
from ytmusicapi import YTMusic
from pydub import AudioSegment  # lib to convert audio formats
# execute os level commands, used to clean temporary files (webm) after download
import os
import music_tag  # lib to add tags and artworks to mp3 files
import wget  # (wget) gather files from the web
from PIL import Image  # image manipulation, used to crop artworks from yt canvas

ytmusic = YTMusic()

# check if the path variable is not define and ask the user to define one
if os.path.isfile("path.txt") == False:
    path = input("Paste here ur preferred download directory: ").strip()
    with open('path.txt', 'w') as f:
        f.write(path)
else:  # if the path_var is defined ask the user if he wanna update it
    with open('path.txt', 'r') as f:
        path = f.read().strip()
    if input(f"Current Download path: {path} - wanna update? (y/n)").strip() == "y":
        path = input("Paste here ur preferred download directory: ").strip()
        with open('path.txt', 'w') as f:  # write the new path_var to the file
            f.write(path)


def term_text(txt):  # formatting text to be terminal friendly
    txt = txt.replace(" ", "\ ")
    txt = txt.replace("(", "\(")
    txt = txt.replace(")", "\)")
    return txt


def down_song(link):  # function to download songs
    yt = YouTube(link)
    print(f"\nDonwloading {yt.title}")
    file_path = f"{path}{yt.author.replace(' - Topic', '')} - {yt.title.replace('/','-')}.webm"
    yt.streams.filter(only_audio=True, abr="160kbps").first().download(
        output_path=f"{path}", filename=f"{yt.author.replace(' - Topic', '')} - {yt.title.replace('/','-')}.webm")
    try:
        mp3_conv(file_path, file_path.replace("webm", "mp3"))
        tags_and_art(file_path.replace("webm", "mp3"),
                     yt.title, None, yt.author.replace(' - Topic', ''), None, yt.thumbnail_url)

    except Exception as e:
        return("mp3_conv or tags_and_art failed: ", e)

    os.system(f"rm -r {term_text(file_path)}")  # clean webm files

    print("\nDONE\n")


def down_plist(link):  # function do download playlists
    p = Playlist(link)
    title = p.title.replace("Album - ", "")
    numb = 1
    print(f"Downloading {title}\n")

    for song in p.videos:
        file_path = f"{path}{song.author.replace(' - Topic', '')} - {title}/webm/"
        print(f"\nDownloading {song.title}")
        song.streams.filter(only_audio=True, abr="160kbps").first().download(
            output_path=f"{path}{song.author.replace(' - Topic', '')} - {title}/webm/", filename=f"{str(numb)} - {song.title.replace('/','-')}.webm")
        try:
            mp3_conv(f"{file_path}{str(numb)} - {song.title.replace('/','-')}.webm",
                     f"{path}{song.author.replace(' - Topic', '')} - {title}/{song.title.replace('/','-')}.mp3")
            tags_and_art(f"{path}{song.author.replace(' - Topic', '')} - {title}/{song.title.replace('/','-')}.mp3",
                         song.title, title, song.author.replace(' - Topic', ''), str(numb), song.thumbnail_url)

        except Exception as e:
            return("mp3_conv or tags_and_art failed: ", e)
        numb += 1

    os.system(f"rm -r {term_text(file_path[:-1])}")
    print("\nDONE\n")  # clean webm files


# add tags and artwork to the mp3 files
def tags_and_art(file_path, song_name, album, author, trk_nmbr, art_link):
    f = music_tag.load_file(file_path)
    f['title'] = song_name
    f['artist'] = author
    if album != None:
        f['album'] = album
    if trk_nmbr != None:
        f['tracknumber'] = trk_nmbr
    file_name = wget.download(art_link)
    fl_nm = Image.open(file_name)
    file_name_cr = fl_nm.crop((140, 60, 500, 420))
    file_name_cr.save("img.jpg")
    with open("img.jpg", 'rb') as img_in:
        f['artwork'] = img_in.read()
    f.save()
    os.system('rm img.jpg')
    os.system(f'rm {term_text(file_name)}')

#convert webm to mp3


def mp3_conv(file_path, out_path):
    webm_audio = AudioSegment.from_file(
        f"{file_path}", format="webm")
    webm_audio.export(
        f"{out_path}", format="mp3")

#main function


def mdown(id, url):
    if id == None:
        link = url
        try:
            down_song(link)

        except Exception as e:
            print("down_song error: ", e)

            try:
                down_plist(link)
            except Exception as err:
                print("down_plist error: ", err)
    else:
        for i in id:
            link = "https://music.youtube.com/"+i
            try:
                down_song(link)

            except Exception as er:
                #print("down_song error: ", er)

                try:
                    down_plist(link)
                except Exception as e:
                    print("down_plist error: ", e)


while True:
    mode = input("Chose MODE album(a) song(s) url(u) quit(q): ")
    if mode.lower().strip() == "a":
        mode = "albums"
        key = "browseId"
    elif mode.lower().strip() == "s":
        mode = "songs"
        key = "videoId"

    elif mode.lower().strip() == "q":
        break

    elif mode.lower().strip() == "u":
        url = input("\npaste URL here: ")
        id = None
        mdown(id, url)
        continue
    else:
        print("sorry try again")
        continue
    try:
        results = ytmusic.search(
            input("\nType the song/album name here (enter to go back) : "), filter=f"{mode}")
    except:
        continue
    print("\nHere's what we found: \n")
    n = 0
    for song in results:
        title = song['title']
        art = song['artists']
        artist = (art[0])['name']
        print(f" {n}: {title} by {artist}")
        n += 1
    sns = input(
        f"\nType in the song/album numbers (0 to {n-1}) separated by a space or search again (enter): ")
    if sns == "":
        continue
    sns = (sns.strip()).split(" ")
    if mode == "albums":
        id = []
        for s in sns:
            alb_dt = ytmusic.get_album(browseId=f"{(results[int(s)])[key]}")
            id.append("playlist?list=" + alb_dt['audioPlaylistId'])
        url = None
        mdown(id, url)
    else:
        id = []
        for s in sns:
            id.append("watch?v=" + (results[int(s)])[key])
        url = None
        mdown(id, url)