1bbebd40b8404bdce93368d7cf216c7fc3f26b2d
[minwii.git] / 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()