DeepakChakravarthy / YoutubeDownloader-FYI

Main.py

# ------------------Import Files---------------------------------------------
import pytube
from pytube import YouTube
from pytube import Playlist
import sys
import re
import requests as r
import wget
from tkinter import PhotoImage, Label, StringVar, Entry, Button, IntVar, \
     Radiobutton, messagebox, filedialog, Frame, X, Y, LEFT, RIGHT, \
         N, FLAT, CENTER, Canvas, Scrollbar, VERTICAL, BOTH, SUNKEN, SOLID
import tkinter as tk
from tkinter import ttk
import clipboard
import webbrowser
from tkinterhtml import HtmlFrame

# ============================ Window Design ================================
top = tk.Tk()
# Styles and Designs Functions for Screens
# Color set for youtube for selectcolor, bg & activebackground #0B1D6F
yt_col = "#0B1D6F"  # Youtube bg Color
# Color set for facebook for selectcolor, bg & activebackground #075C90
fb_col = '#075C90' # Facebook bg Color
col_show = 'white'  # pure white fg
col_select = 'gray'  # gray for activeforeground
COLOR_1 = "#90B3DD" # TabControl Config Color

# Font Style and Size
large_font = font = ('Cascadia Mono', 16)  # style='my.head'
DisFont = font = ('', 14)  # style='my.default'
SubFont = font = ('Consolas', 12)  # style='my.sub'
tab_font = font = ('Consolas', 12)  # Tab font style

# ------------------------- Tab Control Style Config ----------------------

tab_show = "#90B3DD"
tab_select = "#C8D9EE"
style = ttk.Style()
style.theme_create("pop",
                   parent="alt",
                   settings={
                       "TNotebook": {
                           "configure": {
                               "tabmargins": [5, 5, 2, 0]
                           }
                       },
                       "TNotebook.Tab": {
                           "configure": {
                               "padding": [100, 5],
                               "font": tab_font,
                               "background": tab_show
                           },
                           "map": {
                               "background": [("selected", tab_select)],
                               "expand": [("selected", [1, 2, 2, 0])]
                           }
                       }
                   })

style.theme_use("pop")

# Style for Buttons
s_btn = ttk.Style() # Toggle Button
s_btn.configure('my.TButton',
                anchor='center',
                font=('Cascadia Mono', 8))

m_btn = ttk.Style() # Download and Cancel Button for both Tab1 & 2
m_btn.configure('WD.TButton', anchor='center',
                activebackground="lightgreen",
                activeforeground="blue",
                font=('Cascadia Mono', 16))

# Notebook Style Config
noteStyler = ttk.Style()
noteStyler.configure("TNotebook",
                    background=COLOR_1,
                    anchor=CENTER,
                    borderwidth=0)

# ------------------------ Screen and Tab -----------------------------------

tabControl = ttk.Notebook(top)

tab1 = ttk.Frame(tabControl)        # Tab1 - Youtube
tabControl.add(tab1, text='Youtube')

tab2 = ttk.Frame(tabControl)        # Tab2 - Facebook
tabControl.add(tab2, text='Facebook')

tab3 = ttk.Frame(tabControl)        # Tab3 - About
tabControl.add(tab3, text='About')
tabControl.pack(expand=1, fill="both")

top.iconbitmap('Assets/YoutubeDownloader.ico')  # Window Title Icon
top.title("FYIT Download Manager :")  # Title Label
top.geometry("800x500+100+100")  # Screen Size

photo = PhotoImage(file="Assets/youtube_bg.png")  # Tab1 Background
w = Label(tab1, image=photo)
w.pack()

photo2 = PhotoImage(file="Assets/facebook_bg.png")  # Tab2 Background
w2 = Label(tab2, image=photo2)
w2.pack()

top.resizable(0, 0)     # Disable the Maximize & Minimize option

# -----------Close Function--------------------------------------------------

def close_btn():
    if messagebox.askyesno("FYIT..", "Are you Sure you want to exit") is True:
        top.withdraw()
        top.destroy()
        top.exit()

