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
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
