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 setFileName(self, name) :
 181     #    self.fileName.set(name)
 182
 183     def refresh(self, lfa) :
 184         filename = basename(lfa.logfile.name)
 185         self.fileName.set(filename)
 186         metadata = lfa.getMetadata()
 187         self.patientName.set(metadata.get('PatientName', ''))
 188         self.commentsText.delete(1.0, END)
 189         self.commentsText.insert(1.0, metadata.get('Comments', ''))
 190
 191     def createWidgets(self) :
 192         fileLbl = Label(self, text='Fichier :')
 193         fileLbl.grid(row=0, column=0, sticky=E)
 194
 195         fileNameLbl = Label(self, textvariable=self.fileName)
 196         fileNameLbl.grid(row=0, column=1, sticky=W)
 197
 198         nameLbl = Label(self, text='Patient :')
 199         nameLbl.grid(row=1, column=0, sticky=E)
 200
 201         self.nameEntry = Entry(self, width=40, textvariable=self.patientName)
 202         self.nameEntry.grid(row=1, column=1, sticky=W)
 203
 204         commentsLbl = Label(self, text='Commentaires :')
 205         commentsLbl.grid(row=2, column=0, sticky=E)
 206
 207         self.commentsText = Text(self, width=40, height=4, undo=True, wrap=WORD)
 208         self.commentsText.grid(row=2, column=1, sticky=W)
 209
 210         self.saveBtn = Button(self, text='Enregistrer', command=self.saveMetadata)
 211         self.saveBtn.grid(row=3, column=1, sticky=E)
 212
 213     def saveMetadata(self):
 214         app = self.master.master
 215         filepath = app.currentFilePath
 216         lfa = LogFileAnalyser(filepath, mode='r+')
 217         patientName = '%s\n' % self.nameEntry.get().replace('\n', ' ').strip()
 218         comments = '%s\n' % self.commentsText.get(1.0, END).replace('\n', ' ').strip()
 219         metadata = (('PatientName', self.nameEntry.get()),
 220                     ('Comments', comments))
 221         lfa.setMetadata(metadata)
 222
 223
 224 class ResultsFrame(Frame) :
 225
 226     def layResults(self, lfa) :
 227         results = lfa.analyse()
 228         if results :
 229             for i, kv in enumerate(results) :
 230                 k, v = kv
 231                 kl = Label(self, text='%s :' % k)
 232                 kl.grid(row=i, column=0, sticky=E)
 233
 234                 vl = Label(self, text=v)
 235                 vl.grid(row=i, column=1, sticky=W)
 236         else :
 237             msg = Label(self, text="Pas de données exploitables.")
 238             msg.grid()
 239
 240
 241 def main() :
 242     app = Application()
 243     app.master.title("Analyseur des sessions MINWii")
 244     app.mainloop()
 245
 246 if __name__ == '__main__' :
 247     main()