custodial/log_setup.py at main · moneysocket/custodial · 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
# Copyright (c) 2019 PrimeVR
# All rights Reserved

import os
import gzip
import logging
import logging.handlers

class GZipRotator(object):
    def __call__(self, source, dest):
        os.rename(source, dest)
        f_in = open(dest, 'rb')
        f_out = gzip.open("%s.gz" % dest, 'wb')
        f_out.writelines(f_in)
        f_out.close()
        f_in.close()
        os.remove(dest)

def setup_logging(filename, component, console_silent=False,
                  min_level=logging.DEBUG):
    d = os.path.dirname(filename)
    if not os.path.exists(d):
        os.makedirs(d)

    if component:
        s = os.path.join(d, "%s-latest.log" % component)
        try:
            os.remove(s)
        except:
            pass
        os.symlink(filename, s)

    console_level = min_level if min_level > logging.INFO else logging.INFO
    console_handler = logging.StreamHandler()
    console_handler.setLevel(console_level)
    console_format = logging.Formatter('%(asctime)s %(levelname)s: %(message)s',
                                       datefmt='%m-%d %H:%M:%S')
    console_handler.setFormatter(console_format)

    logfile_handler = logging.handlers.TimedRotatingFileHandler(filename,
                                                                'midnight')
    console_level = logging.DEBUG if (logging.DEBUG < min_level) else min_level
    logfile_handler.setLevel(console_level)
    logfile_format = logging.Formatter(
        '%(asctime)s %(filename)s:%(lineno)d %(levelname)s: %(message)s')
    logfile_handler.setFormatter(logfile_format)
    logfile_handler.rotator = GZipRotator()

    handlers = ([logfile_handler] if console_silent else
                [console_handler, logfile_handler])
    logging.basicConfig(handlers=handlers, level=min_level)