class EditDialog(Dialog):
__hosts = None
__window = None
def __init__(self, parent, dpi=(1, 1)):
Dialog.__init__(self,
parent,
id=ID_ANY,
title=u"编辑/添加Hosts",
pos=Point(600, 600),
size=Size(394 * dpi[0], 210 * dpi[1]),
style=DEFAULT_DIALOG_STYLE | FRAME_FLOAT_ON_PARENT)
self.__window = parent
self.SetSizeHints(DefaultSize, DefaultSize)
font = Font(10, FONTFAMILY_DEFAULT, FONTSTYLE_NORMAL,
FONTWEIGHT_NORMAL, False, EmptyString)
inputSize = Size(260 * dpi[0], -1)
self.SetFont(font)
fgSizer3 = FlexGridSizer(0, 2, 0, 0)
fgSizer3.SetFlexibleDirection(BOTH)
fgSizer3.SetNonFlexibleGrowMode(FLEX_GROWMODE_SPECIFIED)
self.localRadio = RadioButton(self, ID_ANY, u"本地Hosts",
DefaultPosition, DefaultSize, 0)
self.localRadio.SetFont(font)
fgSizer3.Add(self.localRadio, 0, ALL, 5)
self.localRadio.Bind(EVT_RADIOBUTTON, self.OnRadioChange)
self.onlineRadio = RadioButton(self, ID_ANY, u"在线Hosts",
DefaultPosition, DefaultSize, 0)
fgSizer3.Add(self.onlineRadio, 0, ALL, 5)
self.onlineRadio.Bind(EVT_RADIOBUTTON, self.OnRadioChange)
self.m_staticText4 = StaticText(self, ID_ANY, u"名称", DefaultPosition,
DefaultSize, 0)
self.m_staticText4.Wrap(-1)
fgSizer3.Add(self.m_staticText4, 0, ALL, 5)
self.nameInput = TextCtrl(self, ID_ANY, EmptyString, DefaultPosition,
inputSize, 0)
fgSizer3.Add(self.nameInput, 0, ALL, 5)
self.m_staticText5 = StaticText(self, ID_ANY, u"地址", DefaultPosition,
DefaultSize, 0)
self.m_staticText5.Wrap(-1)
fgSizer3.Add(self.m_staticText5, 0, ALL, 5)
self.urlInput = TextCtrl(self, ID_ANY, u"http://", DefaultPosition,
inputSize, 0)
fgSizer3.Add(self.urlInput, 0, ALL, 5)
self.m_staticText3 = StaticText(self, ID_ANY, u"图标", DefaultPosition,
DefaultSize, 0)
self.m_staticText3.Wrap(-1)
fgSizer3.Add(self.m_staticText3, 0, ALL, 5)
self.iconComboBox = ComboBox(self, ID_ANY, u"请选择图标", DefaultPosition,
inputSize, list(GetIcons().keys()), 0)
self.iconComboBox.SetFont(font)
fgSizer3.Add(self.iconComboBox, 0, ALL, 5)
self.cancelButton = Button(self, ID_ANY, u"取消", DefaultPosition,
DefaultSize, 0)
fgSizer3.Add(self.cancelButton, 0, ALL, 5)
self.saveButton = Button(self, ID_ANY, u"保存", DefaultPosition,
DefaultSize, 0)
fgSizer3.Add(self.saveButton, 0, ALL, 5)
self.SetSizer(fgSizer3)
self.Layout()
self.Centre(BOTH)
self.cancelButton.Bind(EVT_BUTTON, self.OnButtonClicked)
self.saveButton.Bind(EVT_BUTTON, self.OnButtonClicked)
self.Bind(EVT_CLOSE, self.OnClose)
def __del__(self):
pass
def OnClose(self, event):
self.SetHosts(None)
event.Skip()
def SetHosts(self, hosts):
self.__hosts = hosts
if hosts:
if hosts["url"] and len(hosts["url"] > 0):
self.onlineRadio.SetValue(hosts["url"])
self.localRadio.SetValue(not hosts["url"])
self.nameInput.SetValue(hosts["name"])
self.urlInput.SetValue(hosts["url"] or "")
self.iconComboBox.SetValue(hosts["icon"])
self.onlineRadio.Enable(False)
self.localRadio.Enable(False)
else:
self.onlineRadio.Enable(False)
self.localRadio.Enable(True)
self.localRadio.SetValue(True)
self.urlInput.SetValue("")
self.iconComboBox.SetValue("logo")
self.nameInput.SetValue("")
def OnButtonClicked(self, event):
if event.GetId() == self.cancelButton.GetId():
self.Close()
return
name = self.nameInput.GetValue()
url = self.urlInput.GetValue()
isOnline = self.onlineRadio.GetValue()
if not isOnline:
url = None
if not name or len(name) < 1:
MessageBox("请输入Hosts名称", "提示", ICON_WARNING)
elif isOnline and (not url or len(url) < 1):
MessageBox("请输入在线Hosts地址", "提示", ICON_WARNING)
else:
if not self.iconComboBox.GetValue():
self.iconComboBox.SetValue("logo")
if self.__hosts:
self.__hosts["name"] = name
self.__hosts["url"] = url
self.__hosts["lastUpdateTime"] = Now()
self.__hosts["icon"] = self.iconComboBox.GetValue()
hostsId = self.__hosts['id']
else:
hostsId = 0x1994 + len(Settings.settings["hosts"])
Settings.settings["hosts"].append(
hostsDict(hostsId,
name,
url=url,
lastUpdateTime=Now(),
content="# Created by mHosts v%s, %s\n" %
(Settings.version(), Now()),
icon=self.iconComboBox.GetValue()))
Settings.Save()
self.__window.InitHostsTree(select=hostsId)
self.Close()
def OnRadioChange(self, event):
self.urlInput.Enable(event.GetId() == self.onlineRadio.GetId())
class EditorTab(Panel):
def __init__(self, parent):
Panel.__init__(self, parent)
sizer = GridBagSizer(5, 5)
self.SetSizer(sizer)
self.__cur_po_class = None
self.__create_widgets()
def __create_widgets(self):
sizer = self.GetSizer()
# Next row
inner_sizer = BoxSizer(HORIZONTAL)
label = StaticText(self, label=u'Class path:')
inner_sizer.Add(label, flag=ALL)
self.cb_class_path = ComboBox(self, style=CB_READONLY)
self.cb_class_path.Bind(EVT_COMBOBOX, self.__on_load_po_class)
inner_sizer.Add(self.cb_class_path, 1, flag=FLAG_ALL_AND_EXPAND)
self.btn_reload = Button(self, label=u'Reload')
self.btn_reload.Bind(EVT_BUTTON, self.__on_load_po_class)
inner_sizer.Add(self.btn_reload, flag=ALL)
self.btn_open_class = Button(self, label=u'Open class')
self.btn_open_class.Bind(EVT_BUTTON, self.__open_class)
inner_sizer.Add(self.btn_open_class, flag=ALL)
row = 0
sizer.Add(inner_sizer, pos=(row, 0), flag=FLAG_ALL_AND_EXPAND)
# Next row
row += 1
splitter = SplitterWindow(self, style=SP_3D | SP_LIVE_UPDATE)
self.image_panel = ImageWithElements(splitter)
self.image_panel.static_bitmap.Bind(EVT_MOTION, self.__on_mouse_move)
self.image_panel.static_bitmap.Bind(EVT_RIGHT_DOWN, self.__on_right_click)
self.table_and_test_file_tabs = FieldsTableAndTestFilesTabs(splitter, self)
splitter.SplitHorizontally(self.image_panel, self.table_and_test_file_tabs)
sizer.Add(splitter, pos=(row, 0), flag=FLAG_ALL_AND_EXPAND)
sizer.AddGrowableRow(row, 1)
sizer.AddGrowableCol(0, 1)
def __get_parsed_classes(self, field):
classes = ParsedPageObjectClass.get_parsed_classes(self.__cur_po_class.file_path)
if len(classes) > 0 and len(classes[0].methods) == 0:
classes = []
classes += [ParsedModule.get_parsed_module(tools)]
if field:
classes += ParsedMouseClass.get_parsed_classes()
classes += ParsedBrowserClass.get_parsed_classes()
return classes
def show_content_menu(self, field):
tabs = self.table_and_test_file_tabs.tabs
count = tabs.GetPageCount()
if count > 1:
selected_tab = tabs.GetPage(tabs.GetSelection())
if type(selected_tab) in (TestFileUI, PyFileUI):
file_path = selected_tab.get_file_path()
txt_ctrl_ui = tabs.GetPage(tabs.GetSelection())
parsed_classes = self.__get_parsed_classes(field)
context_menu = FieldContextMenu(field,
parsed_classes,
file_path,
txt_ctrl_ui)
self.PopupMenu(context_menu)
context_menu.Destroy()
else:
show_dialog(self, u'Please select tab with test file.',
u'Tab with test file was not selected')
else:
show_dialog(self, u'Please create/open test file.',
u'Test file was not created/opened')
def __on_load_po_class(self, evt):
path = self.cb_class_path.GetValue()
if len(path) > 0:
self.__load_po_class(path)
def __on_right_click(self, evt):
field = self.__get_current_field(evt)
self.show_content_menu(field)
def __on_mouse_move(self, evt):
if self.__cur_po_class:
ImageAndTableHelper.select_field_on_mouse_move(
evt,
self.__cur_po_class.fields,
self.image_panel,
self.table_and_test_file_tabs.table
)
def __get_current_field(self, evt):
return self.image_panel.get_field(evt.GetPosition())
def __open_class(self, evt):
folder = self.GetTopLevelParent().get_root_folder()
if folder:
if RootFolder.PO_FOLDER in os.listdir(folder):
folder = os.path.join(folder, RootFolder.PO_FOLDER)
dialog = FileDialog(self, defaultDir=folder, wildcard=u'*.py')
if dialog.ShowModal() == ID_OK:
self.__load_po_class(dialog.GetPath())
else:
show_dialog_path_doesnt_exist(self, folder)
def __load_po_class(self, path):
self.table_and_test_file_tabs.table.clear_table()
if not os.path.exists(path):
show_dialog_path_doesnt_exist(self, path)
if not is_correct_python_file(path):
show_dialog(self, u'File name is incorrect: %s' % path,
u'Bad file name')
else:
folder = os.path.dirname(path)
files = [os.path.join(folder, p) for p in os.listdir(folder)
if is_correct_python_file(p)]
self.cb_class_path.Clear()
self.cb_class_path.AppendItems(files)
self.cb_class_path.Select(files.index(path))
try:
self.__cur_po_class = PageObjectClass.parse_string_to_po_class(read_file(path))
area = self.__cur_po_class.area
self.image_panel.set_po_fields(self.__cur_po_class.fields)
self.image_panel.load_image(self.__cur_po_class.img_path, area)
self.cb_class_path.SetValue(self.__cur_po_class.file_path)
self.table_and_test_file_tabs.load_po_class(self.__cur_po_class)
except Exception:
self.__cur_po_class = None
show_error_dialog(self, traceback.format_exc(),
u'Failed to open file %s' % path)