Désactivation du logging dans des fichiers par positionnement de la variable 'WINWII_...
[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 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()