src/minwii/logapp.py

   1 # -*- coding: utf-8 -*-
   2 """
   3 Interface graphique pour l'analyse des fichiers de log minwii.
   4
   5 $Id$
   6 $URL$
   7 """
   8
   9 import os
  10 os.environ['WINWII_NO_LOG'] = '1'
  11 from Tkinter import *
  12 import tkFileDialog
  13 from glob import glob
  14 from os.path import join as pjoin
  15 from os.path import basename
  16 from os.path import getsize
  17 import os.path
  18 from minwii.loganalyse import LogFileAnalyser
  19 from minwii.config import LOGS_DIR
  20 from pprint import pprint
  21
  22 class Application(Frame) :
  23     def __init__(self, master=None) :
  24         Frame.__init__(self, master)
  25         self.configureStretching()
  26         self.createWidgets()
  27         self.logDir = ''
  28         self.logFiles = []
  29         self.currentFilePath = ''
  30         self.resultsFrame = None
  31
  32         if os.path.exists(LOGS_DIR) :
  33             self.chooseDirDialog(dir=LOGS_DIR)
  34
  35     def configureStretching(self) :
  36         top=self.winfo_toplevel()
  37         top.rowconfigure(0, weight=1)
  38         top.columnconfigure(0, weight=1)
  39
  40         self.grid(sticky=N+S+E+W, padx=10, pady=10)
  41         self.rowconfigure(0, weight=1)
  42         self.columnconfigure(0, weight=1)
  43
  44     def createWidgets(self) :
  45         # zone d'affichage des données'
  46         self.dataFrame = df = Frame(self)
  47
  48         self.identFrame = Identification(df)
  49         self.identFrame.grid(sticky=NW)
  50
  51         # barre de boutons
  52         self.btnFrame = bf = Frame(self)
  53         bf.grid(row=1, column=0, sticky=W+S+E)
  54         bf.rowconfigure(0, weight=1)
  55         for i in range(3) :
  56             bf.columnconfigure(i, weight=1)
  57
  58         self.chooseLogDir = Button(bf, text="Parcourir…", command=self.chooseDirDialog)
  59         self.chooseLogDir.grid(row=0, column=0, sticky=W)
  60
  61         self.nav = Navbar(bf, incCallback=self.loadLogFile, decCallback=self.loadLogFile)
  62
  63         self.quitButton = Button(bf, text='Terminer', command=self.quit)
  64         self.quitButton.grid(row=0, column=2, sticky=E)
  65
  66     def chooseDirDialog(self, dir=None) :
  67         if dir is None :
  68             self.logDir = tkFileDialog.askdirectory(title='Sélectionnez un dossier de fichiers de logs')
  69         else :
  70             self.logDir = dir
  71         if self.logDir :
  72             self.logFiles = glob(pjoin(self.logDir, '*.log'))
  73             self._cleanupJunkFiles()
  74             self.logFiles.sort()
  75             self.logFiles.reverse()
  76             self.dataFrame.grid(row=0, column=0, sticky=NW)
  77             self.nav.setSize(len(self.logFiles))
  78             self.nav.grid(row=0, column=1)
  79             self.loadLogFile(self.nav)
  80
  81     def _cleanupJunkFiles(self) :
  82         files = []
  83         while self.logFiles :
  84             f = self.logFiles.pop()
  85             if not getsize(f) :
  86                 os.remove(f)
  87                 continue
  88             else :
  89                 of = open(f)
  90                 lfa = LogFileAnalyser(of)
  91                 if lfa.getLastEventTicks() is None :
  92                     of.close()
  93                     os.remove(f)
  94                     continue
  95                 else :
  96                     of.close()
  97
  98                 files.append(f)
  99
 100         self.logFiles = files
 101
 102
 103     def loadLogFile(self, nav) :
 104         index = nav.index - 1
 105         filepath = self.logFiles[index]
 106         self.currentFilePath = filepath
 107         lfa = LogFileAnalyser(self.currentFilePath)
 108         self.identFrame.refresh(lfa)
 109         if self.resultsFrame :
 110             self.resultsFrame.destroy()
 111         self.resultsFrame = ResultsFrame(self.dataFrame)
 112         self.resultsFrame.layResults(lfa)
 113         lfa.close()
 114         self.resultsFrame.grid()
 115
 116
 117 class Navbar(Frame) :
 118     def __init__(self, master=None, size=1, incCallback=None, decCallback=None) :
 119         Frame.__init__(self, master)
 120         self.caption = StringVar()
 121         self.createWidgets()
 122         self.setSize(size)
 123         self.incCallback = incCallback if incCallback else lambda x : None
 124         self.decCallback = decCallback if decCallback else lambda x : None
 125         self.caption.set('%d / %d' % (self.index, self.to))
 126
 127     def createWidgets(self) :
 128         self.backBtn = Button(self,
 129                               text='◀',
 130                               command = self.dec
 131                               )
 132         self.backBtn.grid(row=0, column=0)
 133
 134         self.lbl = Label(self, textvariable=self.caption)
 135         self.lbl.grid(row=0, column=1)
 136
 137         self.nextBtn = Button(self,
 138                               text='▶',
 139                               command = self.inc)
 140         self.nextBtn.grid(row=0, column=2)
 141
 142     def refreshStates(self) :
 143         if self.index == self.from_ :
 144             self.backBtn.configure(state=DISABLED)
 145         else :
 146             self.backBtn.configure(state=NORMAL)
 147
 148         if self.index < self.to :
 149             self.nextBtn.configure(state=NORMAL)
 150         else :
 151             self.nextBtn.configure(state=DISABLED)
 152
 153         self.caption.set('%d / %d' % (self.index, self.to))
 154
 155
 156     def dec(self) :
 157         self.index = self.index - 1
 158         self.refreshStates()
 159         self.decCallback(self)
 160
 161     def inc(self) :
 162         self.index = self.index + 1
 163         self.refreshStates()
 164         self.incCallback(self)
 165
 166     def setSize(self, size) :
 167         self.from_ = 1
 168         self.to = size
 169         self.index = 1
 170         self.refreshStates()
 171
 172
 173 class Identification(Frame) :
 174     def __init__(self, master=None) :
 175         Frame.__init__(self, master)
 176         self.fileName = StringVar()
 177         self.patientName = StringVar()
 178         self.createWidgets()
 179
 180     def refresh(self, lfa) :
 181         filename = basename(lfa.logfile.name)
 182         self.fileName.set(filename)
 183         metadata = lfa.getMetadata()
 184         self.patientName.set(metadata.get('PatientName', ''))
 185         self.commentsText.delete(1.0, END)
 186         self.commentsText.insert(1.0, metadata.get('Comments', ''))
 187
 188     def createWidgets(self) :
 189         fileLbl = Label(self, text='Fichier :')
 190         fileLbl.grid(row=0, column=0, sticky=E)
 191
 192         fileNameLbl = Label(self, textvariable=self.fileName)
 193         fileNameLbl.grid(row=0, column=1, sticky=W)
 194
 195         nameLbl = Label(self, text='Patient :')
 196         nameLbl.grid(row=1, column=0, sticky=E)
 197
 198         self.nameEntry = Entry(self, width=40, textvariable=self.patientName)
 199         self.nameEntry.grid(row=1, column=1, sticky=W)
 200
 201         commentsLbl = Label(self, text='Commentaires :')
 202         commentsLbl.grid(row=2, column=0, sticky=E)
 203
 204         self.commentsText = Text(self, width=40, height=4, undo=True, wrap=WORD)
 205         self.commentsText.grid(row=2, column=1, sticky=W)
 206
 207         self.saveBtn = Button(self, text='Enregistrer', command=self.saveMetadata)
 208         self.saveBtn.grid(row=3, column=1, sticky=E)
 209
 210     def saveMetadata(self):
 211         app = self.master.master
 212         filepath = app.currentFilePath
 213         lfa = LogFileAnalyser(filepath, mode='r+')
 214         patientName = '%s\n' % self.nameEntry.get().replace('\n', ' ').strip()
 215         comments = '%s\n' % self.commentsText.get(1.0, END).replace('\n', ' ').strip()
 216         metadata = (('PatientName', self.nameEntry.get()),
 217                     ('Comments', comments))
 218         lfa.setMetadata(metadata)
 219
 220
 221 class ResultsFrame(Frame) :
 222
 223     def layResults(self, lfa) :
 224         results = lfa.analyse()
 225         if results :
 226             for i, kvt in enumerate(results) :
 227                 k, v, timeBased = kvt
 228                 kl = Label(self, text='%s :' % k)
 229                 kl.grid(row=i, column=0, sticky=E)
 230
 231                 if not timeBased :
 232                     vl = Label(self, text=v)
 233                     vl.grid(row=i, column=1, sticky=W)
 234                 else :
 235                     maxv = max(v)
 236                     if maxv :
 237                         cw, ch = 200, 100
 238                         c = Canvas(self, background='#fff', width=cw, height=ch)
 239                         rectW = int(float(cw) / len(v))
 240                         unitRectH = float(ch) / maxv
 241                         for j, fv in enumerate(v) :
 242                             if not fv : continue
 243                             x0 = j * rectW
 244                             y0 = ch - int(unitRectH * fv)
 245                             x1 = (j + 1) * rectW
 246                             y1 = ch
 247                             c.create_rectangle(x0, y0, x1, y1, fill="#9085ba")
 248                         c.grid(row=i, column=1, sticky=W)
 249
 250                     else :
 251                         vl = Label(self, text='—')
 252                         vl.grid(row=i, column=1, sticky=W)
 253         else :
 254             msg = Label(self, text="Pas de données exploitables.")
 255             msg.grid()
 256
 257
 258 def main() :
 259     app = Application()
 260     app.master.title("Analyseur des sessions MINWii")
 261     app.mainloop()
 262
 263 if __name__ == '__main__' :
 264     main()