#! /usr/bin/env python ''' This file contains all gui classes MainFrame Custom wx.ListCtrl class OptionsFrame ConnectionPanel AudioPanel videoPanel DownloadPanel SubtitlesPanel GeneralPanel OtherPanel ''' import os import wx from wx.lib.pubsub import setuparg1 from wx.lib.pubsub import pub as Publisher from .version import __version__ from .UpdateThread import UpdateThread from .DownloadThread import DownloadManager from .OptionsHandler import OptionsHandler from .YoutubeDLInterpreter import YoutubeDLInterpreter from .SignalHandler import DownloadHandler if os.name == 'nt': YOUTUBE_DL_FILENAME = 'youtube-dl.exe' else: YOUTUBE_DL_FILENAME = 'youtube-dl' TITLE = 'Youtube-dlG' AUDIOFORMATS = ["mp3", "wav", "aac", "m4a"] VIDEOFORMATS = ["highest available", "mp4 [1280x720]", "mp4 [640x360]", "webm [640x360]", "flv [400x240]", "3gp [320x240]", "mp4 1080p(DASH)", "mp4 720p(DASH)", "mp4 480p(DASH)", "mp4 360p(DASH)"] LANGUAGES = ["English", "Greek", "Portuguese", "French", "Italian", "Russian", "Spanish", "German"] class MainFrame(wx.Frame): def __init__(self, parent=None, id=-1): wx.Frame.__init__(self, parent, id, TITLE+' '+__version__, size=(600, 410), style = wx.DEFAULT_FRAME_STYLE ^ wx.RESIZE_BORDER) # set sizers for status box (Windows & Linux) if os.name == 'nt': statusListSizer = (580, 165) statusBarSizer = (15, 365) else: statusListSizer = (580, 195) statusBarSizer = (15, 390) # create panel, trackList, statusBox using global statusBoxSizer self.panel = wx.Panel(self) wx.StaticText(self.panel, -1, "URLs", (15, 10)) self.trackList = wx.TextCtrl(self.panel, -1, pos=(10, 25), size=(580, 110), style = wx.TE_MULTILINE | wx.TE_DONTWRAP) self.statusList = ListCtrl(self.panel, -1, pos=(10, 190), size=statusListSizer, style = wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES) self.statusBar = wx.StaticText(self.panel, -1, 'Author: Sotiris Papadopoulos', pos=statusBarSizer) # create buttons self.downloadButton = wx.Button(self.panel, label="Download", pos=(100, 145), size=(-1, 30)) self.updateButton = wx.Button(self.panel, label="Update", pos=(250, 145), size=(-1, 30)) self.optionsButton = wx.Button(self.panel, label="Options", pos=(390, 145), size=(-1, 30)) # bind events self.Bind(wx.EVT_BUTTON, self.OnDownload, self.downloadButton) self.Bind(wx.EVT_BUTTON, self.OnUpdate, self.updateButton) self.Bind(wx.EVT_BUTTON, self.OnOptions, self.optionsButton) self.Bind(wx.EVT_TEXT, self.OnTrackListChange, self.trackList) self.Bind(wx.EVT_CLOSE, self.OnClose) # set app icon icon = wx.Icon('../icons/ytube.png', wx.BITMAP_TYPE_ICO) self.SetIcon(icon) # set publisher for update thread Publisher.subscribe(self.update_handler, "update") # set publisher for download thread Publisher.subscribe(self.download_handler, "download") # init Options and DownloadHandler objects self.optionsList = OptionsHandler() self.downloadHandler = None # init some thread variables self.downloadThread = None self.updateThread = None # init urlList for evt_text on self.trackList self.urlList = [] # check & update libraries (youtube-dl) self.check_if_youtube_dl_exist() if (self.optionsList.autoUpdate): self.status_bar_write("Auto update enable") self.update_youtube_dl() def check_if_youtube_dl_exist(self): if not os.path.exists(YOUTUBE_DL_FILENAME): self.status_bar_write("Youtube-dl is missing, trying to download it...") self.update_youtube_dl() def update_youtube_dl(self): self.downloadButton.Disable() self.updateThread = UpdateThread(YOUTUBE_DL_FILENAME) def status_bar_write(self, msg): self.statusBar.SetLabel(msg) def download_handler(self, msg): self.downloadHandler.handle(msg) if self.downloadHandler._has_closed(): self.status_bar_write('Stoping downloads') if self.downloadHandler._has_finished(): self.finished_popup() self.status_bar_write('Done') self.downloadButton.SetLabel('Download') self.updateButton.Enable() self.downloadThread = None self.urlList = [] self.downloadHandler = None def update_handler(self, msg): if msg.data == 'finish': self.downloadButton.Enable() self.updateThread = None else: self.status_bar_write(msg.data) def stop_download(self): self.downloadThread.close() self.downloadThread.join() self.downloadThread = None self.urlList = [] def start_download(self, trackList): self.statusList._clear_list() for url in trackList: if url != '': self.urlList.append(url) self.statusList._add_item(url) if not self.statusList._is_empty(): options = YoutubeDLInterpreter(self.optionsList, YOUTUBE_DL_FILENAME).get_options() self.status_bar_write('Download started') self.downloadThread = DownloadManager(options, self.statusList._get_items()) self.downloadHandler = DownloadHandler(self.statusList) self.downloadButton.SetLabel('Stop') self.updateButton.Disable() else: self.no_url_popup() def save_options(self): self.optionsList.save_to_file() def finished_popup(self): wx.MessageBox('Downloads completed.', 'Info', wx.OK | wx.ICON_INFORMATION) def no_url_popup(self): wx.MessageBox('You need to provide at least one url.', 'Error', wx.OK | wx.ICON_EXCLAMATION) def OnTrackListChange(self, event): if self.downloadThread != None: ''' Get current url list from trackList textCtrl ''' curList = self.trackList.GetValue().split('\n') ''' For each url in current url list ''' for url in curList: ''' If url is not in self.urlList (original downloads list) and url is not empty ''' if url not in self.urlList and url != '': ''' Add url into original download list ''' self.urlList.append(url) ''' Add url into statusList ''' index = self.statusList._add_item(url) ''' Retrieve last item as {url:url, index:indexNo} ''' item = self.statusList._get_last_item() ''' Pass that item into downloadThread ''' self.downloadThread.add_download_item(item) def OnDownload(self, event): if self.downloadThread != None: self.stop_download() else: self.start_download(self.trackList.GetValue().split('\n')) def OnUpdate(self, event): if (self.downloadThread == None and self.updateThread == None): self.status_bar_write("Updating youtube-dl...") self.update_youtube_dl() def OnOptions(self, event): optionsFrame = OptionsFrame(self.optionsList, self) optionsFrame.Show() def OnClose(self, event): self.save_options() self.Destroy() class ListCtrl(wx.ListCtrl): ''' Custom ListCtrl class ''' def __init__(self, parent, id, pos, size, style): wx.ListCtrl.__init__(self, parent, id, pos, size, style) self.InsertColumn(0, 'URL', width=150) self.InsertColumn(1, 'Size', width=90) self.InsertColumn(2, 'Percent', width=80) self.InsertColumn(3, 'ETA', width=50) self.InsertColumn(4, 'Speed', width=90) self.InsertColumn(5, 'Status', width=150) self.ListIndex = 0 ''' Add single item on list ''' def _add_item(self, item): self.InsertStringItem(self.ListIndex, item) self.ListIndex += 1 return self.ListIndex ''' Write data on index, column ''' def _write_data(self, index, column, data): self.SetStringItem(index, column, data) ''' Clear list and set index to 0''' def _clear_list(self): self.DeleteAllItems() self.ListIndex = 0 ''' Return True if list is empty ''' def _is_empty(self): if self.ListIndex == 0: return True else: return False ''' Get last item inserted, Returns dictionary ''' def _get_last_item(self): data = {} last_item = self.GetItem(itemId=self.ListIndex-1, col=0) data['url'] = last_item.GetText() data['index'] = self.ListIndex-1 return data ''' Retrieve all items [start, self.ListIndex), Returns list ''' def _get_items(self, start=0): items = [] for row in range(start, self.ListIndex): item = self.GetItem(itemId=row, col=0) data = {} data['url'] = item.GetText() data['index'] = row items.append(data) return items class ConnectionPanel(wx.Panel): def __init__(self, parent, optList): self.optList = optList wx.Panel.__init__(self, parent) wx.StaticText(self, -1, 'User Agent', (15, 10)) self.userAgentBox = wx.TextCtrl(self, -1, pos=(10, 30), size=(230, -1)) wx.StaticText(self, -1, 'Referer', (270, 10)) self.refererBox = wx.TextCtrl(self, -1, pos=(265, 30), size=(230, -1)) wx.StaticText(self, -1, 'Username', (15, 60)) self.usernameBox = wx.TextCtrl(self, -1, pos=(10, 80), size=(230, -1)) wx.StaticText(self, -1, 'Password', (270, 60)) self.passwordBox = wx.TextCtrl(self, -1, pos=(265, 80), size=(230, -1), style = wx.TE_PASSWORD) wx.StaticText(self, -1, 'Proxy', (15, 110)) self.proxyBox = wx.TextCtrl(self, -1, pos=(10, 130), size=(350, -1)) def load_options(self): self.userAgentBox.SetValue(self.optList.userAgent) self.refererBox.SetValue(self.optList.referer) self.usernameBox.SetValue(self.optList.username) self.proxyBox.SetValue(self.optList.proxy) def save_options(self): self.optList.userAgent = self.userAgentBox.GetValue() self.optList.referer = self.refererBox.GetValue() self.optList.username = self.usernameBox.GetValue() self.optList.proxy = self.proxyBox.GetValue() class AudioPanel(wx.Panel): def __init__(self, parent, optList): self.optList = optList wx.Panel.__init__(self, parent) self.toAudioChk = wx.CheckBox(self, -1, 'Convert to Audio', (10, 10)) self.keepVideoChk = wx.CheckBox(self, -1, 'Keep Video', (30, 40)) wx.StaticText(self, -1, 'Audio Format', (35, 80)) self.audioFormatCombo = wx.ComboBox(self, choices=AUDIOFORMATS, pos=(30, 100), size=(95, -1)) wx.StaticText(self, -1, "Audio Quality 0 (best) 9 (worst)", (35, 130)) self.audioQualitySpnr = wx.SpinCtrl(self, -1, "", (30, 150)) self.audioQualitySpnr.SetRange(0, 9) self.Bind(wx.EVT_CHECKBOX, self.OnAudioCheck, self.toAudioChk) def OnAudioCheck(self, event): if (self.toAudioChk.GetValue()): self.keepVideoChk.Enable() self.audioFormatCombo.Enable() self.audioQualitySpnr.Enable() else: self.keepVideoChk.Disable() self.audioFormatCombo.Disable() self.audioQualitySpnr.Disable() def load_options(self): self.toAudioChk.SetValue(self.optList.toAudio) self.keepVideoChk.SetValue(self.optList.keepVideo) self.audioFormatCombo.SetValue(self.optList.audioFormat) self.audioQualitySpnr.SetValue(self.optList.audioQuality) if (self.optList.toAudio == False): self.keepVideoChk.Disable() self.audioFormatCombo.Disable() self.audioQualitySpnr.Disable() def save_options(self): self.optList.toAudio = self.toAudioChk.GetValue() self.optList.keepVideo = self.keepVideoChk.GetValue() self.optList.audioFormat = self.audioFormatCombo.GetValue() self.optList.audioQuality = self.audioQualitySpnr.GetValue() class VideoPanel(wx.Panel): def __init__(self, parent, optList): self.optList = optList wx.Panel.__init__(self, parent) wx.StaticText(self, -1, 'Video Format', (15, 10)) self.videoFormatCombo = wx.ComboBox(self, choices=VIDEOFORMATS, pos=(10, 30), size=(160, 30)) wx.StaticText(self, -1, 'Playlist Options', (300, 30)) wx.StaticText(self, -1, 'Start', (250, 60)) self.startBox = wx.TextCtrl(self, -1, pos=(320, 55), size=(50, -1)) wx.StaticText(self, -1, 'Stop', (250, 100)) self.stopBox = wx.TextCtrl(self, -1, pos=(320, 95), size=(50, -1)) wx.StaticText(self, -1, 'Max DLs', (250, 140)) self.maxBox = wx.TextCtrl(self, -1, pos=(320, 135), size=(50, -1)) def load_options(self): self.videoFormatCombo.SetValue(self.optList.videoFormat) self.startBox.SetValue(self.optList.startTrack) self.stopBox.SetValue(self.optList.endTrack) self.maxBox.SetValue(self.optList.maxDownloads) def save_options(self): self.optList.videoFormat = self.videoFormatCombo.GetValue() self.optList.startTrack = self.startBox.GetValue() self.optList.endTrack = self.stopBox.GetValue() self.optList.maxDownloads = self.maxBox.GetValue() class DownloadPanel(wx.Panel): def __init__(self, parent, optList): self.optList = optList wx.Panel.__init__(self, parent) wx.StaticText(self, -1, 'Rate Limit (e.g. 50k or 44.6m)', (250, 15)) self.limitBox = wx.TextCtrl(self, -1, pos=(245, 35), size=(80, -1)) wx.StaticText(self, -1, 'Retries', (15, 15)) self.retriesBox = wx.TextCtrl(self, -1, pos=(10, 35), size=(50, -1)) self.writeDescriptionChk = wx.CheckBox(self, -1, 'Write description to file', (10, 60)) self.writeInfoChk = wx.CheckBox(self, -1, 'Write info to (.json) file', (10, 85)) self.writeThumbnailChk = wx.CheckBox(self, -1, 'Write thumbnail to disk', (10, 110)) self.ignoreErrorsChk = wx.CheckBox(self, -1, 'Ignore Errors', (10, 135)) wx.StaticText(self, -1, 'Min Filesize (e.g. 50k or 44.6m)', (250, 65)) self.minFilesizeBox = wx.TextCtrl(self, -1, pos=(245, 85), size=(80, -1)) wx.StaticText(self, -1, 'Max Filesize (e.g. 50k or 44.6m)', (250, 115)) self.maxFilesizeBox = wx.TextCtrl(self, -1, pos=(245, 135), size=(80, -1)) def load_options(self): self.limitBox.SetValue(self.optList.rateLimit) self.retriesBox.SetValue(self.optList.retries) self.writeDescriptionChk.SetValue(self.optList.writeDescription) self.writeInfoChk.SetValue(self.optList.writeInfo) self.writeThumbnailChk.SetValue(self.optList.writeThumbnail) self.ignoreErrorsChk.SetValue(self.optList.ignoreErrors) self.minFilesizeBox.SetValue(self.optList.minFileSize) self.maxFilesizeBox.SetValue(self.optList.maxFileSize) def save_options(self): self.optList.rateLimit = self.limitBox.GetValue() self.optList.retries = self.retriesBox.GetValue() self.optList.writeDescription = self.writeDescriptionChk.GetValue() self.optList.writeInfo = self.writeInfoChk.GetValue() self.optList.writeThumbnail = self.writeThumbnailChk.GetValue() self.optList.ignoreErrors = self.ignoreErrorsChk.GetValue() self.optList.minFileSize = self.minFilesizeBox.GetValue() self.optList.maxFileSize = self.maxFilesizeBox.GetValue() class SubtitlesPanel(wx.Panel): def __init__(self, parent, optList): self.optList = optList wx.Panel.__init__(self, parent) self.writeSubsChk = wx.CheckBox(self, -1, 'Write subtitle file', (10, 10)) self.writeAllSubsChk = wx.CheckBox(self, -1, 'Download all available subtitles', (10, 40)) self.writeAutoSubsChk = wx.CheckBox(self, -1, 'Write automatic subtitle file (YOUTUBE ONLY)', (10, 70)) wx.StaticText(self, -1, 'Subtitles Language', (15, 105)) self.subsLangCombo = wx.ComboBox(self, choices=LANGUAGES, pos=(10, 125), size=(140, 30)) self.Bind(wx.EVT_CHECKBOX, self.OnWriteSubsChk, self.writeSubsChk) self.Bind(wx.EVT_CHECKBOX, self.OnWriteAllSubsChk, self.writeAllSubsChk) self.Bind(wx.EVT_CHECKBOX, self.OnWriteAutoSubsChk, self.writeAutoSubsChk) def OnWriteAutoSubsChk(self, event): if (self.writeAutoSubsChk.GetValue()): self.writeAllSubsChk.Disable() self.writeSubsChk.Disable() self.subsLangCombo.Disable() else: self.writeAllSubsChk.Enable() self.writeSubsChk.Enable() self.subsLangCombo.Enable() def OnWriteSubsChk(self, event): if (self.writeSubsChk.GetValue()): self.writeAllSubsChk.Disable() self.writeAutoSubsChk.Disable() else: self.writeAllSubsChk.Enable() self.writeAutoSubsChk.Enable() def OnWriteAllSubsChk(self, event): if (self.writeAllSubsChk.GetValue()): self.writeSubsChk.Disable() self.subsLangCombo.Disable() self.writeAutoSubsChk.Disable() else: self.writeSubsChk.Enable() self.subsLangCombo.Enable() self.writeAutoSubsChk.Enable() def load_options(self): self.writeSubsChk.Enable() self.subsLangCombo.Enable() self.writeAllSubsChk.Enable() self.writeAutoSubsChk.Enable() self.writeSubsChk.SetValue(self.optList.writeSubs) self.writeAllSubsChk.SetValue(self.optList.writeAllSubs) self.subsLangCombo.SetValue(self.optList.subsLang) self.writeAutoSubsChk.SetValue(self.optList.writeAutoSubs) if (self.writeSubsChk.GetValue()): self.writeAllSubsChk.Disable() self.writeAllSubsChk.SetValue(False) self.writeAutoSubsChk.Disable() self.writeAutoSubsChk.SetValue(False) if (self.writeAllSubsChk.GetValue()): self.writeSubsChk.Disable() self.writeSubsChk.SetValue(False) self.subsLangCombo.Disable() self.writeAutoSubsChk.Disable() self.writeAutoSubsChk.SetValue(False) if (self.writeAutoSubsChk.GetValue()): self.writeAllSubsChk.Disable() self.writeAllSubsChk.SetValue(False) self.writeSubsChk.Disable() self.writeSubsChk.SetValue(False) self.subsLangCombo.Disable() def save_options(self): self.optList.writeSubs = self.writeSubsChk.GetValue() self.optList.writeAllSubs = self.writeAllSubsChk.GetValue() self.optList.subsLang = self.subsLangCombo.GetValue() self.optList.writeAutoSubs = self.writeAutoSubsChk.GetValue() class GeneralPanel(wx.Panel): def __init__(self, parent, optList, controlParent): self.optList = optList self.parent = controlParent wx.Panel.__init__(self, parent) wx.StaticText(self, -1, "Save Path", (15, 10)) self.savePathBox = wx.TextCtrl(self, -1, pos=(10, 30), size=(350, -1)) self.idAsNameChk = wx.CheckBox(self, -1, 'ID as Name', (10, 70)) self.autoUpdateChk = wx.CheckBox(self, -1, 'Auto Update', (160, 70)) self.aboutButton = wx.Button(self, label="About", pos=(380, 80), size=(100, 40)) self.openButton = wx.Button(self, label="Open", pos=(380, 20), size=(100, 40)) self.resetButton = wx.Button(self, label="Reset", pos=(380, 140), size=(100, 40)) wx.StaticText(self, -1, "Settings: " + self.optList.settings_abs_path, (20, 155)) self.Bind(wx.EVT_BUTTON, self.OnAbout, self.aboutButton) self.Bind(wx.EVT_BUTTON, self.OnOpen, self.openButton) self.Bind(wx.EVT_BUTTON, self.OnReset, self.resetButton) def OnReset(self, event): self.parent.reset() def OnOpen(self, event): dlg = wx.DirDialog(None, "Choose directory") if dlg.ShowModal() == wx.ID_OK: self.savePathBox.SetValue(dlg.GetPath()) dlg.Destroy() def OnAbout(self, event): description = '''A cross platform front-end GUI of the popular youtube-dl written in Python.''' license = '''This is free and unencumbered software released into the public domain. Anyone is free to copy, modify, publish, use, compile, sell, or distribute this software, either in source code form or as a compiled binary, for any purpose, commercial or non-commercial, and by any means. In jurisdictions that recognize copyright laws, the author or authors of this software dedicate any and all copyright interest in the software to the public domain. We make this dedication for the benefit of the public at large and to the detriment of our heirs and successors. We intend this dedication to be an overt act of relinquishment in perpetuity of all present and future rights to this software under copyright law. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. For more information, please refer to ''' info = wx.AboutDialogInfo() info.SetIcon(wx.Icon('../icons/ytube.png', wx.BITMAP_TYPE_ICO)) info.SetName(TITLE) info.SetVersion(__version__) info.SetDescription(description) info.SetWebSite('https://github.com/MrS0m30n3/youtube-dl-gui') info.SetLicense(license) info.AddDeveloper('Sotiris Papadopoulos') wx.AboutBox(info) def load_options(self): self.savePathBox.SetValue(self.optList.savePath) self.idAsNameChk.SetValue(self.optList.idAsName) self.autoUpdateChk.SetValue(self.optList.autoUpdate) def save_options(self): self.optList.savePath = self.savePathBox.GetValue() self.optList.idAsName = self.idAsNameChk.GetValue() self.optList.autoUpdate = self.autoUpdateChk.GetValue() class OtherPanel(wx.Panel): def __init__(self, parent, optList): self.optList = optList wx.Panel.__init__(self, parent) wx.StaticText(self, -1, 'Command line arguments (e.g. --help)', (25, 20)) self.cmdArgsBox = wx.TextCtrl(self, -1, pos=(20, 40), size=(450, -1)) def load_options(self): self.cmdArgsBox.SetValue(self.optList.cmdArgs) def save_options(self): self.optList.cmdArgs = self.cmdArgsBox.GetValue() class OptionsFrame(wx.Frame): def __init__(self, optionsList, parent=None, id=-1): wx.Frame.__init__(self, parent, id, "Options", size=(540, 250)) self.optionsList = optionsList panel = wx.Panel(self) notebook = wx.Notebook(panel) self.generalTab = GeneralPanel(notebook, self.optionsList, self) self.audioTab = AudioPanel(notebook, self.optionsList) self.connectionTab = ConnectionPanel(notebook, self.optionsList) self.videoTab = VideoPanel(notebook, self.optionsList) self.downloadTab = DownloadPanel(notebook, self.optionsList) self.subtitlesTab = SubtitlesPanel(notebook, self.optionsList) self.otherTab = OtherPanel(notebook, self.optionsList) notebook.AddPage(self.generalTab, "General") notebook.AddPage(self.audioTab, "Audio") notebook.AddPage(self.videoTab, "Video") notebook.AddPage(self.subtitlesTab, "Subtitles") notebook.AddPage(self.downloadTab, "Download") notebook.AddPage(self.connectionTab, "Connection") notebook.AddPage(self.otherTab, "Commands") sizer = wx.BoxSizer() sizer.Add(notebook, 1, wx.EXPAND) panel.SetSizer(sizer) self.Bind(wx.EVT_CLOSE, self.OnClose) self.load_all_options() def OnClose(self, event): self.save_all_options() self.Destroy() def reset(self): self.optionsList.load_default() self.load_all_options() def load_all_options(self): self.generalTab.load_options() self.audioTab.load_options() self.connectionTab.load_options() self.videoTab.load_options() self.downloadTab.load_options() self.subtitlesTab.load_options() self.otherTab.load_options() def save_all_options(self): self.generalTab.save_options() self.audioTab.save_options() self.connectionTab.save_options() self.videoTab.save_options() self.downloadTab.save_options() self.subtitlesTab.save_options() self.otherTab.save_options()