main.py

'''
Script/Appliation: YT Downloader

Description      : The YT Downloader is a simple script/console application which allows us to download yt videos and playlists
                   It is built on top of pytube library, and offers the users freedom to download videos without hassle of 
                   visiting YT video downloading websites and their ads

Author           : Sharath Sunil

Year             : 2020

Version          : 1.10
'''

from Downloader import Downloader
from platform import system 
import pyfiglet
import os


__author__ = 'Sharath Sunil'
__version__= 1.10
yt_downloader_title = pyfiglet.figlet_format('Youtube Downloader',font='slant') #the title for the yt downloader script !

print(yt_downloader_title)
print(f'Version : {__version__}')
print(f'Author  : {__author__}')

YOUTUBE_VID_DIR = 'Youtube Videos' # Parent directory for all Yt videos !
current_user = os.getlogin()

if system()=='Windows':
    DOWNLOADS = f'C:/Users/{current_user}/Downloads' # get the currently logged in user's download dir !
elif system()=='Linux':
    DOWNLOADS = f'/home/{current_user}/Downloads'
else:
    raise Exception("Not Implemented For Current Platform !")
FINAL_PATH = os.path.join(DOWNLOADS, YOUTUBE_VID_DIR) # joining the downloads and the new vids directory !

if not os.path.isdir(FINAL_PATH):
    os.mkdir(FINAL_PATH)
    print(f'New Download Directory Created At {FINAL_PATH}')

download = Downloader()

while True:

    option = int(input('1)Video\n2)Playlist\n3)Exit\n>>'))
    if option == 1:
        link = input('Please Paste A Video Link : ')
        download.download_single_video(FINAL_PATH, link)

    elif option == 2:
        playlist_link = input('Please Paste A Playlist Link :')
        download.download_playlist(FINAL_PATH, playlist_link=playlist_link)

    elif option == 3:
	    print('Exiting........!')
	    break
    
    else:
        print('Invalid Option !!\nPlease Try Again !')