Youtube-DL-GUI.pyw

import os
import subprocess
from io import BytesIO
from configparser import ConfigParser
import webbrowser
import requests
import PySimpleGUI as sg
import PIL.Image as Image
import PIL.ImageTk as ImageTk

# Get the default download path
def get_download_path():
    if os.name == 'nt':
        import winreg
        sub_key = r'SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders'
        downloads_guid = '{374DE290-123F-4565-9164-39C4925E467B}'
        with winreg.OpenKey(winreg.HKEY_CURRENT_USER, sub_key) as key:
            location = winreg.QueryValueEx(key, downloads_guid)[0]
        return location
    else:
        return os.path.join(os.path.expanduser('~'), 'Downloads')

# First is only needed when Tinker has not been initialized
def get_img_data(url, maxsize=(320, 180), first= False):
    # Get the image from the url
    response = requests.get(url, stream=True)

    # Transform the bytes into an Image
    responseImage = Image.open(BytesIO(response.content))

    # Adjust image size
    responseImage.thumbnail(maxsize)

    if first:
        # Create a temp BytesIO
        imageBytesIO = BytesIO()

        # Save in imageBytesIo
        responseImage.save(imageBytesIO, format="PNG")

        # Delete the img
        del responseImage

        # Return the bytes from the image
        return imageBytesIO.getvalue()

    # Return the image
    return ImageTk.PhotoImage(responseImage)

# String to bool
def stringToBool (val):
    return val.lower() in ["true", "1", "enabled"]

# Read the config file
def read_config(section, key):
    # Config file
    defaultsDict = {
        'format': '(bestvideo[ext=mp4][fps>30]/bestvideo[ext=mp4])+bestaudio[ext=m4a]',
        'location': get_download_path(),
        'file': "%%(title)s.%%(ext)s",
        'extra': '--merge-output-format mp4',
        'adv': "False",
        'cmd': "True"
    }
    # Set the config parser with this defaults
    config = ConfigParser(defaults = defaultsDict)

    # Read from same directory
    config.read('config.ini')

    # Check if it has the section
    if not config.has_section(section):
        config.add_section(section)

    return config.get('main', key)

# Write the config file
def write_config(section, key, val):
    # Set the config parser with this defaults
    config = ConfigParser()

    # Read from same directory
    config.read('config.ini')

    # Check if it has the section
    if not config.has_section(section):
        config.add_section(section)

    config.set('main', key, val)

    with open('config.ini', 'w') as configFile:
        config.write(configFile)

def toggle_row (window, value):
    if value:
        window.unhide_row()
    else:
        window.hide_row()
    window.Update(visible = value)

# Theme
sg.theme("DarkAmber")

# Components to get from the net
imgOnline = sg.Text("Image not found, is there an active internet connection?", pad=(0, 30))
iconToShow = None
try:
    imgOnline = sg.Image(data=get_img_data("https://raw.githubusercontent.com/TheFallender/youtube-dl-gui/master/app.png", first = True), size = (320, 180), enable_events=True, key="-IMAGE_CLICK-", pad=(0, 30))
    iconToShow = get_img_data("https://raw.githubusercontent.com/TheFallender/youtube-dl-gui/master/app.ico", maxsize=(256,256), first=True)
except:
    pass

# Advanced visible
advVisible = stringToBool(read_config("main", "adv"))