# ---------------------------------------------------------------------------

var1 = StringVar()
# Entry Widget for Youtube Downloader Tab
url = Entry(tab1, textvariable=var1, font=large_font, fg='darkblue')
url.place(x=70, y=117, width=500, height=30)

# Entry Widget for Facebook Downloader Tab
url_fb = Entry(tab2, textvariable=var1, font=large_font, fg='darkblue')
url_fb.place(x=70, y=117, width=500, height=30)

# Quit_button placed in tab1
quit_button = Button(tab1,
                         text="Quit",
                         relief=SOLID,
                         font=SubFont,
                         cursor='hand2',
                         command=close_btn)
quit_button.place(x=60, y=400, width=200, height=30)

# ----------------------- Auto URL Paste Functions --------------------------
temp = clipboard.paste()
url.insert('end', temp)
url_fb.insert('end', temp)
var1.set(temp)

def paste():  # Paste text from Clipboard - Function
    variable = clipboard.paste()
    url.insert('end', variable)
    url_fb.insert('end', variable)
    var1.set(variable)

def e1_delete():  # Clear text from Entry - Function
    url.delete(0, 'end')
    url_fb.delete(0, 'end')

def toggle(tog=[0]):  # Toggle Button for clear and paste
    tog[0] = not tog[0]
    if tog[0]:
        t_btn.config(text='Paste')
        t_btn1.config(text='Paste')
        e1_delete()
    else:
        t_btn.config(text='Clear')
        t_btn1.config(text='Clear')
        paste()

# tab1 clear & paste
t_btn1 = ttk.Button(tab1, text="Clear", cursor='hand2', style='my.TButton', command=toggle)
t_btn1.place(x=571, y=118, width=45, height=29)

# tab2 clear & paste
t_btn = ttk.Button(tab2, text="Clear", cursor='hand2', style='my.TButton', command=toggle)
t_btn.place(x=571, y=118, width=45, height=29)

# ------------------ Fetching Part from Youtube ----------------------------

new = 1
url1 = "https://bit.ly/site-fyit"
res_url = "https://github.com/DeepakChakravarthy/YoutubeDownloader-FYI/releases/download/V3.0/FYI.DOWNLOAD.EXE"

def openweb():      # Software Update response
	webbrowser.open(url1, new=new)

def update_check():     # AutoCheck Software Update and Sub Screen
    
    response = r.get(res_url)
    
    response.raise_for_status()
    if messagebox.askyesno("Software Update",
            "New Update is Availabe, Click yes to install.") is True:
        openweb()

