class LogMonitor(wx.Frame):
def __init__(self):
wx.Frame.__init__(self, None, id=wx.ID_ANY, title='', size=(600, 400))
self.__init_menu_bar()
self.panel = wx.Panel(self)
main_sizer = wx.BoxSizer(wx.VERTICAL)
self.__filters = [
String.LEVEL_INFO, String.LEVEL_WARM, String.LEVEL_ERROR,
String.LEVEL_RESULT
]
filter_sizer = self.__init_filter_sizer()
self.list_view = FastObjectListView(self.panel,
wx.ID_ANY,
style=wx.LC_REPORT | wx.LC_HRULES
| wx.LC_VRULES | wx.SUNKEN_BORDER
| wx.LC_NO_SORT_HEADER)
main_sizer.Add(filter_sizer, 0, wx.EXPAND | wx.ALL, 1)
main_sizer.Add(self.list_view, 1, wx.EXPAND | wx.ALL, 1)
self.panel.SetSizer(main_sizer)
self.Layout()
self.__set_columns()
self.__set_row_formatter()
self.__set_filter()
self.__log_file = None
self.__logs = list()
self.__tmp_logs = list()
self.__timer = wx.Timer(self)
self.Bind(wx.EVT_TIMER, self.AppendLogs, self.__timer)
self.__timer.Start(500)
self.Bind(wx.EVT_CLOSE, self.on_close)
def __init_menu_bar(self):
menu_bar = wx.MenuBar()
file_menu = wx.Menu()
self.record_log_menu = wx.MenuItem(file_menu,
wx.ID_SAVEAS,
text=u"记录日志",
kind=wx.ITEM_CHECK)
save_log = wx.MenuItem(file_menu,
wx.ID_SAVE,
text=u"保存当前日志",
kind=wx.ITEM_NORMAL)
file_menu.Append(self.record_log_menu)
file_menu.Append(save_log)
menu_bar.Append(file_menu, '&File')
self.Bind(wx.EVT_MENU, self.menu_handler)
view_menu = wx.Menu()
self.auto_scroll = wx.MenuItem(file_menu,
wx.ID_ANY,
text=u"自动滚动",
kind=wx.ITEM_CHECK)
view_menu.Append(self.auto_scroll)
menu_bar.Append(view_menu, '&View')
self.auto_scroll.Check()
self.SetMenuBar(menu_bar)
def menu_handler(self, event):
id = event.GetId()
if id == wx.ID_SAVEAS:
if self.record_log_menu.IsChecked():
self.__log_file = self.__get_log_file_path()
else:
self.__log_file = None
elif id == wx.ID_SAVE:
save_path = self.__get_save_file_path()
if save_path is not None:
with open(save_path, 'w') as f:
for log in self.__logs:
f.write(log.LINE)
def __get_log_file_path(self):
dlg = wx.FileDialog(self,
message="保存日志文件",
wildcard="Text (*.txt)|*.txt|"
"All files (*.*)|*.*",
style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT)
if dlg.ShowModal() == wx.ID_OK:
p = dlg.GetPath()
with open(p, 'w') as f:
pass
return p
else:
return None
def __get_save_file_path(self):
dlg = wx.FileDialog(self,
message="保存日志文件",
wildcard="Text (*.txt)|*.txt|"
"All files (*.*)|*.*",
style=wx.FD_SAVE | wx.FD_OVERWRITE_PROMPT)
if dlg.ShowModal() == wx.ID_OK:
return dlg.GetPath()
else:
return None
def __init_filter_sizer(self):
sizer = wx.BoxSizer(wx.HORIZONTAL)
def init_checkbox(level):
wx_checkbox = wx.CheckBox(self.panel,
wx.ID_ANY,
level,
wx.DefaultPosition,
wx.DefaultSize,
0,
name=level)
wx_checkbox.Bind(wx.EVT_CHECKBOX, self.set_filter)
if level in self.__filters:
wx_checkbox.SetValue(True)
return wx_checkbox
sizer.Add(init_checkbox(String.LEVEL_DEBUG), 0, wx.ALL, 5)
sizer.Add(init_checkbox(String.LEVEL_INFO), 0, wx.ALL, 5)
sizer.Add(init_checkbox(String.LEVEL_WARM), 0, wx.ALL, 5)
sizer.Add(init_checkbox(String.LEVEL_ERROR), 0, wx.ALL, 5)
sizer.Add(init_checkbox(String.LEVEL_RESULT), 0, wx.ALL, 5)
return sizer
def set_filter(self, event):
object = event.GetEventObject()
if object.IsChecked():
self.__filters.append(object.GetName())
else:
self.__filters.remove(object.GetName())
self.__set_filter()
def __set_filter(self):
_filter = Filter.Chain(
Filter.Predicate(lambda item: item._level in self.__filters))
self.list_view.SetFilter(Filter.Chain(_filter))
self.list_view.RepopulateList()
def Info(self, index, msg):
self.__append_log(LogData(runs=index, level=String.LEVEL_INFO,
msg=msg))
def Error(self, index, msg):
self.__append_log(
LogData(runs=index, level=String.LEVEL_ERROR, msg=msg))
def Result(self, index, msg):
self.__append_log(
LogData(runs=index, level=String.LEVEL_RESULT, msg=msg))
def Warm(self, index, msg):
self.__append_log(LogData(runs=index, level=String.LEVEL_WARM,
msg=msg))
def Debug(self, index, msg):
self.__append_log(
LogData(runs=index, level=String.LEVEL_DEBUG, msg=msg))
def __append_log(self, log):
if self.__log_file is not None:
self.write_log(log=log)
self.__logs.append(log)
self.__tmp_logs.append(log)
def write_log(self, log):
with open(self.__log_file, 'a') as f:
f.write(log.LINE)
def AppendLogs(self, event):
if self.__tmp_logs:
self.list_view.AddObjects(self.__tmp_logs)
if self.auto_scroll.IsChecked():
self.list_view.SelectObject(self.__tmp_logs[-1],
ensureVisible=True)
self.__tmp_logs = []
def __set_columns(self):
self.list_view.SetColumns([
ColumnDefn(title=u"", align="right", width=0, valueGetter=''),
ColumnDefn(title=u"Runs",
align="right",
width=70,
valueGetter='_runs'),
ColumnDefn(title=u"Time",
align="center",
width=100,
valueGetter='_time'),
ColumnDefn(title=u"Level",
align="center",
width=70,
valueGetter='_level'),
ColumnDefn(title=u"Message",
align="left",
minimumWidth=150,
valueGetter='_msg',
isSpaceFilling=True),
])
def __set_row_formatter(self):
self.list_view.rowFormatter = self.row_formatter
@staticmethod
def row_formatter(list_view, item):
if item.LEVEL == String.LEVEL_INFO:
list_view.SetBackgroundColour(Color.LemonChiffon)
elif item.LEVEL == String.LEVEL_DEBUG:
list_view.SetBackgroundColour(Color.DarkGrey)
elif item.LEVEL == String.LEVEL_WARM:
list_view.SetBackgroundColour(Color.Coral)
elif item.LEVEL == String.LEVEL_ERROR:
list_view.SetBackgroundColour(Color.OrangeRed)
elif item.LEVEL == String.LEVEL_RESULT:
list_view.SetBackgroundColour(Color.LightCyan)
if item.MESSAGE == u"测试成功":
list_view.SetTextColour(Color.GoogleGreen)
elif item.MESSAGE == u"测试失败":
list_view.SetTextColour(Color.GoogleRed)
elif item.MESSAGE == u"测试异常":
list_view.SetTextColour(Color.GoogleYellow)
else:
list_view.SetTextColour(Color.GoogleBlue)
else:
list_view.SetBackgroundColour(Color.White)
def on_close(self, event):
if self.IsShown():
self.Show(show=False)
return True
def Destroy(self):
self.__timer.Stop()
return super(wx.Frame, self).Destroy()
