from __future__ import absolute_import, division, print_function import wxtbx.bitmaps from libtbx.queuing_system_utils import sge_utils from libtbx.utils import Sorry import wx try : from wx.lib.agw.genericmessagedialog import GenericMessageDialog except ImportError : GenericMessageDialog = wx.MessageBox import sys, time import getpass user = getpass.getuser() job_attrs = ["job_id", "state", "name", "user", "submit", "queue"] job_labels = ["Job ID", "Status", "Name", "User", "Start time", "Queue"] status_codes = ["d", "E", "h", "r", "R", "s", "S", "t", "T", "w"] status_imgs = [3, 4, 0, 1, 0, 0, 0, 0, 0, 2] col_sizes = [wx.LIST_AUTOSIZE] * 4 + [200,200] class qsub_list_data(object): def __init__(self): self._data = [] self._sortby = None self._sort_descending = False def Refresh(self): self._data = sge_utils.qstat_parse() if self._sortby is not None : self.SortItems(self._sortby, swap_order=False) def GetItemCount(self): return len(self._data) def GetItemText(self, item, col): return getattr(self._data[item], job_attrs[col]) def GetItemImage(self, item): status = self._data[item].state[-1] img_id = status_imgs[status_codes.index(status)] return img_id def SortItems(self, col, swap_order=True): if swap_order : if self._sortby == col : self._sort_descending = (not self._sort_descending) else : self._sort_descending = False if col == 0 : self._data.sort(key=lambda element: int(element.job_id)) elif col == 4 : fmt = "%m/%d/%Y %H:%M:%S" self._data.sort(key=lambda element: time.strptime(element.submit, fmt)) else : attr = job_attrs[col] self._data.sort(key=lambda element: getattr(element, attr)) if self._sort_descending : self._data.reverse() self._sortby = col def GetOwners(self, job_ids, as_list=False): names = [] for job in self._data : if job.job_id in job_ids : names.append(job.user) if as_list : return names return list(set(names)) def GetNames(self, job_ids): names = [] for job in self._data : if job.job_id in job_ids : names.append(job.name) return names class qsub_list_view(wx.ListCtrl): def __init__(self, *args, **kwds): wx.ListCtrl.__init__(self, *args, **kwds) self.SetupImages() self.Bind(wx.EVT_LIST_ITEM_SELECTED, self.OnSelect, self) self.Bind(wx.EVT_LIST_ITEM_DESELECTED, self.OnDeSelect, self) self.Bind(wx.EVT_LEFT_DCLICK, self.OnDoubleClick, self) self.Bind(wx.EVT_RIGHT_DOWN, self.OnRightClick, self) self.Bind(wx.EVT_LIST_COL_CLICK, self.OnSort, self) for i, label in enumerate(job_labels): self.InsertColumn(i, label) self.SetColumnWidth(i, col_sizes[i]) #wx.LIST_AUTOSIZE) self.dataSource = qsub_list_data() self.RefreshAllItems() def SetupImages(self): if wxtbx.bitmaps.icon_lib is None : return il = wx.ImageList(16, 16, True) #il.Add(wx.EmptyBitmap(16,16)) #EmptyImage(16,16).ConvertToBitmap()) for icon in ["blank", "run", "recur", "stop", "status_unknown"] : bmp = wxtbx.bitmaps.fetch_icon_bitmap("actions", icon, 16) il.Add(bmp) self.AssignImageList(il, wx.IMAGE_LIST_SMALL) def OnGetItemImage(self, item): return self.dataSource.GetItemImage(item) def OnGetItemAttr(self, item): pass def OnGetItemText(self, item, col): return self.dataSource.GetItemText(item, col) def RefreshAllItems(self): n_items = self.dataSource.GetItemCount() self.SetItemCount(n_items) if (n_items > 0): self.RefreshItems(0, n_items - 1) def GetSelectedJobIDs(self): jobs = [] item = self.GetFirstSelected() while item != -1 : jobs.append(self.dataSource.GetItemText(item, 0)) item = self.GetNextSelected(item) return jobs def GetOwners(self, job_ids, as_list=False): return self.dataSource.GetOwners(job_ids, as_list=as_list) def GetNames(self, job_ids): return self.dataSource.GetNames(job_ids) def OnSelect(self, event): pass def OnDeSelect(self, event): pass def OnDoubleClick(self, event): pass def OnRightClick(self, event): pass def OnSort(self, event): col = event.GetColumn() self.dataSource.SortItems(col) self.RefreshAllItems() def Update(self): self.dataSource.Refresh() self.RefreshAllItems() class queue_list_frame(wx.Frame): def __init__(self, *args, **kwds): wx.Frame.__init__(self, *args, **kwds) self.statusbar = self.CreateStatusBar() self.SetupToolbar() self.list_ctrl = qsub_list_view(parent=self, id=-1, size=(800,600), style=wx.LC_REPORT|wx.LC_VIRTUAL) self.Bind(wx.EVT_CLOSE, self.OnClose) self.Bind(wx.EVT_WINDOW_DESTROY, self.OnDestroy) self.Update() self._timer = wx.Timer(owner=self) self.Bind(wx.EVT_TIMER, self.OnUpdate) self.statusbar.Bind(wx.EVT_LEFT_DCLICK, self.OnUpdate) self._timer.Start(10000) def SetupToolbar(self): if wxtbx.bitmaps.icon_lib is None : return self.toolbar = self.CreateToolBar(style=wx.TB_TEXT) commands = [ ("actions","reload", "OnUpdate", "Refresh list"), ("actions","editdelete", "OnDelete", "Delete selected"), ] for (icon_class, icon_name, fname, label) in commands : bmp = wxtbx.bitmaps.fetch_icon_bitmap(icon_class, icon_name, 32) tool_button = self.toolbar.AddLabelTool(-1, label, bmp, shortHelp=label, kind=wx.ITEM_NORMAL) self.Bind(wx.EVT_MENU, getattr(self, fname), tool_button) self.toolbar.Realize() def OnClose(self, event): self.Destroy() def OnDestroy(self, event): pass def OnUpdate(self, event): self.Update() def OnDelete(self, event): job_ids = self.list_ctrl.GetSelectedJobIDs() if len(job_ids) == 0 : return users = self.list_ctrl.GetOwners(job_ids) if (len(users) > 1) or (not user in users): raise Sorry("At least one job selected for deletion is owned by a " "different user; this interface only allows you to delete your own "+ "jobs.") if self.ConfirmDelete(job_ids): try : success = sge_utils.qdel(job_ids=job_ids) except RuntimeError as e : raise Sorry("Error executing 'qdel' command: %s" % str(e)) else : GenericMessageDialog("Job(s) deleted successfuly.", style=wx.OK) def SetUpdateInterval(self, interval) : # in seconds, not ms self._timer.Stop() self._timer.Start(interval * 1000) def Update(self): self.list_ctrl.Update() self.statusbar.SetStatusText("Last updated at %s" % get_time()) def ConfirmDelete(self, job_ids): pass def get_time(): return time.strftime("%m-%d-%y %H:%M:%S", time.localtime()) #----------------------------------------------------------------------- def run(args): app = wx.App(0) frame = queue_list_frame(None, -1, "SGE Queue Status") frame.Show() frame.Fit() app.MainLoop() if __name__ == "__main__" : run(sys.argv[1:])