-
Notifications
You must be signed in to change notification settings - Fork 13
/
logger.py
73 lines (56 loc) · 1.67 KB
/
logger.py
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
# coding: utf-8
# Copyright (c) Max-Planck-Institut für Eisenforschung GmbH - Computational Materials Design (CM) Department
# Distributed under the terms of "New BSD License", see the LICENSE file.
import logging
from types import MethodType
__author__ = "Joerg Neugebauer"
__copyright__ = (
"Copyright 2021, Max-Planck-Institut für Eisenforschung GmbH - "
"Computational Materials Design (CM) Department"
)
__version__ = "1.0"
__maintainer__ = "Jan Janssen"
__email__ = "janssen@mpie.de"
__status__ = "production"
__date__ = "Sep 1, 2017"
"""
Set the logging level for pyiron
"""
def set_logging_level(self, level, channel=None):
"""
Set level for logger
Args:
level (str): 'DEBUG, INFO, WARN'
channel (int): 0: file_log, 1: stream, None: both
"""
if channel:
self.handlers[channel].setLevel(level)
else:
self.handlers[0].setLevel(level)
self.handlers[1].setLevel(level)
def setup_logger():
"""
Setup logger - logs are written to pyiron.log
Returns:
logging.getLogger: Logger
"""
logger = logging.getLogger("pyiron_log")
logger.setLevel(logging.DEBUG)
formatter = logging.Formatter(
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)
ch = logging.StreamHandler()
ch.setLevel(logging.WARN)
ch.setFormatter(formatter)
logger.addHandler(ch)
try:
fh = logging.FileHandler("pyiron.log")
except PermissionError:
pass
else:
fh.setLevel(logging.INFO)
fh.setFormatter(formatter)
logger.addHandler(fh)
return logger
logger = setup_logger()
logger.set_logging_level = MethodType(set_logging_level, logger)