def get_fetch():
    resolution = Rvideo.get()
    Select = A_Video.get()
    Selection = A_Audio.get()

    try:
        update_check()
    except r.exceptions.HTTPError as err:
	    messagebox.showinfo("FYIT", "Select Location to Save the File.")
    except r.ConnectionError as e:
        messagebox.showinfo("Connection Error", "Check the Internet Connection")

    try:
        if var1 is None:
            print("error")
        dirname = filedialog.askdirectory(parent=tab1,
                                          initialdir="/",
                                          title='Please select a directory:')
        if dirname:
            try:
                # Youtube Video with Resolution
                if resolution <= 3:
                    yt = pytube.YouTube(var1.get())
                    if resolution == 1:
                        progress_bar = ttk.Progressbar(tab1,
                                                       orient='horizontal',
                                                       length=500,
                                                       mode='determinate')
                        progress_bar.place(x=70, y=160)
                        progress_bar.start()
                        messagebox.showinfo(
                            "Download",
                            "Downloading.. Please Wait for a Minute.")
                        video = yt.streams.get_by_itag(22)
                        video.download(dirname)
                        messagebox.showinfo("Downloaded. ", "Thank You..")
                    elif resolution == 2:
                        progress_bar = ttk.Progressbar(tab1,
                                                       orient='horizontal',
                                                       length=500,
                                                       mode='determinate')
                        progress_bar.place(x=70, y=160)
                        progress_bar.start()
                        messagebox.showinfo("Download", "Downloading...")
                        video = yt.streams.first()
                        video.download(dirname)
                        messagebox.showinfo("Downloaded. ", "Thank You..")
                    elif resolution == 3:
                        progress_bar = ttk.Progressbar(tab1,
                                                       orient='horizontal',
                                                       length=500,
                                                       mode='determinate')
                        progress_bar.place(x=70, y=160)
                        progress_bar.start()
                        messagebox.showinfo("Download", "Downloading...")
                        video = yt.streams.get_by_itag(36)
                        video.download(dirname)
                        messagebox.showinfo("Downloaded. ", "Thank You..")

                # Download Playlist
                if Select == 1:
                    playlist = pytube.Playlist(var1.get())
                    progress_bar = ttk.Progressbar(tab1,
                                                   orient='horizontal',
                                                   length=500,
                                                   mode='determinate')
                    progress_bar.place(x=70, y=160)
                    progress_bar.start()
                    playlist.populate_video_urls()  # To load bulk list
                    messagebox.showinfo("Download", "Downloading...")
                    playlist.download_all(dirname)
                    messagebox.showinfo("Downloaded. ", "Thank You..")

                # Audio files
                if Selection <= 2:
                    link = YouTube(var1.get())
                    format_a = link.streams.filter(only_audio=True).all()
                    if Selection == 1:  # mp4
                        progress_bar = ttk.Progressbar(tab1,
                                                       orient='horizontal',
                                                       length=500,
                                                       mode='determinate')
                        progress_bar.place(x=70, y=160)
                        progress_bar.start()
                        messagebox.showinfo("Download", "Downloading...")
                        format_a[0].download(dirname)
                        messagebox.showinfo("Downloaded. ", "Thank You..")
                    elif Selection == 2:  # webm
                        progress_bar = ttk.Progressbar(tab1,
                                                       orient='horizontal',
                                                       length=500,
                                                       mode='determinate')
                        progress_bar.place(x=70, y=160)
                        progress_bar.start()
                        messagebox.showinfo("Download", "Downloading...")
                        format_a[1].download(dirname)
                        messagebox.showinfo("Downloaded. ", "Thank You..")
                messagebox.showinfo("FYIT", "Please Select Valid option")
            except Exception as a:
                messagebox.showwarning(" FYIT.. ", "Failed")
        else:
            messagebox.showwarning(" FYIT. ", "Cancelled")
    except Exception as a:
        messagebox.showwarning(" FYIT. ", "Cancelled")
        sys.exit()

downbtn = Button(tab1,
                 text="Download",
                 relief=SOLID,
                 font=SubFont,
                 command=get_fetch,
                 cursor='hand2' )
downbtn.place(x=500, y=400, width=200, height=30)   # Download button Tab1
yt_label = Label(tab1,
                 text="Enter the Link to Download",
                 font=large_font,
                 bg="#0B1D6F",
                 fg="White")
yt_label.place(x=65, y=65)                          # Label placed in Tab1

# -----------------Radio button for video resolution-------------------------

Rvideo = IntVar()       # Variable Daclaraton for Youtube Video (options)
resolution_label = Label(tab1,
                         text="Video Resolution:",
                         font=DisFont,
                         bg="#0B1D6F",
                         fg="white")
resolution_label.place(x=75, y=200)
R1 = Radiobutton(tab1,
                 text="Mp4 720",
                 cursor='hand2',
                 font=SubFont,
                 variable=Rvideo,
                 value=1,
                 fg=col_show,
                 bg=yt_col,
                 activebackground=yt_col,
                 activeforeground=col_select,
                 selectcolor=yt_col)
R1.place(x=80, y=240)
R2 = Radiobutton(tab1,
                 text="Mp4 360px",
                 cursor='hand2',
                 font=SubFont,
                 variable=Rvideo,
                 value=2,
                 fg=col_show,
                 bg=yt_col,
                 activebackground=yt_col,
                 activeforeground=col_select,
                 selectcolor=yt_col)
