MachineLearningProject/log.py at main · Nonender/MachineLearningProject · GitHub

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
import os
from datetime import datetime
from inspect import currentframe, getframeinfo


class MeowLogger(object):
    def __init__(self):
        self.log_file = None

    def __del__(self):
        if self.log_file is not None:
            self.log_file.close()

    def __header(self, pid):
        now = datetime.now()
        frame_info = getframeinfo(currentframe().f_back.f_back)
        if pid:
            return "[\033[90m{}|\033[0m{}:{}|{}] ".format(
                now.strftime("%Y-%m-%dT%H:%M:%S.%f"),
                os.path.basename(frame_info.filename),
                frame_info.lineno, os.getpid())
        return "[\033[90m{}|\033[0m{}:{}] ".format(
            now.strftime("%Y-%m-%dT%H:%M:%S.%f"),
            os.path.basename(frame_info.filename), frame_info.lineno)

    def set_log_file(self, filename):
        if self.log_file is not None:
            self.log_file.close()
        self.log_file = open(filename, "w")

    def log(self, content, muted=False):
        if muted:
            return
        if self.log_file is not None:
            self.log_file.write(content + "\n")
            self.log_file.flush()
            return
        print(content)

    def inf(self, line, pid=False, muted=False):
        self.log(self.__header(pid) + line, muted)

    def grey(self, line, pid=False, muted=False):
        self.log("\033[90m{}\033[0m".format(line), muted)


    def red(self, line, pid=False, muted=False):
        self.log("{}\033[91m{}\033[0m".format(self.__header(pid), line), muted)

    def green(self, line, pid=False, muted=False):
        self.log("{}\033[92m{}\033[0m".format(self.__header(pid), line), muted)

    def yellow(self, line, pid=False, muted=False):
        self.log("{}\033[93m{}\033[0m".format(self.__header(pid), line), muted)

    def blue(self, line, pid=False, muted=False):
        self.log("{}\033[94m{}\033[0m".format(self.__header(pid), line), muted)

    def pink(self, line, pid=False, muted=False):
        self.log("{}\033[95m{}\033[0m".format(self.__header(pid), line), muted)

    def cyan(self, line, pid=False, muted=False):
        self.log("{}\033[96m{}\033[0m".format(self.__header(pid), line), muted)


log = MeowLogger()