import wx import sys from wx import StatusBar as wxStatusB from wx.lib import newevent import wx.richtext from sas.guiframe.gui_style import GUIFRAME_ICON # Number of fields on the status bar NB_FIELDS = 4 #position of the status bar's fields ICON_POSITION = 0 MSG_POSITION = 1 GAUGE_POSITION = 2 CONSOLE_POSITION = 3 BUTTON_SIZE = 40 STATUS_BAR_ICON_SIZE = 12 CONSOLE_WIDTH = 500 CONSOLE_HEIGHT = 300 if sys.platform.count("win32") > 0: FONT_VARIANT = 0 else: FONT_VARIANT = 1 GREEN = wx.Colour(95, 190, 95) YELLOW = wx.Colour(247, 214, 49) RED = wx.Colour(234, 89, 78) class ConsolePanel(wx.Panel): """ """ def __init__(self, parent, *args, **kwargs): """ """ wx.Panel.__init__(self, parent=parent, *args, **kwargs) self.parent = parent self.sizer = wx.BoxSizer(wx.VERTICAL) self.msg_txt = wx.richtext.RichTextCtrl(self, size=(CONSOLE_WIDTH-40, CONSOLE_HEIGHT-60), style=wx.VSCROLL|wx.HSCROLL|wx.NO_BORDER) self.msg_txt.SetEditable(False) self.msg_txt.SetValue('No message available') self.sizer.Add(self.msg_txt, 1, wx.EXPAND|wx.ALL, 10) self.SetSizer(self.sizer) def set_message(self, status="", event=None): """ """ status = str(status) if status.strip() == "": return color = (0, 0, 0) #black icon_bmp = wx.ArtProvider.GetBitmap(wx.ART_INFORMATION, wx.ART_TOOLBAR) if hasattr(event, "info"): icon_type = event.info.lower() if icon_type == "warning": color = (0, 0, 255) # blue icon_bmp = wx.ArtProvider.GetBitmap(wx.ART_WARNING, wx.ART_TOOLBAR) if icon_type == "error": color = (255, 0, 0) # red icon_bmp = wx.ArtProvider.GetBitmap(wx.ART_ERROR, wx.ART_TOOLBAR) if icon_type == "info": icon_bmp = wx.ArtProvider.GetBitmap(wx.ART_INFORMATION, wx.ART_TOOLBAR) self.msg_txt.Newline() self.msg_txt.WriteBitmap(icon_bmp) self.msg_txt.BeginTextColour(color) self.msg_txt.WriteText("\t") self.msg_txt.AppendText(status) self.msg_txt.EndTextColour() class Console(wx.Frame): """ """ def __init__(self, parent=None, status="", *args, **kwds): kwds["size"] = (CONSOLE_WIDTH, CONSOLE_HEIGHT) kwds["title"] = "Console" wx.Frame.__init__(self, parent=parent, *args, **kwds) self.SetWindowVariant(FONT_VARIANT) self.panel = ConsolePanel(self) self.panel.set_message(status=status) wx.EVT_CLOSE(self, self.Close) def set_multiple_messages(self, messages=[]): """ """ if messages: for status in messages: self.panel.set_message(status=status) def set_message(self, status, event=None): """ """ self.panel.set_message(status=str(status), event=event) def Close(self, event): """ """ self.Hide() class StatusBar(wxStatusB): """ Application status bar """ def __init__(self, parent, id): wxStatusB.__init__(self, parent, id) self.parent = parent self.parent.SetStatusBarPane(MSG_POSITION) #Layout of status bar width = STATUS_BAR_ICON_SIZE height = STATUS_BAR_ICON_SIZE self.SetFieldsCount(NB_FIELDS) # Leave some space for the resize handle in the last field console_btn_width = 80 self.SetStatusWidths([width+4, -2, -1, width+console_btn_width]) self.SetMinHeight(height) #display default message self.msg_position = MSG_POSITION # Create progress bar gauge_width = 5 * width self.gauge = wx.Gauge(self, size=(gauge_width, height), style=wx.GA_HORIZONTAL) self.gauge.Hide() # Create status bar icon reflecting the type of status # for the last message self.status_color = wx.StaticText(self, id=wx.NewId(), label=" ", size=wx.Size(15,15)) self.status_color.SetBackgroundColour(GREEN) self.status_color.SetForegroundColour(GREEN) # Create the button used to show the console dialog self.console_button = wx.Button(self, wx.NewId(), "Console", size=(console_btn_width, -1)) font = self.console_button.GetFont() _, pixel_h = font.GetPixelSize() font.SetPixelSize(wx.Size(0,int(pixel_h*0.9))) self.console_button.SetFont(font) self.console_button.SetToolTipString("History of status bar messages") self.console_button.Bind(wx.EVT_BUTTON, self._onMonitor, id=self.console_button.GetId()) self.reposition() ## Current progress value of the bar self.nb_start = 0 self.nb_progress = 0 self.nb_stop = 0 self.frame = None self.list_msg = [] self.frame = Console(parent=self) if hasattr(self.frame, "IsIconized"): if not self.frame.IsIconized(): try: icon = self.parent.GetIcon() self.frame.SetIcon(icon) except: try: FRAME_ICON = wx.Icon(GUIFRAME_ICON.FRAME_ICON_PATH, wx.BITMAP_TYPE_ICON) self.frame.SetIcon(FRAME_ICON) except: pass self.frame.set_multiple_messages(self.list_msg) self.frame.Hide() self.progress = 0 self.timer = wx.Timer(self, -1) self.timer_stop = wx.Timer(self, -1) self.thread = None self.Bind(wx.EVT_TIMER, self._on_time, self.timer) self.Bind(wx.EVT_TIMER, self._on_time_stop, self.timer_stop) self.Bind(wx.EVT_SIZE, self.OnSize) self.Bind(wx.EVT_IDLE, self.OnIdle) def reposition(self): """ Place the various fields in their proper position """ rect = self.GetFieldRect(GAUGE_POSITION) self.gauge.SetPosition((rect.x, rect.y)) rect = self.GetFieldRect(ICON_POSITION) self.status_color.SetPosition((rect.x, rect.y)) rect = self.GetFieldRect(CONSOLE_POSITION) self.console_button.SetPosition((rect.x, rect.y)) self.sizeChanged = False def OnIdle(self, event): """ """ if self.sizeChanged: self.reposition() def OnSize(self, evt): """ """ self.reposition() self.sizeChanged = True def get_msg_position(self): """ """ return self.msg_position def SetStatusText(self, text="", number=MSG_POSITION, event=None): """ """ wxStatusB.SetStatusText(self, text.split('\n',1)[0], number) self.list_msg.append(text) self.status_color.SetBackgroundColour(GREEN) self.status_color.SetForegroundColour(GREEN) if self.frame is not None : self.frame.set_message(status=text, event=event) def PopStatusText(self, *args, **kwds): """ Override status bar """ wxStatusB.PopStatusText(self, field=MSG_POSITION) def PushStatusText(self, *args, **kwds): """ PushStatusText """ text = "PushStatusText: What is this string?" wxStatusB.PushStatusText(self, field=MSG_POSITION, string=text) def enable_clear_gauge(self): """ clear the progress bar """ flag = True # Why we do this? #if (self.nb_start <= self.nb_stop) or \ # (self.nb_progress <= self.nb_stop): # flag = True return flag def _on_time_stop(self, evt): """ Clear the progress bar :param evt: wx.EVT_TIMER """ count = 0 while(count <= 100): count += 1 self.timer_stop.Stop() self.clear_gauge(msg="") self.nb_progress = 0 self.nb_start = 0 self.nb_stop = 0 def _on_time(self, evt): """ Update the progress bar while the timer is running :param evt: wx.EVT_TIMER """ # Check stop flag that can be set from non main thread if self.timer.IsRunning(): self.gauge.Pulse() def clear_gauge(self, msg=""): """ Hide the gauge """ self.progress = 0 self.gauge.SetValue(0) self.gauge.Hide() def set_icon(self, event): """ Display icons related to the type of message sent to the statusbar when available. No icon is displayed if the message is empty """ if hasattr(event, "status"): status = str(event.status) if status.strip() == "": return else: return if not hasattr(event, "info"): return # Get the size of the button images height = STATUS_BAR_ICON_SIZE msg = event.info.lower() if msg == "warning": self.status_color.SetBackgroundColour(YELLOW) self.status_color.SetForegroundColour(YELLOW) elif msg == "error": self.status_color.SetBackgroundColour(RED) self.status_color.SetForegroundColour(RED) else: self.status_color.SetBackgroundColour(GREEN) self.status_color.SetForegroundColour(GREEN) def set_dialog(self, event): """ Display dialogbox """ if not hasattr(event, "info"): return msg = event.info.lower() if msg == "error": e_msg = "Error(s) Occurred:\n" e_msg += "\t" + event.status + "\n\n" e_msg += "Further information might be available in " e_msg += "the Console log (bottom right corner)." wx.MessageBox(e_msg, style=wx.ICON_ERROR) def set_message(self, event): """ display received message on the statusbar """ if hasattr(event, "status"): self.SetStatusText(text=str(event.status), event=event) def set_gauge(self, event): """ change the state of the gauge according the state of the current job """ if not hasattr(event, "type"): return type = event.type self.gauge.Show(True) if type.lower() == "start": self.nb_start += 1 #self.timer.Stop() self.progress += 5 self.gauge.SetValue(int(self.progress)) self.progress += 5 if self.progress < self.gauge.GetRange() - 20: self.gauge.SetValue(int(self.progress)) if type.lower() == "progress": self.nb_progress += 1 self.timer.Start(1) self.gauge.Pulse() if type.lower() == "update": self.progress += 5 if self.progress < self.gauge.GetRange()- 20: self.gauge.SetValue(int(self.progress)) if type.lower() == "stop": self.nb_stop += 1 self.gauge.Show(True) if self.enable_clear_gauge(): self.timer.Stop() self.progress = 0 self.gauge.SetValue(100) self.timer_stop.Start(5) def set_status(self, event): """ Update the status bar . :param type: type of message send. type must be in ["start","progress","update","stop"] :param msg: the message itself as string :param thread: if updatting using a thread status """ self.set_message(event=event) self.set_icon(event=event) self.set_gauge(event=event) # dialog on error self.set_dialog(event=event) def _onMonitor(self, event): """ Pop up a frame with messages sent to the status bar """ self.frame.Show(False) self.frame.Show(True) class SPageStatusbar(wxStatusB): def __init__(self, parent, timeout=None, *args, **kwds): wxStatusB.__init__(self, parent, *args, **kwds) self.SetFieldsCount(1) self.timeout = timeout width, height = parent.GetSizeTuple() self.gauge = wx.Gauge(self, style=wx.GA_HORIZONTAL, size=(width, height/10)) rect = self.GetFieldRect(0) self.gauge.SetPosition((rect.x , rect.y )) if self.timeout is not None: self.gauge.SetRange(int(self.timeout)) self.timer = wx.Timer(self, -1) self.Bind(wx.EVT_TIMER, self._on_time, self.timer) self.timer.Start(1) self.pos = 0 def _on_time(self, evt): """ Update the progress bar while the timer is running :param evt: wx.EVT_TIMER """ # Check stop flag that can be set from non main thread if self.timeout is None and self.timer.IsRunning(): self.gauge.Pulse() if __name__ == "__main__": app = wx.PySimpleApp() frame = wx.Frame(None, wx.ID_ANY, 'test frame') #statusBar = StatusBar(frame, wx.ID_ANY) statusBar = SPageStatusbar(frame) frame.SetStatusBar(statusBar) frame.Show(True) #event = MessageEvent() #event.type = "progress" #event.status = "statusbar...." #event.info = "error" #statusBar.set_status(event=event) app.MainLoop()