# Layout
appLayout = [
    [sg.Text("  Youtube DL GUI by TheFallender  ", text_color="BLACK", background_color="WHITE", relief=sg.RELIEF_RAISED, border_width=5)],
    [sg.Text("G̲i̲t̲H̲u̲b̲ R̲e̲po̲", background_color="WHITE", text_color="BLUE", relief=sg.RELIEF_GROOVE, border_width=4, enable_events=True, key="-GITHUB_CLICK-")],
    
    [imgOnline],
    
    [
        sg.Text("Youtube Link: "), 
        sg.InputText(size=(25,1), key = "-URL_INPUT-")
    ],
    [
        sg.Checkbox("Show Advanced", default = advVisible, key="-CB_ADV-", enable_events=True),
        sg.Checkbox("Show CMD", default = stringToBool(read_config("main", "cmd")), key="-CB_CMD-")
    ],

    [sg.Text("Format: ", key = "-INFO_F-", visible=advVisible)],
    [sg.InputText(default_text=read_config("main", "format"), size=(40,1), key = "-PARAM_F-", visible=advVisible)],

    [sg.Text("Download Location: ", key = "-INFO_O-", visible=advVisible)],
    [sg.InputText(default_text=read_config("main", "location"), size=(40,1), key = "-PARAM_O-", visible=advVisible)],

    [sg.Text("Filename format: ", key = "-INFO_FN-", visible=advVisible)],
    [sg.InputText(default_text=read_config("main", "file"), size=(40,1), key = "-PARAM_FN-", visible=advVisible)],

    [sg.Text("Extra Parameters: ", key = "-INFO_E-", visible=advVisible)],
    [sg.InputText(default_text=read_config("main", "extra"), size=(40,1), key = "-PARAM_E-", visible=advVisible)],

    [
        sg.Button('Download', key="-Download-", pad=(5,(20,10))),
        sg.Button('Save', key="-Save-", pad=(5,(20,10))),
        sg.Button('Cancel', key="-Cancel-", pad=(5,(20,10))),
    ]
]

# Create the window
window = sg.Window("Youtube DL GUI", appLayout, element_justification='c', icon=iconToShow)

# Event Loop
while True:
    event, values = window.read()

    if event == "-GITHUB_CLICK-":
        webbrowser.open('https://github.com/TheFallender/youtube-dl-gui')

    if event == "-IMAGE_CLICK-":
        webbrowser.open('https://www.youtube.com')

    if event == "-CB_ADV-":
        # Toggle the status of the windows
        status = values['-CB_ADV-']
        toggle_row(window['-INFO_F-'], status)
        toggle_row(window['-PARAM_F-'], status)
        toggle_row(window['-INFO_O-'], status)
        toggle_row(window['-PARAM_O-'], status)
        toggle_row(window['-INFO_FN-'], status)
        toggle_row(window['-PARAM_FN-'], status)
        toggle_row(window['-INFO_E-'], status)
        toggle_row(window['-PARAM_E-'], status)
        # Required to position the elements below
        toggle_row(window['-Cancel-'], False)
        toggle_row(window['-Cancel-'], True)

    if event == "-Download-" and values['-URL_INPUT-'] != "":
        # File path format for UNIX and Windows
        cleanUrl = values['-URL_INPUT-'].replace("?list=WL", "")
        filePath = "\"{0}{1}{2}\"".format(values['-PARAM_O-'], ("\\" if os.name == 'nt' else "/"), values['-PARAM_FN-'])

        # Format the command with the given input
        command = "youtube-dl -f \"{0}\" -o {1} {2} ".format(values['-PARAM_F-'], filePath, values['-PARAM_E-'])
        command += cleanUrl
        
        # Wether or not to show a Command Prompt
        if values['-CB_CMD-']:
            if os.name == 'nt':
                cmdShow = 'cmd /c (echo Youtube-DL GUI by TheFallender: & echo Comand to run: & echo {0} & echo. & {0})'.format(command)
                subprocess.call(cmdShow)
            else:
                cmdShow = 'bash -c "echo Youtube-DL GUI by TheFallender:; echo Comand to run:; echo {0}; echo ""; {0};"'.format(command.replace("\"", "\\\""))
                os.system("gnome-terminal --wait -- {0}".format(cmdShow))
        else:
            subprocess.Popen(command, shell=True).communicate()
        break

    if event == "-Save-":
        write_config('main', 'format', values['-PARAM_F-'].replace('%', "%%"))
        write_config('main', 'location', values['-PARAM_O-'].replace('%', "%%"))
        write_config('main', 'file', values['-PARAM_FN-'].replace('%', "%%"))
        write_config('main', 'extra', values['-PARAM_E-'].replace('%', "%%"))
        write_config('main', 'adv', str(values['-CB_ADV-']))
        write_config('main', 'cmd', str(values['-CB_CMD-']))

    if event == "-Cancel-" or event == sg.WIN_CLOSED:
        break

window.close()