summaryrefslogtreecommitdiff
path: root/jfr_playoff/gui/logframe.py
blob: 28e3487aa511e0e1832fe5281763443b87242599 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#coding=utf-8

import datetime
import logging as log
from collections import OrderedDict

import tkinter as tk
from tkinter import ttk
import tkFileDialog as tkfd

class LogWindow(tk.Toplevel):
    def __init__(self, *args, **kwargs):
        tk.Toplevel.__init__(self, *args, **kwargs)
        self.withdraw()
        self.protocol('WM_DELETE_WINDOW', self.withdraw)
        self.renderContents()
        self._registerLogging()
        self._records = []
        self._counter = -1

    def renderContents(self):
        columns = OrderedDict([
            ('level', 'Poziom komunikatu'),
            ('category', 'Moduł'),
            ('message', 'Komunikat')])
        self.logList = ttk.Treeview(
            self, show='headings',
            columns=columns.keys(),
            selectmode='browse')
        for column, heading in columns.iteritems():
            self.logList.heading(column, text=heading)
        self.logList.pack(side=tk.TOP, fill=tk.BOTH, expand=True)
        btnFrame = tk.Frame(self)
        btnFrame.pack(side=tk.BOTTOM)
        (ttk.Button(
            btnFrame, text='Zapisz dziennik...',
            command=self.onRecordsSave)).pack(side=tk.LEFT)
        (ttk.Button(
            btnFrame, text='Wyczyść dziennik',
            command=self.resetRecords)).pack(side=tk.LEFT)

    def _registerLogging(self):
        logHandler = LogHandler(log.INFO, window=self)
        logger = log.getLogger()
        logger.setLevel(log.INFO)
        logger.addHandler(logHandler)

    def addRecord(self, record):
        self._counter += 1
        self._records.append((record, datetime.datetime.now()))
        self.logList.insert(
            '', tk.END, tag=self._counter, values=[
                record.levelname, record.name, record.message
            ])

    def resetRecords(self):
        self._records = []
        self.logList.delete(*self.logList.get_children())

    def onRecordsSave(self, *args):
        filename = tkfd.asksaveasfilename(
            title='Wybierz plik dziennika',
            filetypes=(('Log files', '*.log'),))
        if filename:
            if not filename.lower().endswith('.log'):
                filename = filename + '.log'
            self._saveRecords(filename)

    def _saveRecords(self, filename):
        with open(filename, 'w') as fileObj:
            for record, timestamp in self._records:
                fileObj.write('%s\t%s\t%s\t%s' % (
                    timestamp, record.levelname, record.name, record.message))


class LogHandler(log.Handler):
    def __init__(self, *args, **kwargs):
        self._window = kwargs['window']
        del kwargs['window']
        log.Handler.__init__(self, *args, **kwargs)

    def handle(self, record):
        self.format(record)
        self._window.addRecord(record)