R2.place(x=80, y=280)
R3 = Radiobutton(tab1,
                 text="3gp 144px",
                 cursor='hand2',
                 font=SubFont,
                 variable=Rvideo,
                 value=3,
                 fg=col_show,
                 bg=yt_col,
                 activebackground=yt_col,
                 activeforeground=col_select,
                 selectcolor=yt_col)
R3.place(x=80, y=320)

# -----------------Radio button for video to Audio----------------------------

A_Audio = IntVar()      # Variable Daclaraton for Youtube Audio (options)
format_label = Label(tab1,
                     text="Only Audio:",
                     font=DisFont,
                     bg="#0B1D6F",
                     fg="white")
format_label.place(x=260, y=200)
R4 = Radiobutton(tab1,
                 text="Mp4 Audio",
                 cursor='hand2',
                 font=SubFont,
                 variable=A_Audio,
                 value=1,
                 fg=col_show,
                 bg=yt_col,
                 activebackground=yt_col,
                 activeforeground=col_select,
                 selectcolor=yt_col)
R4.place(x=265, y=240)
R5 = Radiobutton(tab1,
                 text="webm",
                 cursor='hand2',
                 font=SubFont,
                 variable=A_Audio,
                 value=2,
                 fg=col_show,
                 bg=yt_col,
                 activebackground=yt_col,
                 activeforeground=col_select,
                 selectcolor=yt_col)
R5.place(x=265, y=280)

# -----------------Radio button for Playlist video --------------------------

A_Video = IntVar()      # Variable Daclaraton for Youtube Playlist (options)
list_label = Label(tab1,
                   text="Download Playlist:",
                   font=DisFont,
                   bg="#0B1D6F",
                   fg="white")
list_label.place(x=415, y=200)
R7 = Radiobutton(tab1,
                 text="High (Default)",
                 cursor='hand2',
                 font=SubFont,
                 variable=A_Video,
                 value=1,
                 fg=col_show,
                 bg=yt_col,
                 activebackground=yt_col,
                 activeforeground=col_select,
                 selectcolor=yt_col)
R7.place(x=420, y=240)

# ======================= Facebook Control ==================================

quit_button = Button(tab2,
                         text="Quit",
                         relief=SOLID,
                         font=SubFont,
                         cursor='hand2',
                         command=close_btn)
quit_button.place(x=60, y=400, width=200, height=30)    # Quit_button placed in tab2

fb_label = Label(tab2,
                 text="Enter the Link to Download",
                 font=large_font,
                 bg="#075C90",
                 fg="White")
fb_label.place(x=65, y=65)      # Label placed in Tab2

def FacebookDownload():         # FacebookDownloader Main Fuction
    try:
        update_check()
    except r.exceptions.HTTPError as err:
	    messagebox.showinfo("FYIT", "Select Location to Download.")
    except r.ConnectionError as e:
        messagebox.showinfo("Connection Error",
                            "Check the Internet Connection")

    try:
        html = r.get(var1.get())
        dirname = filedialog.askdirectory(parent=tab2,
                                          initialdir="/",
                                          title='Please select a directory:')
        if dirname:
            hdvideo_url = re.search('hd_src:"(.+?)"', html.text)[1]
        try:
            hd_url = hdvideo_url.replace('hd_src:"', '')

The variable hdvideo_url does not seem to be defined in case dirname on line 444 is False. Are you sure this can never be the case?

            messagebox.showinfo("FYIT", "[+] Video Started Downloading")
            progress_bar = ttk.Progressbar(tab2,
                                           orient='horizontal',
                                           length=500,
                                           mode='determinate')
            progress_bar.place(x=60, y=180)
            progress_bar.start()
            wget.download(hd_url, dirname)
            ERASE_LINE = '\x1b[2K'
            sys.stdout.write(ERASE_LINE)
            messagebox.showinfo("FYIT", "Video downloaded")
            progress_bar.stop()
        except r.ConnectionError as e:
            messagebox.showerror("FYIT", "ConnectionError")
        except r.Timeout as e:
            messagebox.showinfo("FYIT", "Tomeout")
        except r.RequestException as e:
            messagebox.showerror("FYIT", "General Error or Invalid URL")
        except (KeyboardInterrupt, SystemExit):
            messagebox.showinfo("FYIT", "SystemExit")
            sys.exit()
        except TypeError:
            messagebox.showerror("FYIT", "Video May Private or InvalidURL")
    except Exception as e:
        messagebox.showwarning("FYIT", "Cancelled")

