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