#! /usr/bin/env python ''' This file contains all gui classes MainFrame Custom wx.ListCtrl class OptionsFrame GeneralPanel AudioPanel ConnectionPanel VideoPanel FilesystemPanel SubtitlesPanel OtherPanel UpdatePanel AuthenticationPanel PlaylistPanel ''' import wx from wx.lib.pubsub import setuparg1 from wx.lib.pubsub import pub as Publisher from wx.lib.mixins.listctrl import ListCtrlAutoWidthMixin from .version import __version__ from .UpdateThread import UpdateThread from .DownloadThread import DownloadManager from .OptionsHandler import OptionsHandler from .YoutubeDLInterpreter import YoutubeDLInterpreter from .SignalHandler import DownloadHandler from .LogManager import LogManager, LogGUI from .Utils import ( video_is_dash, have_dash_audio, get_os_type, file_exist, fix_path, get_abs_path, get_icon_path ) if get_os_type() == 'nt': YOUTUBE_DL_FILENAME = 'youtube-dl.exe' else: YOUTUBE_DL_FILENAME = 'youtube-dl' TITLE = 'Youtube-dlG' AUDIOFORMATS = ["mp3", "wav", "aac", "m4a"] VIDEOFORMATS = ["default", "mp4 [1280x720]", "mp4 [640x360]", "webm [640x360]", "flv [400x240]", "3gp [320x240]", "mp4 1080p(DASH)", "mp4 720p(DASH)", "mp4 480p(DASH)", "mp4 360p(DASH)"] DASH_AUDIO_FORMATS = ["NO SOUND", "DASH m4a audio 128k", "DASH webm audio 48k"] LANGUAGES = ["English", "Greek", "Portuguese", "French", "Italian", "Russian", "Spanish", "German"] ICON = get_icon_path(['ytube.png', 'icons'], __file__) class MainFrame(wx.Frame): def __init__(self, parent=None, id=-1): wx.Frame.__init__(self, parent, id, TITLE+' '+__version__, size=(600, 420)) # init gui self.InitGUI() # 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(ICON, 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.status_bar_write) self.downloadHandler = None # init log manager self.logManager = None if self.optionsList.enableLog: self.logManager = LogManager( self.optionsList.get_config_path(), self.optionsList.writeTimeToLog ) # 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 InitGUI(self): self.panel = wx.Panel(self) mainBoxSizer = wx.BoxSizer(wx.VERTICAL) urlTextBox = wx.BoxSizer(wx.HORIZONTAL) urlTextBox.Add(wx.StaticText(self.panel, label='URLs'), flag = wx.TOP, border=10) mainBoxSizer.Add(urlTextBox, flag = wx.LEFT, border=20) trckListBox = wx.BoxSizer(wx.HORIZONTAL) self.trackList = wx.TextCtrl(self.panel, size=(-1, 120), style = wx.TE_MULTILINE | wx.TE_DONTWRAP) trckListBox.Add(self.trackList, 1) mainBoxSizer.Add(trckListBox, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=20) buttonsBox = wx.BoxSizer(wx.HORIZONTAL) self.downloadButton = wx.Button(self.panel, label='Download', size=(90, 30)) buttonsBox.Add(self.downloadButton) self.updateButton = wx.Button(self.panel, label='Update', size=(90, 30)) buttonsBox.Add(self.updateButton, flag = wx.LEFT | wx.RIGHT, border=80) self.optionsButton = wx.Button(self.panel, label='Options', size=(90, 30)) buttonsBox.Add(self.optionsButton) mainBoxSizer.Add(buttonsBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.BOTTOM | wx.TOP, border=10) stListBox = wx.BoxSizer(wx.HORIZONTAL) self.statusList = ListCtrl(self.panel, style = wx.LC_REPORT | wx.LC_HRULES | wx.LC_VRULES) stListBox.Add(self.statusList, 1, flag = wx.EXPAND) mainBoxSizer.Add(stListBox, 1, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=20) stBarBox = wx.BoxSizer(wx.HORIZONTAL) self.statusBar = wx.StaticText(self.panel, label='Author: Sotiris Papadopoulos') stBarBox.Add(self.statusBar, flag = wx.TOP | wx.BOTTOM, border=5) mainBoxSizer.Add(stBarBox, flag = wx.LEFT, border=20) self.panel.SetSizer(mainBoxSizer) def check_if_youtube_dl_exist(self): path = fix_path(self.optionsList.updatePath)+YOUTUBE_DL_FILENAME if not file_exist(path): 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(self.optionsList.updatePath, YOUTUBE_DL_FILENAME) def status_bar_write(self, msg): self.statusBar.SetLabel(msg) def fin_task(self, msg): self.status_bar_write(msg) self.downloadButton.SetLabel('Download') self.updateButton.Enable() self.downloadThread.join() self.downloadThread = None self.downloadHandler = None self.urlList = [] self.finished_popup() 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(): if self.downloadHandler._has_error(): if self.logManager != None: msg = 'An error occured while downloading. See Options>Log.' else: msg = 'An error occured while downloading' else: msg = 'Done' self.fin_task(msg) def update_handler(self, msg): if msg.data == 'finish': self.downloadButton.Enable() self.updateThread.join() self.updateThread = None else: self.status_bar_write(msg.data) def stop_download(self): self.downloadThread.close() def load_tracklist(self, trackList): for url in trackList: if url != '': self.urlList.append(url) self.statusList._add_item(url) def start_download(self): self.statusList._clear_list() self.load_tracklist(self.trackList.GetValue().split('\n')) if not self.statusList._is_empty(): options = YoutubeDLInterpreter(self.optionsList, YOUTUBE_DL_FILENAME).get_options() self.downloadThread = DownloadManager( options, self.statusList._get_items(), self.optionsList.clearDashFiles, self.logManager ) self.downloadHandler = DownloadHandler(self.statusList) self.status_bar_write('Download started') 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 ''' 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() def OnUpdate(self, event): if (self.downloadThread == None and self.updateThread == None): self.update_youtube_dl() def OnOptions(self, event): optionsFrame = OptionsFrame(self.optionsList, parent=self, logger=self.logManager) optionsFrame.Show() def OnClose(self, event): self.save_options() self.Destroy() class ListCtrl(wx.ListCtrl, ListCtrlAutoWidthMixin): ''' Custom ListCtrl class ''' def __init__(self, parent=None, id=-1, pos=wx.DefaultPosition, size=wx.DefaultSize, style=0): wx.ListCtrl.__init__(self, parent, id, pos, size, style) ListCtrlAutoWidthMixin.__init__(self) self.InsertColumn(0, 'Video', width=150) self.InsertColumn(1, 'Size', width=80) self.InsertColumn(2, 'Percent', width=65) self.InsertColumn(3, 'ETA', width=45) self.InsertColumn(4, 'Speed', width=90) self.InsertColumn(5, 'Status', width=160) self.setResizeColumn(0) self.ListIndex = 0 ''' Add single item on list ''' def _add_item(self, item): self.InsertStringItem(self.ListIndex, item) self.ListIndex += 1 ''' 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): return self.ListIndex == 0 ''' 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 LogPanel(wx.Panel): size = '' path = '' win_box_border = 0 def __init__(self, parent, optList, log): wx.Panel.__init__(self, parent) self.SetBoxBorder() self.optList = optList self.log = log self.set_data() mainBoxSizer = wx.BoxSizer(wx.VERTICAL) enLogBox = wx.BoxSizer(wx.HORIZONTAL) self.enableLogChk = wx.CheckBox(self, label='Enable Log') enLogBox.Add(self.enableLogChk) mainBoxSizer.Add(enLogBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=20+self.win_box_border) wrTimeBox = wx.BoxSizer(wx.HORIZONTAL) self.enableTimeChk = wx.CheckBox(self, label='Write Time') wrTimeBox.Add(self.enableTimeChk) mainBoxSizer.Add(wrTimeBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=5+self.win_box_border) butBox = wx.BoxSizer(wx.HORIZONTAL) self.clearLogButton = wx.Button(self, label='Clear Log') butBox.Add(self.clearLogButton) self.viewLogButton = wx.Button(self, label='View Log') butBox.Add(self.viewLogButton, flag = wx.LEFT, border=20) mainBoxSizer.Add(butBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=15) if self.optList.enableLog: self.SetDataSizers(mainBoxSizer) self.SetSizer(mainBoxSizer) self.Bind(wx.EVT_CHECKBOX, self.OnEnable, self.enableLogChk) self.Bind(wx.EVT_CHECKBOX, self.OnTime, self.enableTimeChk) self.Bind(wx.EVT_BUTTON, self.OnClear, self.clearLogButton) self.Bind(wx.EVT_BUTTON, self.OnView, self.viewLogButton) def SetBoxBorder(self): ''' Set border for windows ''' if get_os_type() == 'nt': self.win_box_border = 10 def SetDataSizers(self, box): logPathText = wx.BoxSizer(wx.HORIZONTAL) logPathText.Add(wx.StaticText(self, label='Path: ' + self.path)) box.Add(logPathText, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=20) logSizeText = wx.BoxSizer(wx.HORIZONTAL) self.sizeText = wx.StaticText(self, label='Log Size: ' + self.size) logSizeText.Add(self.sizeText) box.Add(logSizeText, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=10) def set_data(self): if self.log != None: self.size = str(self.log.size()) + ' Bytes' self.path = self.log.path def OnTime(self, event): if self.log != None: self.log.add_time = self.enableTimeChk.GetValue() def OnEnable(self, event): if self.enableLogChk.GetValue(): self.enableTimeChk.Enable() else: self.enableTimeChk.Disable() self.restart_popup() def OnClear(self, event): if self.log != None: self.log.clear() self.sizeText.SetLabel('Log Size: 0 Bytes') def OnView(self, event): if self.log != None: log_gui = LogGUI(self.path, parent=self) log_gui.Show() def load_options(self): self.enableLogChk.SetValue(self.optList.enableLog) self.enableTimeChk.SetValue(self.optList.writeTimeToLog) if self.optList.enableLog == False: self.enableTimeChk.Disable() if self.log == None: self.clearLogButton.Disable() self.viewLogButton.Disable() def save_options(self): self.optList.enableLog = self.enableLogChk.GetValue() self.optList.writeTimeToLog = self.enableTimeChk.GetValue() def restart_popup(self): wx.MessageBox('Please restart ' + TITLE, 'Restart', wx.OK | wx.ICON_INFORMATION) class UpdatePanel(wx.Panel): def __init__(self, parent, optList): wx.Panel.__init__(self, parent) self.optList = optList mainBoxSizer = wx.BoxSizer(wx.VERTICAL) text = '''Enter the path where youtube-dlG should download the latest youtube-dl.''' helpText = wx.BoxSizer(wx.HORIZONTAL) helpText.Add(wx.StaticText(self, label=text), flag = wx.TOP, border=10) mainBoxSizer.Add(helpText, flag = wx.LEFT, border=80) pathText = wx.BoxSizer(wx.HORIZONTAL) pathText.Add(wx.StaticText(self, label='Path'), flag = wx.TOP, border=20) mainBoxSizer.Add(pathText, flag = wx.LEFT, border=95) upPathBox = wx.BoxSizer(wx.HORIZONTAL) self.updatePathBox = wx.TextCtrl(self) upPathBox.Add(self.updatePathBox, 1, flag = wx.TOP, border=5) mainBoxSizer.Add(upPathBox, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=90) autoUpBox = wx.BoxSizer(wx.HORIZONTAL) self.autoUpdateChk = wx.CheckBox(self, label='Auto Update youtube-dl') autoUpBox.Add(self.autoUpdateChk, flag = wx.TOP, border=30) mainBoxSizer.Add(autoUpBox, flag = wx.LEFT, border=100) self.SetSizer(mainBoxSizer) def load_options(self): self.updatePathBox.SetValue(self.optList.updatePath) self.autoUpdateChk.SetValue(self.optList.autoUpdate) def save_options(self): self.optList.updatePath = get_abs_path(self.updatePathBox.GetValue()) self.optList.autoUpdate = self.autoUpdateChk.GetValue() class PlaylistPanel(wx.Panel): def __init__(self, parent, optList): wx.Panel.__init__(self, parent) self.optList = optList mainBoxSizer = wx.StaticBoxSizer(wx.StaticBox(self, label='Playlist Options'), wx.VERTICAL) mainBoxSizer.Add((-1, 10)) startBox = wx.BoxSizer(wx.HORIZONTAL) startBox.Add(wx.StaticText(self, label='Start'), flag = wx.RIGHT, border=32) self.startSpnr = wx.SpinCtrl(self, size=(60, -1)) self.startSpnr.SetRange(1, 999) startBox.Add(self.startSpnr) mainBoxSizer.Add(startBox, flag = wx.TOP | wx.LEFT, border=20) stopBox = wx.BoxSizer(wx.HORIZONTAL) stopBox.Add(wx.StaticText(self, label='Stop'), flag = wx.RIGHT, border=34) self.stopSpnr = wx.SpinCtrl(self, size=(60, -1)) self.stopSpnr.SetRange(0, 999) stopBox.Add(self.stopSpnr) mainBoxSizer.Add(stopBox, flag = wx.TOP | wx.LEFT, border=20) maxBox = wx.BoxSizer(wx.HORIZONTAL) maxBox.Add(wx.StaticText(self, label='Max DLs'), flag = wx.RIGHT, border=self.get_border()) self.maxSpnr = wx.SpinCtrl(self, size=(60, -1)) self.maxSpnr.SetRange(0, 999) maxBox.Add(self.maxSpnr) mainBoxSizer.Add(maxBox, flag = wx.TOP | wx.LEFT, border=20) self.SetSizer(mainBoxSizer) def get_border(self): if get_os_type() == 'nt': return 16 return 10 def load_options(self): self.startSpnr.SetValue(self.optList.startTrack) self.stopSpnr.SetValue(self.optList.endTrack) self.maxSpnr.SetValue(self.optList.maxDownloads) def save_options(self): self.optList.startTrack = self.startSpnr.GetValue() self.optList.endTrack = self.stopSpnr.GetValue() self.optList.maxDownloads = self.maxSpnr.GetValue() class ConnectionPanel(wx.Panel): def __init__(self, parent, optList): wx.Panel.__init__(self, parent) self.optList = optList mainBoxSizer = wx.BoxSizer(wx.VERTICAL) retBox = wx.BoxSizer(wx.HORIZONTAL) retBox.Add(wx.StaticText(self, label='Retries'), flag = wx.RIGHT, border=8) self.retriesSpnr = wx.SpinCtrl(self, size=(50, -1)) self.retriesSpnr.SetRange(1, 99) retBox.Add(self.retriesSpnr) mainBoxSizer.Add(retBox, flag = wx.LEFT | wx.TOP, border=10) uaText = wx.BoxSizer(wx.HORIZONTAL) uaText.Add(wx.StaticText(self, label='User Agent'), flag = wx.LEFT, border=15) mainBoxSizer.Add(uaText, flag = wx.TOP, border=10) mainBoxSizer.Add((-1, 5)) uaBox = wx.BoxSizer(wx.HORIZONTAL) self.userAgentBox = wx.TextCtrl(self) uaBox.Add(self.userAgentBox, 1, flag = wx.LEFT, border=10) mainBoxSizer.Add(uaBox, flag = wx.EXPAND | wx.RIGHT, border=200) refText = wx.BoxSizer(wx.HORIZONTAL) refText.Add(wx.StaticText(self, label='Referer'), flag = wx.LEFT, border=15) mainBoxSizer.Add(refText, flag = wx.TOP, border=10) mainBoxSizer.Add((-1, 5)) refBox = wx.BoxSizer(wx.HORIZONTAL) self.refererBox = wx.TextCtrl(self) refBox.Add(self.refererBox, 1, flag = wx.LEFT, border=10) mainBoxSizer.Add(refBox, flag = wx.EXPAND | wx.RIGHT, border=200) prxyText = wx.BoxSizer(wx.HORIZONTAL) prxyText.Add(wx.StaticText(self, label='Proxy'), flag = wx.LEFT, border=15) mainBoxSizer.Add(prxyText, flag = wx.TOP, border=10) mainBoxSizer.Add((-1, 5)) prxyBox = wx.BoxSizer(wx.HORIZONTAL) self.proxyBox = wx.TextCtrl(self) prxyBox.Add(self.proxyBox, 1, flag = wx.LEFT, border=10) mainBoxSizer.Add(prxyBox, flag = wx.EXPAND | wx.RIGHT, border=100) self.SetSizer(mainBoxSizer) def load_options(self): self.userAgentBox.SetValue(self.optList.userAgent) self.refererBox.SetValue(self.optList.referer) self.proxyBox.SetValue(self.optList.proxy) self.retriesSpnr.SetValue(self.optList.retries) def save_options(self): self.optList.userAgent = self.userAgentBox.GetValue() self.optList.referer = self.refererBox.GetValue() self.optList.proxy = self.proxyBox.GetValue() self.optList.retries = self.retriesSpnr.GetValue() class AuthenticationPanel(wx.Panel): def __init__(self, parent, optList): wx.Panel.__init__(self,parent) self.optList = optList mainBoxSizer = wx.BoxSizer(wx.VERTICAL) usrTextBox = wx.BoxSizer(wx.HORIZONTAL) usrTextBox.Add(wx.StaticText(self, label='Username')) mainBoxSizer.Add(usrTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=15) usrBox = wx.BoxSizer(wx.HORIZONTAL) self.usernameBox = wx.TextCtrl(self, size=(-1, 25)) usrBox.Add(self.usernameBox, 1, flag = wx.TOP, border=5) mainBoxSizer.Add(usrBox, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=160) passTextBox = wx.BoxSizer(wx.HORIZONTAL) passTextBox.Add(wx.StaticText(self, label='Password')) mainBoxSizer.Add(passTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=15) passBox = wx.BoxSizer(wx.HORIZONTAL) self.passwordBox = wx.TextCtrl(self, size=(-1, 25), style = wx.TE_PASSWORD) passBox.Add(self.passwordBox, 1, flag = wx.TOP, border=5) mainBoxSizer.Add(passBox, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=160) vPassTextBox = wx.BoxSizer(wx.HORIZONTAL) vPassTextBox.Add(wx.StaticText(self, label='Video Password (vimeo, smotri)')) mainBoxSizer.Add(vPassTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=15) vPassBox = wx.BoxSizer(wx.HORIZONTAL) self.videopassBox = wx.TextCtrl(self, size=(-1, 25), style = wx.TE_PASSWORD) vPassBox.Add(self.videopassBox, 1, flag = wx.TOP, border=5) mainBoxSizer.Add(vPassBox, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=160) self.SetSizer(mainBoxSizer) def load_options(self): self.usernameBox.SetValue(self.optList.username) self.passwordBox.SetValue(self.optList.password) self.videopassBox.SetValue(self.optList.videoPass) def save_options(self): self.optList.username = self.usernameBox.GetValue() self.optList.password = self.passwordBox.GetValue() self.optList.videoPass = self.videopassBox.GetValue() class AudioPanel(wx.Panel): win_box_border = 0 def __init__(self, parent, optList): wx.Panel.__init__(self, parent) self.SetBoxBorder() self.optList = optList mainBoxSizer = wx.BoxSizer(wx.VERTICAL) toaBox = wx.BoxSizer(wx.HORIZONTAL) self.toAudioChk = wx.CheckBox(self, label='Convert to Audio') toaBox.Add(self.toAudioChk) mainBoxSizer.Add(toaBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=15+self.win_box_border) keepVBox = wx.BoxSizer(wx.HORIZONTAL) self.keepVideoChk = wx.CheckBox(self, label='Keep Video') keepVBox.Add(self.keepVideoChk) mainBoxSizer.Add(keepVBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=5+self.win_box_border) afTextBox = wx.BoxSizer(wx.HORIZONTAL) afTextBox.Add(wx.StaticText(self, label='Audio Format')) mainBoxSizer.Add(afTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=10) afComboBox = wx.BoxSizer(wx.HORIZONTAL) self.audioFormatCombo = wx.ComboBox(self, choices=AUDIOFORMATS, size=(160, 30)) afComboBox.Add(self.audioFormatCombo) mainBoxSizer.Add(afComboBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=5) aqTextBox = wx.BoxSizer(wx.HORIZONTAL) aqTextBox.Add(wx.StaticText(self, label='Audio Quality 0 (best) 9 (worst)')) mainBoxSizer.Add(aqTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=10) aqComboBox = wx.BoxSizer(wx.HORIZONTAL) self.audioQualitySpnr = wx.SpinCtrl(self, size=(90, 20)) self.audioQualitySpnr.SetRange(0, 9) aqComboBox.Add(self.audioQualitySpnr) mainBoxSizer.Add(aqComboBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=5) self.SetSizer(mainBoxSizer) self.Bind(wx.EVT_CHECKBOX, self.OnAudioCheck, self.toAudioChk) def SetBoxBorder(self): ''' Set border for windows ''' if get_os_type() == 'nt': self.win_box_border = 5 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): wx.Panel.__init__(self, parent) self.optList = optList mainBoxSizer = wx.BoxSizer(wx.VERTICAL) vfTextBox = wx.BoxSizer(wx.HORIZONTAL) vfTextBox.Add(wx.StaticText(self, label='Video Format')) mainBoxSizer.Add(vfTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=20) vfComboBox = wx.BoxSizer(wx.HORIZONTAL) self.videoFormatCombo = wx.ComboBox(self, choices=VIDEOFORMATS, size=(160, 30)) vfComboBox.Add(self.videoFormatCombo) mainBoxSizer.Add(vfComboBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=5) daTextBox = wx.BoxSizer(wx.HORIZONTAL) daTextBox.Add(wx.StaticText(self, label='DASH Audio')) mainBoxSizer.Add(daTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=10) daComboBox = wx.BoxSizer(wx.HORIZONTAL) self.dashAudioFormatCombo = wx.ComboBox(self, choices=DASH_AUDIO_FORMATS, size=(160, 30)) daComboBox.Add(self.dashAudioFormatCombo) mainBoxSizer.Add(daComboBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=5) clrDashBox = wx.BoxSizer(wx.HORIZONTAL) self.clearDashFilesChk = wx.CheckBox(self, label='Clear DASH audio/video files') clrDashBox.Add(self.clearDashFilesChk) mainBoxSizer.Add(clrDashBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=20) self.SetSizer(mainBoxSizer) self.Bind(wx.EVT_COMBOBOX, self.OnVideoFormatPick, self.videoFormatCombo) self.Bind(wx.EVT_COMBOBOX, self.OnAudioFormatPick, self.dashAudioFormatCombo) def OnAudioFormatPick(self, event): if have_dash_audio(self.dashAudioFormatCombo.GetValue()): self.clearDashFilesChk.Enable() else: self.clearDashFilesChk.SetValue(False) self.clearDashFilesChk.Disable() def OnVideoFormatPick(self, event): if video_is_dash(self.videoFormatCombo.GetValue()): self.dashAudioFormatCombo.Enable() if have_dash_audio(self.dashAudioFormatCombo.GetValue()): self.clearDashFilesChk.Enable() else: self.clearDashFilesChk.SetValue(False) self.clearDashFilesChk.Disable() self.dashAudioFormatCombo.Disable() def load_options(self): self.videoFormatCombo.SetValue(self.optList.videoFormat) self.dashAudioFormatCombo.SetValue(self.optList.dashAudioFormat) self.clearDashFilesChk.SetValue(self.optList.clearDashFiles) if not video_is_dash(self.optList.videoFormat): self.dashAudioFormatCombo.Disable() if not have_dash_audio(self.optList.dashAudioFormat): self.clearDashFilesChk.SetValue(False) self.clearDashFilesChk.Disable() def save_options(self): self.optList.videoFormat = self.videoFormatCombo.GetValue() self.optList.dashAudioFormat = self.dashAudioFormatCombo.GetValue() self.optList.clearDashFiles = self.clearDashFilesChk.GetValue() class FilesystemPanel(wx.Panel): win_box_border = 0 def __init__(self, parent, optList): wx.Panel.__init__(self, parent) self.SetBoxBorder() self.optList = optList mainBoxSizer = wx.BoxSizer(wx.HORIZONTAL) leftBoxSizer = wx.BoxSizer(wx.VERTICAL) rightBoxSizer = wx.StaticBoxSizer(wx.StaticBox(self, label='Filesize (e.g. 50k or 44.6m)'), wx.VERTICAL) self.SetLeftBox(leftBoxSizer) self.SetRightBox(rightBoxSizer) mainBoxSizer.Add(leftBoxSizer, 1, flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=10) mainBoxSizer.Add(rightBoxSizer, 1, flag = wx.EXPAND | wx.ALL | wx.ALIGN_CENTER, border=10) self.SetSizer(mainBoxSizer) def SetBoxBorder(self): ''' Set border for windows ''' if get_os_type() == 'nt': self.win_box_border = 15 def SetLeftBox(self, box): idBox = wx.BoxSizer(wx.HORIZONTAL) self.idAsNameChk = wx.CheckBox(self, label='ID as Name') idBox.Add(self.idAsNameChk, flag = wx.LEFT, border=5) box.Add(idBox, flag = wx.TOP, border=20) ignrBox = wx.BoxSizer(wx.HORIZONTAL) self.ignoreErrorsChk = wx.CheckBox(self, label='Ignore Errors') ignrBox.Add(self.ignoreErrorsChk, flag = wx.LEFT, border=5) box.Add(ignrBox, flag = wx.TOP, border=5+self.win_box_border) wrtDescBox = wx.BoxSizer(wx.HORIZONTAL) self.writeDescriptionChk = wx.CheckBox(self, label='Write description to file') wrtDescBox.Add(self.writeDescriptionChk, flag = wx.LEFT, border=5) box.Add(wrtDescBox, flag = wx.TOP, border=5+self.win_box_border) wrtInfoBox = wx.BoxSizer(wx.HORIZONTAL) self.writeInfoChk = wx.CheckBox(self, label='Write info to (.json) file') wrtInfoBox.Add(self.writeInfoChk, flag = wx.LEFT, border=5) box.Add(wrtInfoBox, flag = wx.TOP, border=5+self.win_box_border) wrtThumBox = wx.BoxSizer(wx.HORIZONTAL) self.writeThumbnailChk = wx.CheckBox(self, label='Write thumbnail to disk') wrtThumBox.Add(self.writeThumbnailChk, flag = wx.LEFT, border=5) box.Add(wrtThumBox, flag = wx.TOP, border=5+self.win_box_border) def SetRightBox(self, box): minBox = wx.BoxSizer(wx.HORIZONTAL) minBox.Add(wx.StaticText(self, label='Min'), flag = wx.RIGHT, border=12) self.minFilesizeBox = wx.TextCtrl(self, size=(80, -1)) minBox.Add(self.minFilesizeBox) box.Add(minBox, flag = wx.TOP | wx.ALIGN_CENTER_HORIZONTAL, border=10) maxBox = wx.BoxSizer(wx.HORIZONTAL) maxBox.Add(wx.StaticText(self, label='Max'), flag = wx.RIGHT, border=8) self.maxFilesizeBox = wx.TextCtrl(self, size=(80, -1)) maxBox.Add(self.maxFilesizeBox) box.Add(maxBox, flag = wx.TOP | wx.ALIGN_CENTER_HORIZONTAL, border=10) def load_options(self): 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.idAsNameChk.SetValue(self.optList.idAsName) self.minFilesizeBox.SetValue(self.optList.minFileSize) self.maxFilesizeBox.SetValue(self.optList.maxFileSize) def save_options(self): 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.idAsName = self.idAsNameChk.GetValue() self.optList.minFileSize = self.minFilesizeBox.GetValue() self.optList.maxFileSize = self.maxFilesizeBox.GetValue() self.check_input() def check_input(self): self.optList.minFileSize.replace('-', '') self.optList.maxFileSize.replace('-', '') if self.optList.minFileSize == '': self.optList.minFileSize = '0' if self.optList.maxFileSize == '': self.optList.maxFileSize = '0' class SubtitlesPanel(wx.Panel): win_box_border = 0 def __init__(self, parent, optList): wx.Panel.__init__(self, parent) self.SetBoxBorder() self.optList = optList mainBoxSizer = wx.BoxSizer(wx.VERTICAL) dlSubsBox = wx.BoxSizer(wx.HORIZONTAL) self.writeSubsChk = wx.CheckBox(self, label='Download subtitle file by language') dlSubsBox.Add(self.writeSubsChk, flag = wx.LEFT, border=10) mainBoxSizer.Add(dlSubsBox, flag = wx.TOP, border=15) dlAllSubBox = wx.BoxSizer(wx.HORIZONTAL) self.writeAllSubsChk = wx.CheckBox(self, label='Download all available subtitles') dlAllSubBox.Add(self.writeAllSubsChk, flag = wx.LEFT, border=10) mainBoxSizer.Add(dlAllSubBox, flag = wx.TOP, border=5+self.win_box_border) dlAutoSubBox = wx.BoxSizer(wx.HORIZONTAL) self.writeAutoSubsChk = wx.CheckBox(self, label='Download automatic subtitle file (YOUTUBE ONLY)') dlAutoSubBox.Add(self.writeAutoSubsChk, flag = wx.LEFT, border=10) mainBoxSizer.Add(dlAutoSubBox, flag = wx.TOP, border=5+self.win_box_border) embSubBox = wx.BoxSizer(wx.HORIZONTAL) self.embedSubsChk = wx.CheckBox(self, label='Embed subtitles in the video (only for mp4 videos)') self.embedSubsChk.Disable() embSubBox.Add(self.embedSubsChk, flag = wx.LEFT, border=10) mainBoxSizer.Add(embSubBox, flag = wx.TOP, border=5+self.win_box_border) slangTextBox = wx.BoxSizer(wx.HORIZONTAL) slangTextBox.Add(wx.StaticText(self, label='Subtitles Language'), flag = wx.LEFT, border=15) mainBoxSizer.Add(slangTextBox, flag = wx.TOP, border=10+self.win_box_border) slangBox = wx.BoxSizer(wx.HORIZONTAL) self.subsLangCombo = wx.ComboBox(self, choices=LANGUAGES, size=(140, 30)) slangBox.Add(self.subsLangCombo, flag = wx.LEFT, border=10) mainBoxSizer.Add(slangBox, flag = wx.TOP, border=5) self.SetSizer(mainBoxSizer) 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 SetBoxBorder(self): ''' Set border for windows ''' if get_os_type() == 'nt': self.win_box_border = 10 def subs_are_on(self): return self.writeAutoSubsChk.GetValue() or self.writeSubsChk.GetValue() def OnWriteAutoSubsChk(self, event): if self.writeAutoSubsChk.GetValue(): self.writeAllSubsChk.Disable() self.writeSubsChk.Disable() self.subsLangCombo.Disable() self.embedSubsChk.Enable() else: self.writeAllSubsChk.Enable() self.writeSubsChk.Enable() self.subsLangCombo.Enable() self.embedSubsChk.Disable() self.embedSubsChk.SetValue(False) def OnWriteSubsChk(self, event): if self.writeSubsChk.GetValue(): self.writeAllSubsChk.Disable() self.writeAutoSubsChk.Disable() self.embedSubsChk.Enable() else: self.writeAllSubsChk.Enable() self.writeAutoSubsChk.Enable() self.embedSubsChk.Disable() self.embedSubsChk.SetValue(False) 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) self.embedSubsChk.SetValue(self.optList.embedSubs) if self.optList.writeSubs: self.writeAllSubsChk.Disable() self.writeAutoSubsChk.Disable() self.embedSubsChk.Enable() if self.optList.writeAllSubs: self.writeSubsChk.Disable() self.subsLangCombo.Disable() self.writeAutoSubsChk.Disable() if self.optList.writeAutoSubs: self.writeAllSubsChk.Disable() self.writeSubsChk.Disable() self.subsLangCombo.Disable() self.embedSubsChk.Enable() if not self.subs_are_on(): self.embedSubsChk.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() self.optList.embedSubs = self.embedSubsChk.GetValue() class GeneralPanel(wx.Panel): def __init__(self, parent, optList, resetHandler): wx.Panel.__init__(self, parent) self.optList = optList self.resetHandler = resetHandler mainBoxSizer = wx.BoxSizer(wx.VERTICAL) svTextBox = wx.BoxSizer(wx.HORIZONTAL) svTextBox.Add(wx.StaticText(self, label='Save Path')) mainBoxSizer.Add(svTextBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.TOP, border=20) svPathBox = wx.BoxSizer(wx.HORIZONTAL) self.savePathBox = wx.TextCtrl(self) svPathBox.Add(self.savePathBox, 1, flag = wx.TOP, border=10) mainBoxSizer.Add(svPathBox, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=40) buttonsBox = wx.BoxSizer(wx.HORIZONTAL) self.aboutButton = wx.Button(self, label='About', size=(110, 40)) buttonsBox.Add(self.aboutButton) self.openButton = wx.Button(self, label='Open', size=(110, 40)) buttonsBox.Add(self.openButton, flag = wx.LEFT | wx.RIGHT, border=50) self.resetButton = wx.Button(self, label='Reset Options', size=(110, 40)) buttonsBox.Add(self.resetButton) mainBoxSizer.Add(buttonsBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.BOTTOM | wx.TOP, border=20) setngsBox = wx.BoxSizer(wx.HORIZONTAL) text = 'Settings: ' + self.optList.settings_abs_path setngsBox.Add(wx.StaticText(self, label=text), flag = wx.TOP, border=20) mainBoxSizer.Add(setngsBox, flag = wx.ALIGN_CENTER_HORIZONTAL | wx.BOTTOM, border=10) self.SetSizer(mainBoxSizer) 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.resetHandler() 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(ICON, wx.BITMAP_TYPE_ICO)) info.SetName(TITLE) info.SetVersion(__version__) info.SetDescription(description) info.SetWebSite('http://mrs0m30n3.github.io/youtube-dl-gui/') info.SetLicense(license) info.AddDeveloper('Sotiris Papadopoulos') wx.AboutBox(info) def load_options(self): self.savePathBox.SetValue(self.optList.savePath) def save_options(self): self.optList.savePath = get_abs_path(self.savePathBox.GetValue()) class OtherPanel(wx.Panel): def __init__(self, parent, optList): wx.Panel.__init__(self, parent) self.optList = optList mainBoxSizer = wx.BoxSizer(wx.VERTICAL) textBox = wx.BoxSizer(wx.HORIZONTAL) textBox.Add(wx.StaticText(self, label='Command line arguments (e.g. --help)'), flag = wx.TOP, border=30) mainBoxSizer.Add(textBox, flag = wx.LEFT, border=50) inputBox = wx.BoxSizer(wx.HORIZONTAL) self.cmdArgsBox = wx.TextCtrl(self) inputBox.Add(self.cmdArgsBox, 1, flag = wx.TOP, border=10) mainBoxSizer.Add(inputBox, flag = wx.EXPAND | wx.LEFT | wx.RIGHT, border=50) self.SetSizer(mainBoxSizer) 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, logger=None): wx.Frame.__init__(self, parent, id, "Options", size=(580, 250)) self.optionsList = optionsList panel = wx.Panel(self) notebook = wx.Notebook(panel) self.generalTab = GeneralPanel(notebook, self.optionsList, self.reset) self.audioTab = AudioPanel(notebook, self.optionsList) self.connectionTab = ConnectionPanel(notebook, self.optionsList) self.videoTab = VideoPanel(notebook, self.optionsList) self.filesysTab = FilesystemPanel(notebook, self.optionsList) self.subtitlesTab = SubtitlesPanel(notebook, self.optionsList) self.otherTab = OtherPanel(notebook, self.optionsList) self.updateTab = UpdatePanel(notebook, self.optionsList) self.authTab = AuthenticationPanel(notebook, self.optionsList) self.videoselTab = PlaylistPanel(notebook, self.optionsList) self.logTab = LogPanel(notebook, self.optionsList, logger) notebook.AddPage(self.generalTab, "General") notebook.AddPage(self.videoTab, "Video") notebook.AddPage(self.audioTab, "Audio") notebook.AddPage(self.videoselTab, "Playlist") notebook.AddPage(self.subtitlesTab, "Subtitles") notebook.AddPage(self.filesysTab, "Filesystem") notebook.AddPage(self.connectionTab, "Connection") notebook.AddPage(self.authTab, "Authentication") notebook.AddPage(self.updateTab, "Update") notebook.AddPage(self.logTab, "Log") 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() if not file_exist(fix_path(self.optionsList.updatePath)+YOUTUBE_DL_FILENAME): self.wrong_youtubedl_path() self.Destroy() def wrong_youtubedl_path(self): text = '''The path under Options>Update is invalid please do one of the following: *) restart youtube-dlG *) click the update button *) change the path to point where youtube-dl is''' wx.MessageBox(text, 'Error', wx.OK | wx.ICON_EXCLAMATION) 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.filesysTab.load_options() self.subtitlesTab.load_options() self.otherTab.load_options() self.updateTab.load_options() self.authTab.load_options() self.videoselTab.load_options() self.logTab.load_options() def save_all_options(self): self.generalTab.save_options() self.audioTab.save_options() self.connectionTab.save_options() self.videoTab.save_options() self.filesysTab.save_options() self.subtitlesTab.save_options() self.otherTab.save_options() self.updateTab.save_options() self.authTab.save_options() self.videoselTab.save_options() self.logTab.save_options()