downbtnfb = Button(tab2,
                   text="Download",
                   relief=SOLID,
                   font=SubFont,
                   command=FacebookDownload,
                   cursor='hand2')
downbtnfb.place(x=500, y=400, width=200, height=30)    # Download placed in Tab2

# ======================= About Tab Control ===================================

def check_it(hyper_link):       # About tab Redirect Link (Frame1)
    webbrowser.open_new(hyper_link)


def find_update():              # Check Software Update - 2
    try:
        update_check()
        messagebox.showinfo("FYIT", "Thank You.")
    except r.ConnectionError as e:
        messagebox.showinfo("Connection Error",
                            "Check the Internet Connection")
    except r.exceptions.HTTPError as err:
        messagebox.showinfo("FYIT","No Update yet..")

tab3_style = ttk.Style()

frame1 = tk.Frame(master=tab3, width=260)           # Tab3_Frame1
frame1.pack(fill=tk.BOTH, side=tk.LEFT, expand=True)

imge = PhotoImage(file='Assets/side_bar.png')
pht = Label(frame1, image=imge)
pht.pack()

link_1 = Label(frame1,
                text="How to use..?",
                font=tab_font,
                fg="blue",
                bg="#90B3DD",
                activeforeground="red",
                activebackground="green",
                cursor="hand2",
                underline=0)
link_1.place(x=20, y=260)
link_1.bind("<Button-1>",
            lambda e: check_it())

link_2 = Label(frame1,
                text="Help..!",
                font=tab_font,
                fg="blue",
                bg="#90B3DD",
                activeforeground="red",
                activebackground="green",
                cursor="hand2",
                underline=0)
link_2.place(x=20, y=300)
link_2.bind("<Button-1>",
            lambda e: check_it("http://bit.ly/site-fyit"))

link_3 = Label(frame1,
               text="Contact us..",
               font=tab_font,
               fg="blue",
               bg="#90B3DD",
               activeforeground="red",
               activebackground="green",
               cursor="hand2",
               underline=0)
link_3.place(x=20, y=340)
link_3.bind("<Button-1>",
            lambda e: check_it("https://gitter.im/FYIT-DOWNLOADER/DEV-FYI?utm_source=share-link&utm_medium=link&utm_campaign=share-link"))

link_4 = Label(frame1,
                text="Visit us..",
                font=tab_font,
                fg="blue",
                bg="#90B3DD",
                activeforeground="red",
                activebackground="green",
                cursor="hand2",
                underline=0)
link_4.place(x=20, y=380)
link_4.bind("<Button-1>",
            lambda e: check_it("http://bit.ly/site-fyit"))

link_5 = Button(frame1,
                text="Check Update..",
                font=tab_font,
                command=find_update,
                fg="blue",
                bg="#90B3DD",
                activeforeground="red",
                activebackground="#90B3DD",
                cursor="hand2",
                relief=FLAT,
                underline=0)
link_5.place(x=20, y=420)
link_5.config(highlightthickness=0)

frame2 = HtmlFrame(master=tab3, width=640, horizontal_scrollbar="auto")     # Tab3_Frame2
frame2.pack(fill=tk.BOTH, side=tk.LEFT, expand=True)

frame2.set_content(open("Assets/help.html", "r").read())

import AddOn
top.mainloop()


# ----------------------------- Close Function -------------------------------

def on_close():
    AddOn.window.destroy()
    top.destroy()
    sys.exit()

top.protocol("WM_DELETE_WINDOW", on_close)
# ----call for other Screens------
def minimize():
    top.withdraw()
# ----------------------------------------------------------------------------