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.hid = StringVar()
 178         self.patientName = StringVar()
 179         self.createWidgets()
 180
 181     def refresh(self, lfa) :
 182         filename = basename(lfa.logfile.name)
 183         self.fileName.set(filename)
 184         self.hid.set(lfa.getHID())
 185         metadata = lfa.getMetadata()
 186         self.patientName.set(metadata.get('PatientName', ''))
 187         self.commentsText.delete(1.0, END)
 188         self.commentsText.insert(1.0, metadata.get('Comments', ''))
 189
 190     def createWidgets(self) :
 191         fileLbl = Label(self, text='Fichier :')
 192         fileLbl.grid(row=0, column=0, sticky=E)
 193
 194         fileNameLbl = Label(self, textvariable=self.fileName)
 195         fileNameLbl.grid(row=0, column=1, sticky=W)
 196
 197         hidLbl = Label(self, text='HID :')
 198         hidLbl.grid(row=1, column=0, sticky=E)
 199
 200         hidNameLbl = Label(self, textvariable=self.hid)
 201         hidNameLbl.grid(row=1, column=1, sticky=W)
 202
 203         nameLbl = Label(self, text='Patient :')
 204         nameLbl.grid(row=2, column=0, sticky=E)
 205
 206         self.nameEntry = Entry(self, width=40, textvariable=self.patientName)
 207         self.nameEntry.grid(row=2, column=1, sticky=W)
 208
 209         commentsLbl = Label(self, text='Commentaires :')
 210         commentsLbl.grid(row=3, column=0, sticky=E)
 211
 212         self.commentsText = Text(self, width=40, height=4, undo=True, wrap=WORD)
 213         self.commentsText.grid(row=3, column=1, sticky=W)
 214
 215         self.saveBtn = Button(self, text='Enregistrer', command=self.saveMetadata)
 216         self.saveBtn.grid(row=4, column=1, sticky=E)
 217
 218     def saveMetadata(self):
 219         app = self.master.master
 220         filepath = app.currentFilePath
 221         lfa = LogFileAnalyser(filepath, mode='r+')
 222         patientName = '%s\n' % self.nameEntry.get().replace('\n', ' ').strip()
 223         comments = '%s\n' % self.commentsText.get(1.0, END).replace('\n', ' ').strip()
 224         metadata = (('PatientName', self.nameEntry.get()),
 225                     ('Comments', comments))
 226         lfa.setMetadata(metadata)
 227
 228
 229 class ResultsFrame(Frame) :
 230
 231     def layResults(self, lfa) :
 232         results = lfa.analyse()
 233         if results :
 234             for i, kvt in enumerate(results) :
 235                 k, v, timeBased = kvt
 236                 kl = Label(self, text='%s :' % k)
 237                 kl.grid(row=i, column=0, sticky=E)
 238
 239                 if not timeBased :
 240                     vl = Label(self, text=v)
 241                     vl.grid(row=i, column=1, sticky=W)
 242                 else :
 243                     maxv = max(v)
 244                     if maxv :
 245                         cw, ch = 200, 100
 246                         c = Canvas(self, background='#fff', width=cw, height=ch)
 247                         rectW = int(float(cw) / len(v))
 248                         unitRectH = float(ch) / maxv
 249                         for j, fv in enumerate(v) :
 250                             if not fv : continue
 251                             x0 = j * rectW
 252                             y0 = ch - int(unitRectH * fv)
 253                             x1 = (j + 1) * rectW
 254                             y1 = ch
 255                             c.create_rectangle(x0, y0, x1, y1, fill="#9085ba")
 256                         c.grid(row=i, column=1, sticky=W)
 257
 258                     else :
 259                         vl = Label(self, text='—')
 260                         vl.grid(row=i, column=1, sticky=W)
 261         else :
 262             msg = Label(self, text="Pas de données exploitables.")
 263             msg.grid()
 264
 265
 266 def main() :
 267     app = Application()
 268     app.master.title("Analyseur des sessions MINWii")
 269     app.mainloop()
 270
 271 if __name__ == '__main__' :
 272     main()