You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

797 lines
30 KiB

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""Youtubedlg module responsible for the options window. """
from __future__ import unicode_literals
import os
import gettext
import wx
from .version import __version__
from .info import (
__descriptionfull__,
__licensefull__,
__projecturl__,
__appname__,
__author__
)
from .utils import (
TwoWayOrderedDict as twodict,
os_path_exists,
get_icon_file,
read_formats
)
from .formats import OUTPUT_FORMATS
#TODO Bind events
#TODO Adjust layout
#TODO Set frame's min size
#TODO Add labels to gettext
#TODO Move all formats, etc to formats.py
class OptionsFrame(wx.Frame):
"""Youtubedlg options frame class.
Args:
parent (mainframe.MainFrame): Parent class.
"""
FRAME_TITLE = _("Options")
def __init__(self, parent):
wx.Frame.__init__(self, parent, title=self.FRAME_TITLE, size=parent.opt_manager.options["opts_win_size"])
self.opt_manager = parent.opt_manager
self.log_manager = parent.log_manager
self.app_icon = None
# Set the app icon
app_icon_path = get_icon_file()
if app_icon_path is not None:
self.app_icon = wx.Icon(app_icon_path, wx.BITMAP_TYPE_PNG)
self.SetIcon(self.app_icon)
self._was_shown = False
# Create options frame basic components
self.panel = wx.Panel(self)
self.notebook = wx.Notebook(self.panel)
self.separator_line = wx.StaticLine(self.panel)
self.reset_button = wx.Button(self.panel, label="Reset")
self.close_button = wx.Button(self.panel, label="Close")
# Create tabs
tab_args = (self, self.notebook)
self.tabs = (
(GeneralTab(*tab_args), "General"),
(FormatsTab(*tab_args), "Formats"),
(DownloadsTab(*tab_args), "Downloads"),
(AdvancedTab(*tab_args), "Advanced"),
(ExtraTab(*tab_args), "Extra")
)
# Add tabs on notebook
for tab, label in self.tabs:
self.notebook.AddPage(tab, label)
# Bind events
self.Bind(wx.EVT_BUTTON, self._on_reset, self.reset_button)
self.Bind(wx.EVT_BUTTON, self._on_close, self.close_button)
self.Bind(wx.EVT_CLOSE, self._on_close)
self._set_layout()
self.load_all_options()
def _set_layout(self):
main_sizer = wx.BoxSizer(wx.VERTICAL)
main_sizer.Add(self.notebook, 1, wx.EXPAND | wx.ALL, border=5)
main_sizer.Add(self.separator_line, 0, wx.EXPAND)
buttons_sizer = wx.BoxSizer(wx.HORIZONTAL)
buttons_sizer.Add(self.reset_button)
buttons_sizer.Add(self.close_button)
main_sizer.Add(buttons_sizer, flag=wx.ALIGN_RIGHT | wx.ALL, border=5)
self.panel.SetSizer(main_sizer)
def _on_close(self, event):
"""Event handler for wx.EVT_CLOSE event."""
self.save_all_options()
self.Hide()
def _on_reset(self, event):
"""Event handler for the reset button wx.EVT_BUTTON event."""
self.reset()
def reset(self):
"""Reset the default options."""
self.opt_manager.load_default()
self.load_all_options()
def load_all_options(self):
"""Load all the options on each tab."""
for tab, _ in self.tabs:
tab.load_options()
def save_all_options(self):
"""Save all the options from all the tabs back to the OptionsManager."""
for tab, _ in self.tabs:
tab.save_options()
def Show(self, *args, **kwargs):
# CenterOnParent can't go to main frame's __init__ as main frame may change
# own position and options frame won't be centered on main frame anymore.
if not self._was_shown:
self._was_shown = True
self.CenterOnParent()
return wx.Frame.Show(self, *args, **kwargs)
class TabPanel(wx.Panel):
"""Main tab class from which each tab inherits.
Args:
parent (OptionsFrame): The parent of all tabs.
notebook (wx.Notebook): The container for each tab.
Notes:
In order to use a different size you must overwrite the below *_SIZE
attributes on the corresponding child object.
"""
CHECKBOX_SIZE = (-1, -1)
if os.name == "nt":
# Make checkboxes look the same on Windows
CHECKBOX_SIZE = (-1, 25)
BUTTONS_SIZE = (-1, -1)
TEXTCTRL_SIZE = (-1, -1)
SPINCTRL_SIZE = (70, -1)
CHECKLISTBOX_SIZE = (-1, 100)
LISTBOX_SIZE = (-1, 100)
def __init__(self, parent, notebook):
wx.Panel.__init__(self, notebook)
self.opt_manager = parent.opt_manager
self.log_manager = parent.log_manager
self.app_icon = parent.app_icon
self.reset_handler = parent.reset
# Shortcut methods below
def crt_button(self, label, event_handler=None):
button = wx.Button(self, label=label, size=self.BUTTONS_SIZE)
if event_handler is not None:
button.Bind(wx.EVT_BUTTON, event_handler)
return button
def crt_checkbox(self, label, event_handler=None):
checkbox = wx.CheckBox(self, label=label, size=self.CHECKBOX_SIZE)
if event_handler is not None:
checkbox.Bind(wx.EVT_CHECKBOX, event_handler)
return checkbox
def crt_textctrl(self, style=None):
if style is None:
textctrl = wx.TextCtrl(self, size=self.TEXTCTRL_SIZE)
else:
textctrl = wx.TextCtrl(self, size=self.TEXTCTRL_SIZE, style=style)
return textctrl
def crt_combobox(self, choices, size=(-1, -1), event_handler=None):
combobox = wx.ComboBox(self, choices=choices, size=size, style=wx.CB_READONLY)
if event_handler is not None:
combobox.Bind(wx.EVT_COMBOBOX, event_handler)
return combobox
def crt_spinctrl(self, spin_range=(0, 9999)):
spinctrl = wx.SpinCtrl(self, size=self.SPINCTRL_SIZE)
spinctrl.SetRange(*spin_range)
return spinctrl
def crt_statictext(self, label):
return wx.StaticText(self, wx.ID_ANY, label)
def crt_staticbox(self, label):
return wx.StaticBox(self, wx.ID_ANY, label)
def crt_checklistbox(self, choices, style=None):
if style is None:
checklistbox = wx.CheckListBox(self, choices=choices, size=self.CHECKLISTBOX_SIZE)
else:
checklistbox = wx.CheckListBox(self, choices=choices, style=style, size=self.CHECKLISTBOX_SIZE)
return checklistbox
def crt_listbox(self, choices, style=None):
if style is None:
listbox = wx.ListBox(self, choices=choices, size=self.LISTBOX_SIZE)
else:
listbox = wx.ListBox(self, choices=choices, style=style, size=self.LISTBOX_SIZE)
return listbox
class GeneralTab(TabPanel):
LOCALE_NAMES = twodict([
('ar_AR', 'Arabic'),
('cs_CZ', 'Czech'),
('en_US', 'English'),
('fr_FR', 'French'),
('de_DE', 'German'),
('it_IT', 'Italian'),
('he_IS', 'Hebrew'),
('hu_HU', 'Hungarian'),
('pt_BR', 'Portuguese'),
('ru_RU', 'Russian'),
('es_ES', 'Spanish'),
('es_MX', 'Mexican Spanish'),
('tr_TR', 'Turkish')
])
def __init__(self, *args, **kwargs):
super(GeneralTab, self).__init__(*args, **kwargs)
self.language_label = self.crt_statictext("Language")
self.language_combobox = self.crt_combobox(self.LOCALE_NAMES.values())
self.filename_format_label = self.crt_statictext("Filename format")
self.filename_format_combobox = self.crt_combobox(OUTPUT_FORMATS.values())
self.filename_custom_format = self.crt_textctrl()
self.filename_opts_label = self.crt_statictext("Filename options")
self.filename_ascii_checkbox = self.crt_checkbox("Restrict filenames to ASCII")
self.more_opts_label = self.crt_statictext("More options")
self.confirm_exit_checkbox = self.crt_checkbox("Confirm on exit")
self.shutdown_checkbox = self.crt_checkbox("Shutdown")
self.sudo_textctrl = self.crt_textctrl(wx.TE_PASSWORD)
self._set_layout()
def _set_layout(self):
main_sizer = wx.BoxSizer(wx.HORIZONTAL)
vertical_sizer = wx.BoxSizer(wx.VERTICAL)
vertical_sizer.Add(self.language_label)
vertical_sizer.Add(self.language_combobox, flag=wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.filename_format_label)
vertical_sizer.Add(self.filename_format_combobox, flag=wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.filename_custom_format, flag=wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.filename_opts_label)
vertical_sizer.Add(self.filename_ascii_checkbox, flag=wx.ALL, border=5)
vertical_sizer.Add(self.more_opts_label)
vertical_sizer.Add(self.confirm_exit_checkbox, flag=wx.ALL, border=5)
shutdown_sizer = wx.BoxSizer(wx.HORIZONTAL)
shutdown_sizer.Add(self.shutdown_checkbox)
shutdown_sizer.AddSpacer((-1, -1), 1)
shutdown_sizer.Add(self.sudo_textctrl, 1)
vertical_sizer.Add(shutdown_sizer, flag=wx.EXPAND | wx.LEFT | wx.RIGHT, border=5)
main_sizer.Add(vertical_sizer, 1, wx.EXPAND | wx.ALL, border=5)
self.SetSizer(main_sizer)
#TODO Implement load-save for confirm_exit_checkbox widget
def load_options(self):
self.language_combobox.SetValue(self.LOCALE_NAMES[self.opt_manager.options["locale_name"]])
self.filename_format_combobox.SetValue(OUTPUT_FORMATS[self.opt_manager.options["output_format"]])
self.filename_custom_format.SetValue(self.opt_manager.options["output_template"])
self.filename_ascii_checkbox.SetValue(self.opt_manager.options["restrict_filenames"])
self.shutdown_checkbox.SetValue(self.opt_manager.options["shutdown"])
self.sudo_textctrl.SetValue(self.opt_manager.options["sudo_password"])
def save_options(self):
self.opt_manager.options["locale_name"] = self.LOCALE_NAMES[self.language_combobox.GetValue()]
self.opt_manager.options["output_format"] = OUTPUT_FORMATS[self.filename_format_combobox.GetValue()]
self.opt_manager.options["output_template"] = self.filename_custom_format.GetValue()
self.opt_manager.options["restrict_filenames"] = self.filename_ascii_checkbox.GetValue()
self.opt_manager.options["shutdown"] = self.shutdown_checkbox.GetValue()
self.opt_manager.options["sudo_password"] = self.sudo_textctrl.GetValue()
class FormatsTab(TabPanel):
AUDIO_QUALITY = twodict([("0", _("high")), ("5", _("mid")), ("9", _("low"))])
VIDEO_FORMATS = read_formats()
AUDIO_FORMATS = ["mp3", "wav", "aac", "m4a", "vorbis", "opus"]
def __init__(self, *args, **kwargs):
super(FormatsTab, self).__init__(*args, **kwargs)
self.video_formats_label = self.crt_statictext("Video formats")
self.video_formats_checklistbox = self.crt_checklistbox(self.VIDEO_FORMATS.values())
self.audio_formats_label = self.crt_statictext("Audio formats")
self.audio_formats_checklistbox = self.crt_checklistbox(self.AUDIO_FORMATS)
self.post_proc_opts_label = self.crt_statictext("Post-Process options")
self.keep_video_checkbox = self.crt_checkbox("Keep original video")
self.audio_quality_label = self.crt_statictext("Audio quality")
self.audio_quality_combobox = self.crt_combobox(self.AUDIO_QUALITY.values())
self._set_layout()
def _set_layout(self):
main_sizer = wx.BoxSizer(wx.HORIZONTAL)
vertical_sizer = wx.BoxSizer(wx.VERTICAL)
vertical_sizer.Add(self.video_formats_label)
vertical_sizer.Add(self.video_formats_checklistbox, 1, wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.audio_formats_label)
vertical_sizer.Add(self.audio_formats_checklistbox, 1, wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.post_proc_opts_label)
vertical_sizer.Add(self.keep_video_checkbox, flag=wx.ALL, border=5)
audio_quality_sizer = wx.BoxSizer(wx.HORIZONTAL)
audio_quality_sizer.Add(self.audio_quality_label, flag=wx.ALIGN_CENTER_VERTICAL)
audio_quality_sizer.AddSpacer((20, -1))
audio_quality_sizer.Add(self.audio_quality_combobox)
vertical_sizer.Add(audio_quality_sizer, flag=wx.LEFT | wx.RIGHT, border=10)
main_sizer.Add(vertical_sizer, 1, wx.EXPAND | wx.ALL, border=5)
self.SetSizer(main_sizer)
def load_options(self):
#TODO Add video_formats_checklistbox
#TODO Add audio_formats_checklistbox
self.keep_video_checkbox.SetValue(self.opt_manager.options["keep_video"])
self.audio_quality_combobox.SetValue(self.AUDIO_QUALITY[self.opt_manager.options["audio_quality"]])
def save_options(self):
#TODO Add video_formats_checklistbox
#TODO Add audio_formats_checklistbox
self.opt_manager.options["keep_video"] = self.keep_video_checkbox.GetValue()
self.opt_manager.options["audio_quality"] = self.AUDIO_QUALITY[self.audio_quality_combobox.GetValue()]
class DownloadsTab(TabPanel):
SUBS_LANG = twodict([
("en", _("English")),
("gr", _("Greek")),
("pt", _("Portuguese")),
("fr", _("French")),
("it", _("Italian")),
("ru", _("Russian")),
("es", _("Spanish")),
("tr", _("Turkish")),
("de", _("German"))
])
FILESIZES = twodict([
("", "Bytes"),
("k", "Kilobytes"),
("m", "Megabytes"),
("g", "Gigabytes"),
("t", "Terabytes"),
("p", "Petabytes"),
("e", "Exabytes"),
("z", "Zettabytes"),
("y", "Yottabytes")
])
SUBS_CHOICES = [
"None",
"Automatic subtitles (YOUTUBE ONLY)",
"All available subtitles",
"Subtitles by language"
]
def __init__(self, *args, **kwargs):
super(DownloadsTab, self).__init__(*args, **kwargs)
self.subtitles_label = self.crt_statictext("Subtitles")
self.subtitles_combobox = self.crt_combobox(self.SUBS_CHOICES)
self.subtitles_lang_listbox = self.crt_listbox(self.SUBS_LANG.values())
self.subtitles_opts_label = self.crt_statictext("Subtitles options")
self.embed_subs_checkbox = self.crt_checkbox("Embed subtitles into video file (mp4 ONLY)")
self.playlist_box = self.crt_staticbox("Playlist")
self.playlist_start_label = self.crt_statictext("Start")
self.playlist_start_spinctrl = self.crt_spinctrl()
self.playlist_stop_label = self.crt_statictext("Stop")
self.playlist_stop_spinctrl = self.crt_spinctrl()
self.playlist_max_label = self.crt_statictext("Max")
self.playlist_max_spinctrl = self.crt_spinctrl()
self.filesize_box = self.crt_staticbox("Filesize")
self.filesize_min_label = self.crt_statictext("Min")
self.filesize_min_spinctrl = self.crt_spinctrl()
self.filesize_min_sizeunit_combobox = self.crt_combobox(self.FILESIZES.values())
self.filesize_max_label = self.crt_statictext("Max")
self.filesize_max_spinctrl = self.crt_spinctrl()
self.filesize_max_sizeunit_combobox = self.crt_combobox(self.FILESIZES.values())
self._set_layout()
def _set_layout(self):
main_sizer = wx.BoxSizer(wx.HORIZONTAL)
vertical_sizer = wx.BoxSizer(wx.VERTICAL)
vertical_sizer.Add(self.subtitles_label)
vertical_sizer.Add(self.subtitles_combobox, flag=wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.subtitles_lang_listbox, 1, wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.subtitles_opts_label)
vertical_sizer.Add(self.embed_subs_checkbox, flag=wx.ALL, border=5)
plist_and_fsize_sizer = wx.BoxSizer(wx.HORIZONTAL)
plist_and_fsize_sizer.Add(self._build_playlist_sizer(), 1, wx.EXPAND)
plist_and_fsize_sizer.AddSpacer((10, -1))
plist_and_fsize_sizer.Add(self._build_filesize_sizer(), 1, wx.EXPAND)
vertical_sizer.Add(plist_and_fsize_sizer, 1, wx.EXPAND | wx.TOP, border=5)
main_sizer.Add(vertical_sizer, 1, wx.EXPAND | wx.ALL, border=5)
self.SetSizer(main_sizer)
def _build_playlist_sizer(self):
left_right_border = 80
playlist_box_sizer = wx.StaticBoxSizer(self.playlist_box, wx.VERTICAL)
playlist_box_sizer.AddSpacer((-1, 10))
start_plist_sizer = wx.BoxSizer(wx.HORIZONTAL)
start_plist_sizer.AddSpacer((left_right_border, -1))
start_plist_sizer.Add(self.playlist_start_label, flag=wx.ALIGN_CENTER_VERTICAL)
start_plist_sizer.AddSpacer((-1, -1), 1)
start_plist_sizer.Add(self.playlist_start_spinctrl)
start_plist_sizer.AddSpacer((left_right_border, -1))
stop_plist_sizer = wx.BoxSizer(wx.HORIZONTAL)
stop_plist_sizer.AddSpacer((left_right_border, -1))
stop_plist_sizer.Add(self.playlist_stop_label, flag=wx.ALIGN_CENTER_VERTICAL)
stop_plist_sizer.AddSpacer((-1, -1), 1)
stop_plist_sizer.Add(self.playlist_stop_spinctrl)
stop_plist_sizer.AddSpacer((left_right_border, -1))
max_plist_sizer = wx.BoxSizer(wx.HORIZONTAL)
max_plist_sizer.AddSpacer((left_right_border, -1))
max_plist_sizer.Add(self.playlist_max_label, flag=wx.ALIGN_CENTER_VERTICAL)
max_plist_sizer.AddSpacer((-1, -1), 1)
max_plist_sizer.Add(self.playlist_max_spinctrl)
max_plist_sizer.AddSpacer((left_right_border, -1))
playlist_box_sizer.Add(start_plist_sizer, flag=wx.EXPAND | wx.TOP, border=5)
playlist_box_sizer.Add(stop_plist_sizer, flag=wx.EXPAND | wx.TOP, border=5)
playlist_box_sizer.Add(max_plist_sizer, flag=wx.EXPAND | wx.TOP, border=5)
return playlist_box_sizer
def _build_filesize_sizer(self):
left_right_border = 40
filesize_box_sizer = wx.StaticBoxSizer(self.filesize_box, wx.VERTICAL)
filesize_box_sizer.AddSpacer((-1, 10))
max_filesize_sizer = wx.BoxSizer(wx.HORIZONTAL)
max_filesize_sizer.AddSpacer((left_right_border, -1))
max_filesize_sizer.Add(self.filesize_max_spinctrl)
max_filesize_sizer.AddSpacer((-1, -1), 1)
max_filesize_sizer.Add(self.filesize_max_sizeunit_combobox)
max_filesize_sizer.AddSpacer((left_right_border, -1))
min_filesize_sizer = wx.BoxSizer(wx.HORIZONTAL)
min_filesize_sizer.AddSpacer((left_right_border, -1))
min_filesize_sizer.Add(self.filesize_min_spinctrl)
min_filesize_sizer.AddSpacer((-1, -1), 1)
min_filesize_sizer.Add(self.filesize_min_sizeunit_combobox)
min_filesize_sizer.AddSpacer((left_right_border, -1))
filesize_box_sizer.Add(self.filesize_max_label, flag=wx.ALIGN_CENTER_HORIZONTAL)
filesize_box_sizer.Add(max_filesize_sizer, flag=wx.EXPAND | wx.TOP, border=5)
filesize_box_sizer.Add(self.filesize_min_label, flag=wx.ALIGN_CENTER_HORIZONTAL)
filesize_box_sizer.Add(min_filesize_sizer, flag=wx.EXPAND | wx.TOP, border=5)
return filesize_box_sizer
def load_options(self):
if self.opt_manager.options["write_subs"]:
self.subtitles_combobox.SetValue(self.SUBS_CHOICES[3])
elif self.opt_manager.options["write_all_subs"]:
self.subtitles_combobox.SetValue(self.SUBS_CHOICES[2])
elif self.opt_manager.options["write_auto_subs"]:
self.subtitles_combobox.SetValue(self.SUBS_CHOICES[1])
else:
self.subtitles_combobox.SetValue(self.SUBS_CHOICES[0])
self.subtitles_lang_listbox.SetStringSelection(self.SUBS_LANG[self.opt_manager.options["subs_lang"]])
self.embed_subs_checkbox.SetValue(self.opt_manager.options["embed_subs"])
self.playlist_start_spinctrl.SetValue(self.opt_manager.options["playlist_start"])
self.playlist_stop_spinctrl.SetValue(self.opt_manager.options["playlist_end"])
self.playlist_max_spinctrl.SetValue(self.opt_manager.options["max_downloads"])
self.filesize_min_spinctrl.SetValue(self.opt_manager.options["min_filesize"])
self.filesize_max_spinctrl.SetValue(self.opt_manager.options["max_filesize"])
self.filesize_min_sizeunit_combobox.SetValue(self.FILESIZES[self.opt_manager.options["min_filesize_unit"]])
self.filesize_max_sizeunit_combobox.SetValue(self.FILESIZES[self.opt_manager.options["max_filesize_unit"]])
def save_options(self):
subs_choice = self.SUBS_CHOICES.index(self.subtitles_combobox.GetValue())
if subs_choice == 1:
self.opt_manager.options["write_subs"] = False
self.opt_manager.options["write_all_subs"] = False
self.opt_manager.options["write_auto_subs"] = True
elif subs_choice == 2:
self.opt_manager.options["write_subs"] = False
self.opt_manager.options["write_all_subs"] = True
self.opt_manager.options["write_auto_subs"] = False
elif subs_choice == 3:
self.opt_manager.options["write_subs"] = True
self.opt_manager.options["write_all_subs"] = False
self.opt_manager.options["write_auto_subs"] = False
else:
self.opt_manager.options["write_subs"] = False
self.opt_manager.options["write_all_subs"] = False
self.opt_manager.options["write_auto_subs"] = False
self.opt_manager.options["subs_lang"] = self.SUBS_LANG[self.subtitles_lang_listbox.GetStringSelection()]
self.opt_manager.options["embed_subs"] = self.embed_subs_checkbox.GetValue()
self.opt_manager.options["playlist_start"] = self.playlist_start_spinctrl.GetValue()
self.opt_manager.options["playlist_end"] = self.playlist_stop_spinctrl.GetValue()
self.opt_manager.options["max_downloads"] = self.playlist_max_spinctrl.GetValue()
self.opt_manager.options["min_filesize"] = self.filesize_min_spinctrl.GetValue()
self.opt_manager.options["max_filesize"] = self.filesize_max_spinctrl.GetValue()
self.opt_manager.options["min_filesize_unit"] = self.FILESIZES[self.filesize_min_sizeunit_combobox.GetValue()]
self.opt_manager.options["max_filesize_unit"] = self.FILESIZES[self.filesize_max_sizeunit_combobox.GetValue()]
class AdvancedTab(TabPanel):
TEXTCTRL_SIZE = (250, -1)
def __init__(self, *args, **kwargs):
super(AdvancedTab, self).__init__(*args, **kwargs)
self.retries_label = self.crt_statictext("Retries")
self.retries_spinctrl = self.crt_spinctrl()
self.auth_label = self.crt_statictext("Authentication")
self.username_label = self.crt_statictext("Username")
self.username_textctrl = self.crt_textctrl()
self.password_label = self.crt_statictext("Password")
self.password_textctrl = self.crt_textctrl(wx.TE_PASSWORD)
self.video_pass_label = self.crt_statictext("Video password")
self.video_pass_textctrl = self.crt_textctrl(wx.TE_PASSWORD)
self.network_label = self.crt_statictext("Network")
self.proxy_label = self.crt_statictext("Proxy")
self.proxy_textctrl = self.crt_textctrl()
self.useragent_label = self.crt_statictext("User agent")
self.useragent_textctrl = self.crt_textctrl()
self.referer_label = self.crt_statictext("Referer")
self.referer_textctrl = self.crt_textctrl()
self.logging_label = self.crt_statictext("Logging")
self.enable_log_checkbox = self.crt_checkbox("Enable log")
self.view_log_button = self.crt_button("View")
self.clear_log_button = self.crt_button("Clear")
self._set_layout()
def _set_layout(self):
main_sizer = wx.BoxSizer(wx.HORIZONTAL)
vertical_sizer = wx.BoxSizer(wx.VERTICAL)
retries_sizer = wx.BoxSizer(wx.HORIZONTAL)
retries_sizer.AddSpacer((10, -1))
retries_sizer.Add(self.retries_label, flag=wx.ALIGN_CENTER_VERTICAL)
retries_sizer.AddSpacer((20, -1))
retries_sizer.Add(self.retries_spinctrl)
retries_sizer.AddSpacer((10, -1))
vertical_sizer.Add(retries_sizer, flag=wx.ALIGN_RIGHT | wx.TOP, border=5)
vertical_sizer.Add(self.auth_label, flag=wx.TOP, border=5)
username_sizer = wx.BoxSizer(wx.HORIZONTAL)
username_sizer.AddSpacer((10, -1))
username_sizer.Add(self.username_label, flag=wx.ALIGN_CENTER_VERTICAL)
username_sizer.AddSpacer((-1, -1), 1)
username_sizer.Add(self.username_textctrl)
username_sizer.AddSpacer((10, -1))
vertical_sizer.Add(username_sizer, flag=wx.EXPAND | wx.TOP, border=5)
password_sizer = wx.BoxSizer(wx.HORIZONTAL)
password_sizer.AddSpacer((10, -1))
password_sizer.Add(self.password_label, flag=wx.ALIGN_CENTER_VERTICAL)
password_sizer.AddSpacer((-1, -1), 1)
password_sizer.Add(self.password_textctrl)
password_sizer.AddSpacer((10, -1))
vertical_sizer.Add(password_sizer, flag=wx.EXPAND | wx.TOP, border=5)
video_pass_sizer = wx.BoxSizer(wx.HORIZONTAL)
video_pass_sizer.AddSpacer((10, -1))
video_pass_sizer.Add(self.video_pass_label, flag=wx.ALIGN_CENTER_VERTICAL)
video_pass_sizer.AddSpacer((-1, -1), 1)
video_pass_sizer.Add(self.video_pass_textctrl)
video_pass_sizer.AddSpacer((10, -1))
vertical_sizer.Add(video_pass_sizer, flag=wx.EXPAND | wx.TOP, border=5)
vertical_sizer.Add(self.network_label, flag=wx.TOP, border=15)
proxy_sizer = wx.BoxSizer(wx.HORIZONTAL)
proxy_sizer.AddSpacer((10, -1))
proxy_sizer.Add(self.proxy_label, flag=wx.ALIGN_CENTER_VERTICAL)
proxy_sizer.AddSpacer((-1, -1), 1)
proxy_sizer.Add(self.proxy_textctrl)
proxy_sizer.AddSpacer((10, -1))
vertical_sizer.Add(proxy_sizer, flag=wx.EXPAND | wx.TOP, border=5)
useragent_sizer = wx.BoxSizer(wx.HORIZONTAL)
useragent_sizer.AddSpacer((10, -1))
useragent_sizer.Add(self.useragent_label, flag=wx.ALIGN_CENTER_VERTICAL)
useragent_sizer.AddSpacer((-1, -1), 1)
useragent_sizer.Add(self.useragent_textctrl)
useragent_sizer.AddSpacer((10, -1))
vertical_sizer.Add(useragent_sizer, flag=wx.EXPAND | wx.TOP, border=5)
referer_sizer = wx.BoxSizer(wx.HORIZONTAL)
referer_sizer.AddSpacer((10, -1))
referer_sizer.Add(self.referer_label, flag=wx.ALIGN_CENTER_VERTICAL)
referer_sizer.AddSpacer((-1, -1), 1)
referer_sizer.Add(self.referer_textctrl)
referer_sizer.AddSpacer((10, -1))
vertical_sizer.Add(referer_sizer, flag=wx.EXPAND | wx.TOP, border=5)
vertical_sizer.Add(self.logging_label, flag=wx.TOP, border=15)
logging_sizer = wx.BoxSizer(wx.HORIZONTAL)
logging_sizer.AddSpacer((10, -1))
logging_sizer.Add(self.enable_log_checkbox)
logging_sizer.AddSpacer((-1, -1), 1)
logging_sizer.Add(self.view_log_button)
logging_sizer.Add(self.clear_log_button)
logging_sizer.AddSpacer((10, -1))
vertical_sizer.Add(logging_sizer, flag=wx.EXPAND | wx.TOP, border=5)
main_sizer.Add(vertical_sizer, 1, wx.EXPAND | wx.ALL, border=5)
self.SetSizer(main_sizer)
def load_options(self):
self.retries_spinctrl.SetValue(self.opt_manager.options["retries"])
self.username_textctrl.SetValue(self.opt_manager.options["username"])
self.password_textctrl.SetValue(self.opt_manager.options["password"])
self.video_pass_textctrl.SetValue(self.opt_manager.options["video_password"])
self.proxy_textctrl.SetValue(self.opt_manager.options["proxy"])
self.useragent_textctrl.SetValue(self.opt_manager.options["user_agent"])
self.referer_textctrl.SetValue(self.opt_manager.options["referer"])
self.enable_log_checkbox.SetValue(self.opt_manager.options["enable_log"])
def save_options(self):
self.opt_manager.options["retries"] = self.retries_spinctrl.GetValue()
self.opt_manager.options["username"] = self.username_textctrl.GetValue()
self.opt_manager.options["password"] = self.password_textctrl.GetValue()
self.opt_manager.options["video_password"] = self.video_pass_textctrl.GetValue()
self.opt_manager.options["proxy"] = self.proxy_textctrl.GetValue()
self.opt_manager.options["user_agent"] = self.useragent_textctrl.GetValue()
self.opt_manager.options["referer"] = self.referer_textctrl.GetValue()
self.opt_manager.options["enable_log"] = self.enable_log_checkbox.GetValue()
class ExtraTab(TabPanel):
def __init__(self, *args, **kwargs):
super(ExtraTab, self).__init__(*args, **kwargs)
self.cmdline_args_label = self.crt_statictext("Command line arguments (e.g. --help)")
self.cmdline_args_textctrl = self.crt_textctrl(wx.TE_MULTILINE | wx.TE_LINEWRAP)
self.extra_opts_label = self.crt_statictext("Extra options")
self.ignore_errors_checkbox = self.crt_checkbox("Ignore errors")
self._set_layout()
def _set_layout(self):
main_sizer = wx.BoxSizer(wx.HORIZONTAL)
vertical_sizer = wx.BoxSizer(wx.VERTICAL)
vertical_sizer.Add(self.cmdline_args_label)
vertical_sizer.Add(self.cmdline_args_textctrl, 1, wx.EXPAND | wx.ALL, border=5)
vertical_sizer.Add(self.extra_opts_label)
extra_opts_sizer = wx.WrapSizer()
extra_opts_sizer.Add(self.ignore_errors_checkbox)
vertical_sizer.Add(extra_opts_sizer, flag=wx.ALL, border=5)
main_sizer.Add(vertical_sizer, 1, wx.EXPAND | wx.ALL, border=5)
self.SetSizer(main_sizer)
def load_options(self):
self.cmdline_args_textctrl.SetValue(self.opt_manager.options["cmd_args"])
self.ignore_errors_checkbox.SetValue(self.opt_manager.options["ignore_errors"])
def save_options(self):
self.opt_manager.options["cmd_args"] = self.cmdline_args_textctrl.GetValue()
self.opt_manager.options["ignore_errors"] = self.ignore_errors_checkbox.GetValue()
class LogGUI(wx.Frame):
"""Simple window for reading the STDERR.
Attributes:
TITLE (string): Frame title.
FRAME_SIZE (tuple): Tuple that holds the frame size (width, height).
Args:
parent (wx.Window): Frame parent.
"""
# REFACTOR move it on widgets module
TITLE = _("Log Viewer")
FRAME_SIZE = (650, 200)
def __init__(self, parent=None):
wx.Frame.__init__(self, parent, title=self.TITLE, size=self.FRAME_SIZE)
panel = wx.Panel(self)
self._text_area = wx.TextCtrl(
panel,
style=wx.TE_MULTILINE | wx.TE_READONLY | wx.HSCROLL
)
sizer = wx.BoxSizer()
sizer.Add(self._text_area, 1, wx.EXPAND)
panel.SetSizerAndFit(sizer)
def load(self, filename):
"""Load file content on the text area. """
if os_path_exists(filename):
self._text_area.LoadFile(filename)