def __open_url(self, evt):
url = self.__txt_url.GetValue()
if StringUtils.is_url_correct(url):
self.bth_open_url.Disable()
name = self.get_browser_initials()
try:
if self.__browser and self.__browser.get_browser_initials(
) != name:
self.__browser.quit()
self.__browser = Browser(name)
elif not self.__browser:
self.__browser = Browser(name)
except Exception:
show_error_dialog(self, traceback.format_exc(),
u'Failed to open browser')
self.__browser = None
if self.__browser:
self.__browser.open(url)
# TODO: if generator or selector -> load image
self.bth_open_url.Enable()
else:
show_dialog(self, u'Bad url: %s' % url, u'Bad url')
def __create_widgets(self):
panel = Panel(self)
sizer = GridBagSizer(5, 5)
row = 0
col = 0
label = StaticText(panel, label=u'Root folder:')
sizer.Add(label, pos=(row, col))
col += 1
col_width = 4
self.__txt_root_path = TextCtrl(panel)
sizer.Add(self.__txt_root_path, pos=(row, col), span=(1, col_width), flag=FLAG_ALL_AND_EXPAND)
col += col_width
self.__btn_set_root = Button(panel, label=u'Set root folder')
self.__btn_set_root.Bind(EVT_BUTTON, self.__set_root_folder)
sizer.Add(self.__btn_set_root, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
label = StaticText(panel, label=u'Url:')
sizer.Add(label, pos=(row, col))
col += 1
self.__txt_url = TextCtrl(panel, value=u'https://www.google.com/') # TODO: remove url
sizer.Add(self.__txt_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
label = StaticText(panel, label=u'Browser:')
sizer.Add(label, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.__cb_browser = Choice(panel, choices=Browser.get_supported_browsers())
self.__cb_browser.Select(0)
sizer.Add(self.__cb_browser, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_open_url = Button(panel, label=u'Open url')
self.bth_open_url.Bind(EVT_BUTTON, self.__open_url)
sizer.Add(self.bth_open_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_close_url = Button(panel, label=u'Close browser')
self.bth_close_url.Bind(EVT_BUTTON, self.__close_browser)
sizer.Add(self.bth_close_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
tabs = Tabs(panel, [(GeneratorTab, "Generator"),
(EditorTab, "Editor"),
(TestRunnerTab, "Test runner"),
(SelectorFinderTab, "Selector finder")])
sizer.Add(tabs, pos=(row, col), span=(1, 6), flag=FLAG_ALL_AND_EXPAND)
sizer.AddGrowableCol(1, 1)
sizer.AddGrowableRow(2, 1)
panel.SetSizer(sizer)
self.Layout()
def options(self, parser, env):
browsers = Browser.get_supported_browsers()
parser.add_option('-b', '--browser',
help="Specify browser by using initials. " \
"If value was not passed then 'ff' will be used. " \
"Supported choices: %s" % browsers,
choices=browsers)
Plugin.options(self, parser, env)
def options(self, parser, env):
browsers = Browser.get_supported_browsers()
parser.add_option('-b',
'--browser',
help="Specify browser by using initials. "
"If value was not passed then 'ff' will be used. "
"Supported choices: %s" % browsers,
choices=browsers)
Plugin.options(self, parser, env)
def __open_url(self, evt):
url = self.__txt_url.GetValue()
if StringUtils.is_url_correct(url):
self.bth_open_url.Disable()
name = self.get_browser_initials()
try:
if self.__browser and self.__browser.get_browser_initials() != name:
self.__browser.quit()
self.__browser = Browser(name)
elif not self.__browser:
self.__browser = Browser(name)
except Exception:
show_error_dialog(self, traceback.format_exc(), u'Failed to open browser')
self.__browser = None
if self.__browser:
self.__browser.open(url)
# TODO: if generator or selector -> load image
self.bth_open_url.Enable()
else:
show_dialog(self, u'Bad url: %s' % url, u'Bad url')
def __get_links(url):
links = []
browser = None
try:
browser = Browser('gc')
browser.get(url)
links += __get_links_from_current_page(browser)
links += __get_links_from_frames(browser)
except:
try:
browser.save_screenshot()
except:
pass
traceback.print_exc()
finally:
if browser:
browser.quit()
return links
def __init__(self, *args, **kwargs):
Panel.__init__(self, *args, **kwargs)
sizer = GridBagSizer(5, 5)
row = 0
col = 0
self.cb_html_output = CheckBox(self, label=u'Report in HTML')
self.cb_html_output.Bind(EVT_CHECKBOX, self.__on_check)
sizer.Add(self.cb_html_output,
pos=(row, col),
flag=FLAG_ALL_AND_EXPAND)
col += 1
self.txt_html_report = TextCtrl(self, style=TE_READONLY)
self.txt_html_report.Disable()
sizer.Add(self.txt_html_report,
pos=(row, col),
span=(1, 3),
flag=FLAG_ALL_AND_EXPAND)
col += 3
self.btn_select_html = Button(self, label=u'Select HTML file')
self.btn_select_html.Disable()
self.btn_select_html.Bind(EVT_BUTTON, self.__on_select_file)
sizer.Add(self.btn_select_html,
pos=(row, col),
flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
self.cb_xml_output = CheckBox(self, label=u'Report in XML')
self.cb_xml_output.Bind(EVT_CHECKBOX, self.__on_check)
sizer.Add(self.cb_xml_output, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.txt_xml_report = TextCtrl(self, style=TE_READONLY)
self.txt_xml_report.Disable()
sizer.Add(self.txt_xml_report,
pos=(row, col),
span=(1, 3),
flag=FLAG_ALL_AND_EXPAND)
col += 3
self.btn_select_xml = Button(self, label=u'Select XML file')
self.btn_select_xml.Disable()
self.btn_select_xml.Bind(EVT_BUTTON, self.__on_select_file)
sizer.Add(self.btn_select_xml,
pos=(row, col),
flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
self.cb_options = CheckBox(self, label=u'Additional options')
self.cb_options.Bind(EVT_CHECKBOX, self.__on_check)
sizer.Add(self.cb_options, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.txt_options = TextCtrl(self)
self.txt_options.Disable()
sizer.Add(self.txt_options,
pos=(row, col),
span=(1, 3),
flag=FLAG_ALL_AND_EXPAND)
col += 3
self.btn_nose_help = Button(self, label=u'Show help')
self.btn_nose_help.Bind(EVT_BUTTON, self.__on_show_help)
sizer.Add(self.btn_nose_help, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
self.btn_load_tests_from_files = Button(self,
label=u'Load tests from files')
self.btn_load_tests_from_files.Bind(EVT_BUTTON,
self.__load_tests_from_files)
sizer.Add(self.btn_load_tests_from_files,
pos=(row, col),
flag=FLAG_ALL_AND_EXPAND)
col += 1
self.btn_load_tests_from_dir = Button(
self, label=u'Load tests from directory')
self.btn_load_tests_from_dir.Bind(EVT_BUTTON,
self.__load_tests_from_directory)
sizer.Add(self.btn_load_tests_from_dir,
pos=(row, col),
flag=FLAG_ALL_AND_EXPAND)
col += 1
dummy_label = StaticText(self)
sizer.Add(dummy_label, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.cb_browser = Choice(self,
choices=Browser.get_supported_browsers())
self.cb_browser.Select(0)
sizer.Add(self.cb_browser, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.btn_run = Button(self, label=u'Run test cases')
self.btn_run.Bind(EVT_BUTTON, self.__run_tests)
sizer.Add(self.btn_run, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
window = SplitterWindow(self, style=SP_3D | SP_LIVE_UPDATE)
self.tree_ctrl = CustomTreeCtrl(
window,
style=TR_SINGLE | TR_HAS_BUTTONS | TR_AUTO_CHECK_CHILD
| TR_AUTO_CHECK_PARENT | TR_AUTO_TOGGLE_CHILD)
self.tree_ctrl.SetBackgroundColour(self.GetBackgroundColour())
self.tree_ctrl.SetForegroundColour(self.GetForegroundColour())
self.tree_ctrl.Bind(EVT_TREE_ITEM_CHECKED, self.__on_tree_check)
self.txt_ctrl = TextCtrl(window,
style=TE_MULTILINE | TE_READONLY | HSCROLL
| VSCROLL)
font_size = self.txt_ctrl.GetFont().GetPointSize()
self.txt_ctrl.SetFont(
Font(font_size, FONTFAMILY_TELETYPE, NORMAL, NORMAL))
window.SplitVertically(self.tree_ctrl, self.txt_ctrl)
sizer.Add(window,
pos=(row, col),
span=(1, 5),
flag=FLAG_ALL_AND_EXPAND)
sizer.AddGrowableCol(2, 1)
sizer.AddGrowableRow(row, 1)
self.SetSizerAndFit(sizer)
def __init__(self, *args, **kwargs):
Panel.__init__(self, *args, **kwargs)
sizer = GridBagSizer(5, 5)
row = 0
col = 0
self.cb_html_output = CheckBox(self, label=u'Report in HTML')
self.cb_html_output.Bind(EVT_CHECKBOX, self.__on_check)
sizer.Add(self.cb_html_output, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.txt_html_report = TextCtrl(self, style=TE_READONLY)
self.txt_html_report.Disable()
sizer.Add(self.txt_html_report, pos=(row, col), span=(1, 3), flag=FLAG_ALL_AND_EXPAND)
col += 3
self.btn_select_html = Button(self, label=u'Select HTML file')
self.btn_select_html.Disable()
self.btn_select_html.Bind(EVT_BUTTON, self.__on_select_file)
sizer.Add(self.btn_select_html, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
self.cb_xml_output = CheckBox(self, label=u'Report in XML')
self.cb_xml_output.Bind(EVT_CHECKBOX, self.__on_check)
sizer.Add(self.cb_xml_output, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.txt_xml_report = TextCtrl(self, style=TE_READONLY)
self.txt_xml_report.Disable()
sizer.Add(self.txt_xml_report, pos=(row, col), span=(1, 3), flag=FLAG_ALL_AND_EXPAND)
col += 3
self.btn_select_xml = Button(self, label=u'Select XML file')
self.btn_select_xml.Disable()
self.btn_select_xml.Bind(EVT_BUTTON, self.__on_select_file)
sizer.Add(self.btn_select_xml, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
self.cb_options = CheckBox(self, label=u'Additional options')
self.cb_options.Bind(EVT_CHECKBOX, self.__on_check)
sizer.Add(self.cb_options, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.txt_options = TextCtrl(self)
self.txt_options.Disable()
sizer.Add(self.txt_options, pos=(row, col), span=(1, 3), flag=FLAG_ALL_AND_EXPAND)
col += 3
self.btn_nose_help = Button(self, label=u'Show help')
self.btn_nose_help.Bind(EVT_BUTTON, self.__on_show_help)
sizer.Add(self.btn_nose_help, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
self.btn_load_tests_from_files = Button(self, label=u'Load tests from files')
self.btn_load_tests_from_files.Bind(EVT_BUTTON, self.__load_tests_from_files)
sizer.Add(self.btn_load_tests_from_files, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.btn_load_tests_from_dir = Button(self, label=u'Load tests from directory')
self.btn_load_tests_from_dir.Bind(EVT_BUTTON, self.__load_tests_from_directory)
sizer.Add(self.btn_load_tests_from_dir, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
dummy_label = StaticText(self)
sizer.Add(dummy_label, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.cb_browser = Choice(self, choices=Browser.get_supported_browsers())
self.cb_browser.Select(0)
sizer.Add(self.cb_browser, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.btn_run = Button(self, label=u'Run test cases')
self.btn_run.Bind(EVT_BUTTON, self.__run_tests)
sizer.Add(self.btn_run, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
window = SplitterWindow(self, style=SP_3D | SP_LIVE_UPDATE)
self.tree_ctrl = CustomTreeCtrl(window,
style=TR_SINGLE | TR_HAS_BUTTONS | TR_AUTO_CHECK_CHILD | TR_AUTO_CHECK_PARENT | TR_AUTO_TOGGLE_CHILD)
self.tree_ctrl.SetBackgroundColour(self.GetBackgroundColour())
self.tree_ctrl.SetForegroundColour(self.GetForegroundColour())
self.tree_ctrl.Bind(EVT_TREE_ITEM_CHECKED, self.__on_tree_check)
self.txt_ctrl = TextCtrl(window, style=TE_MULTILINE | TE_READONLY | HSCROLL | VSCROLL)
font_size = self.txt_ctrl.GetFont().GetPointSize()
self.txt_ctrl.SetFont(Font(font_size, FONTFAMILY_TELETYPE, NORMAL, NORMAL))
window.SplitVertically(self.tree_ctrl, self.txt_ctrl)
sizer.Add(window, pos=(row, col), span=(1, 5), flag=FLAG_ALL_AND_EXPAND)
sizer.AddGrowableCol(2, 1)
sizer.AddGrowableRow(row, 1)
self.SetSizerAndFit(sizer)
def __create_widgets(self):
panel = Panel(self)
sizer = GridBagSizer(5, 5)
row = 0
col = 0
label = StaticText(panel, label=u'Root folder:')
sizer.Add(label, pos=(row, col))
col += 1
col_width = 4
self.__txt_root_path = TextCtrl(panel)
sizer.Add(self.__txt_root_path,
pos=(row, col),
span=(1, col_width),
flag=FLAG_ALL_AND_EXPAND)
col += col_width
self.__btn_set_root = Button(panel, label=u'Set root folder')
self.__btn_set_root.Bind(EVT_BUTTON, self.__set_root_folder)
sizer.Add(self.__btn_set_root,
pos=(row, col),
flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
label = StaticText(panel, label=u'Url:')
sizer.Add(label, pos=(row, col))
col += 1
self.__txt_url = TextCtrl(
panel, value=u'https://www.google.com/') # TODO: remove url
sizer.Add(self.__txt_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
label = StaticText(panel, label=u'Browser:')
sizer.Add(label, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.__cb_browser = Choice(panel,
choices=Browser.get_supported_browsers())
self.__cb_browser.Select(0)
sizer.Add(self.__cb_browser, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_open_url = Button(panel, label=u'Open url')
self.bth_open_url.Bind(EVT_BUTTON, self.__open_url)
sizer.Add(self.bth_open_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_close_url = Button(panel, label=u'Close browser')
self.bth_close_url.Bind(EVT_BUTTON, self.__close_browser)
sizer.Add(self.bth_close_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
tabs = Tabs(panel, [(GeneratorTab, "Generator"), (EditorTab, "Editor"),
(TestRunnerTab, "Test runner"),
(SelectorFinderTab, "Selector finder")])
sizer.Add(tabs, pos=(row, col), span=(1, 6), flag=FLAG_ALL_AND_EXPAND)
sizer.AddGrowableCol(1, 1)
sizer.AddGrowableRow(2, 1)
panel.SetSizer(sizer)
self.Layout()
class MainFrame(Frame):
def __init__(self, parent):
Frame.__init__(self, parent)
self.__browser = None
self.__tmp_dir = mkdtemp()
self.SetTitle("Easy Selenium UI")
self.SetSize((800, 600))
self.__create_widgets()
self.Bind(EVT_CLOSE, self.__on_close)
def __create_widgets(self):
panel = Panel(self)
sizer = GridBagSizer(5, 5)
row = 0
col = 0
label = StaticText(panel, label=u'Root folder:')
sizer.Add(label, pos=(row, col))
col += 1
col_width = 4
self.__txt_root_path = TextCtrl(panel)
sizer.Add(self.__txt_root_path,
pos=(row, col),
span=(1, col_width),
flag=FLAG_ALL_AND_EXPAND)
col += col_width
self.__btn_set_root = Button(panel, label=u'Set root folder')
self.__btn_set_root.Bind(EVT_BUTTON, self.__set_root_folder)
sizer.Add(self.__btn_set_root,
pos=(row, col),
flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
label = StaticText(panel, label=u'Url:')
sizer.Add(label, pos=(row, col))
col += 1
self.__txt_url = TextCtrl(
panel, value=u'https://www.google.com/') # TODO: remove url
sizer.Add(self.__txt_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
label = StaticText(panel, label=u'Browser:')
sizer.Add(label, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.__cb_browser = Choice(panel,
choices=Browser.get_supported_browsers())
self.__cb_browser.Select(0)
sizer.Add(self.__cb_browser, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_open_url = Button(panel, label=u'Open url')
self.bth_open_url.Bind(EVT_BUTTON, self.__open_url)
sizer.Add(self.bth_open_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_close_url = Button(panel, label=u'Close browser')
self.bth_close_url.Bind(EVT_BUTTON, self.__close_browser)
sizer.Add(self.bth_close_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
tabs = Tabs(panel, [(GeneratorTab, "Generator"), (EditorTab, "Editor"),
(TestRunnerTab, "Test runner"),
(SelectorFinderTab, "Selector finder")])
sizer.Add(tabs, pos=(row, col), span=(1, 6), flag=FLAG_ALL_AND_EXPAND)
sizer.AddGrowableCol(1, 1)
sizer.AddGrowableRow(2, 1)
panel.SetSizer(sizer)
self.Layout()
def get_root_folder(self):
text = self.__txt_root_path.GetValue()
if text and len(text) > 0:
return text
else:
return None
def __set_root_folder(self, evt):
dialog = DirDialog(self)
if dialog.ShowModal() == ID_OK:
path = dialog.GetPath()
RootFolder.prepare_folder(path)
self.__txt_root_path.SetValue(path)
def __open_url(self, evt):
url = self.__txt_url.GetValue()
if StringUtils.is_url_correct(url):
self.bth_open_url.Disable()
name = self.get_browser_initials()
try:
if self.__browser and self.__browser.get_browser_initials(
) != name:
self.__browser.quit()
self.__browser = Browser(name)
elif not self.__browser:
self.__browser = Browser(name)
except Exception:
show_error_dialog(self, traceback.format_exc(),
u'Failed to open browser')
self.__browser = None
if self.__browser:
self.__browser.open(url)
# TODO: if generator or selector -> load image
self.bth_open_url.Enable()
else:
show_dialog(self, u'Bad url: %s' % url, u'Bad url')
def __close_browser(self, evt):
if self.__browser:
self.__browser.quit()
self.__browser = None
def __on_close(self, evt):
self.__close_browser(evt)
shutil.rmtree(self.__tmp_dir)
self.Destroy()
def get_url(self):
return self.__txt_url.GetValue()
def set_url(self, url):
self.__txt_url.SetValue(url)
def get_browser(self):
return self.__browser
def get_browser_initials(self):
return self.__cb_browser.GetStringSelection()
def get_tmp_dir(self):
return self.__tmp_dir
def setUpClass(cls):
super(BaseTest, cls).setUpClass()
if cls.BROWSER_NAME and not Browser.DEFAULT_BROWSER:
Browser.DEFAULT_BROWSER = cls.BROWSER_NAME
cls.browser = Browser(logger=cls.logger)
class MainFrame(Frame):
def __init__(self, parent):
Frame.__init__(self, parent)
self.__browser = None
self.__tmp_dir = mkdtemp()
self.SetTitle("Easy Selenium UI")
self.SetSize((800, 600))
self.__create_widgets()
self.Bind(EVT_CLOSE, self.__on_close)
def __create_widgets(self):
panel = Panel(self)
sizer = GridBagSizer(5, 5)
row = 0
col = 0
label = StaticText(panel, label=u'Root folder:')
sizer.Add(label, pos=(row, col))
col += 1
col_width = 4
self.__txt_root_path = TextCtrl(panel)
sizer.Add(self.__txt_root_path, pos=(row, col), span=(1, col_width), flag=FLAG_ALL_AND_EXPAND)
col += col_width
self.__btn_set_root = Button(panel, label=u'Set root folder')
self.__btn_set_root.Bind(EVT_BUTTON, self.__set_root_folder)
sizer.Add(self.__btn_set_root, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
label = StaticText(panel, label=u'Url:')
sizer.Add(label, pos=(row, col))
col += 1
self.__txt_url = TextCtrl(panel, value=u'https://www.google.com/') # TODO: remove url
sizer.Add(self.__txt_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
label = StaticText(panel, label=u'Browser:')
sizer.Add(label, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.__cb_browser = Choice(panel, choices=Browser.get_supported_browsers())
self.__cb_browser.Select(0)
sizer.Add(self.__cb_browser, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_open_url = Button(panel, label=u'Open url')
self.bth_open_url.Bind(EVT_BUTTON, self.__open_url)
sizer.Add(self.bth_open_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
col += 1
self.bth_close_url = Button(panel, label=u'Close browser')
self.bth_close_url.Bind(EVT_BUTTON, self.__close_browser)
sizer.Add(self.bth_close_url, pos=(row, col), flag=FLAG_ALL_AND_EXPAND)
row += 1
col = 0
tabs = Tabs(panel, [(GeneratorTab, "Generator"),
(EditorTab, "Editor"),
(TestRunnerTab, "Test runner"),
(SelectorFinderTab, "Selector finder")])
sizer.Add(tabs, pos=(row, col), span=(1, 6), flag=FLAG_ALL_AND_EXPAND)
sizer.AddGrowableCol(1, 1)
sizer.AddGrowableRow(2, 1)
panel.SetSizer(sizer)
self.Layout()
def get_root_folder(self):
text = self.__txt_root_path.GetValue()
if text and len(text) > 0:
return text
else:
return None
def __set_root_folder(self, evt):
dialog = DirDialog(self)
if dialog.ShowModal() == ID_OK:
path = dialog.GetPath()
RootFolder.prepare_folder(path)
self.__txt_root_path.SetValue(path)
def __open_url(self, evt):
url = self.__txt_url.GetValue()
if StringUtils.is_url_correct(url):
self.bth_open_url.Disable()
name = self.get_browser_initials()
try:
if self.__browser and self.__browser.get_browser_initials() != name:
self.__browser.quit()
self.__browser = Browser(name)
elif not self.__browser:
self.__browser = Browser(name)
except Exception:
show_error_dialog(self, traceback.format_exc(), u'Failed to open browser')
self.__browser = None
if self.__browser:
self.__browser.open(url)
# TODO: if generator or selector -> load image
self.bth_open_url.Enable()
else:
show_dialog(self, u'Bad url: %s' % url, u'Bad url')
def __close_browser(self, evt):
if self.__browser:
self.__browser.quit()
self.__browser = None
def __on_close(self, evt):
self.__close_browser(evt)
shutil.rmtree(self.__tmp_dir)
self.Destroy()
def get_url(self):
return self.__txt_url.GetValue()
def set_url(self, url):
self.__txt_url.SetValue(url)
def get_browser(self):
return self.__browser
def get_browser_initials(self):
return self.__cb_browser.GetStringSelection()
def get_tmp_dir(self):
return self.__tmp_dir
