stable/freqtrade/loggers.py

102 lines
3.9 KiB
Python
Raw Normal View History

2019-07-06 22:40:52 +00:00
import logging
import sys
2019-10-25 06:50:37 +00:00
from logging import Formatter
from logging.handlers import (BufferingHandler, RotatingFileHandler,
SysLogHandler)
2020-08-14 12:53:21 +00:00
from typing import Any, Dict
2019-07-06 22:40:52 +00:00
from freqtrade.exceptions import OperationalException
2019-10-25 06:50:37 +00:00
2019-07-06 22:40:52 +00:00
logger = logging.getLogger(__name__)
2020-08-14 12:41:46 +00:00
LOGFORMAT = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
2019-07-06 22:40:52 +00:00
# Initialize bufferhandler - will be used for /log endpoints
bufferHandler = BufferingHandler(1000)
bufferHandler.setFormatter(Formatter(LOGFORMAT))
2019-07-06 22:40:52 +00:00
def _set_loggers(verbosity: int = 0, api_verbosity: str = 'info') -> None:
2019-07-06 22:40:52 +00:00
"""
2019-07-07 07:17:01 +00:00
Set the logging level for third party libraries
2019-07-06 22:40:52 +00:00
:return: None
"""
logging.getLogger('requests').setLevel(
2020-03-10 19:30:36 +00:00
logging.INFO if verbosity <= 1 else logging.DEBUG
2019-07-06 22:40:52 +00:00
)
logging.getLogger("urllib3").setLevel(
2020-03-10 19:30:36 +00:00
logging.INFO if verbosity <= 1 else logging.DEBUG
2019-07-06 22:40:52 +00:00
)
logging.getLogger('ccxt.base.exchange').setLevel(
2020-03-10 19:30:36 +00:00
logging.INFO if verbosity <= 2 else logging.DEBUG
2019-07-06 22:40:52 +00:00
)
logging.getLogger('telegram').setLevel(logging.INFO)
logging.getLogger('werkzeug').setLevel(
logging.ERROR if api_verbosity == 'error' else logging.INFO
)
2019-07-06 22:40:52 +00:00
2020-08-14 12:41:46 +00:00
def setup_logging_pre() -> None:
"""
2020-08-14 18:12:59 +00:00
Early setup for logging.
Uses INFO loglevel and only the Streamhandler.
Early messages (before proper logging setup) will therefore only be available
after the proper logging setup.
2020-08-14 12:41:46 +00:00
"""
2020-08-14 12:53:21 +00:00
logging.basicConfig(
level=logging.INFO,
format=LOGFORMAT,
handlers=[logging.StreamHandler(sys.stderr)]
2020-08-14 12:53:21 +00:00
)
2020-08-14 12:41:46 +00:00
2019-07-06 22:40:52 +00:00
def setup_logging(config: Dict[str, Any]) -> None:
"""
Process -v/--verbose, --logfile options
2019-07-06 22:40:52 +00:00
"""
# Log level
verbosity = config['verbosity']
logging.root.addHandler(bufferHandler)
2019-07-06 22:40:52 +00:00
2019-10-25 06:50:37 +00:00
logfile = config.get('logfile')
2019-10-25 06:50:37 +00:00
if logfile:
s = logfile.split(':')
if s[0] == 'syslog':
# Address can be either a string (socket filename) for Unix domain socket or
# a tuple (hostname, port) for UDP socket.
# Address can be omitted (i.e. simple 'syslog' used as the value of
# config['logfilename']), which defaults to '/dev/log', applicable for most
# of the systems.
address = (s[1], int(s[2])) if len(s) > 2 else s[1] if len(s) > 1 else '/dev/log'
handler = SysLogHandler(address=address)
# No datetime field for logging into syslog, to allow syslog
# to perform reduction of repeating messages if this is set in the
# syslog config. The messages should be equal for this.
handler.setFormatter(Formatter('%(name)s - %(levelname)s - %(message)s'))
2020-08-14 12:53:21 +00:00
logging.root.addHandler(handler)
2019-10-25 06:50:37 +00:00
elif s[0] == 'journald':
try:
from systemd.journal import JournaldLogHandler
except ImportError:
raise OperationalException("You need the systemd python package be installed in "
"order to use logging to journald.")
2020-08-15 06:15:18 +00:00
handler_jd = JournaldLogHandler()
2019-10-25 06:50:37 +00:00
# No datetime field for logging into journald, to allow syslog
# to perform reduction of repeating messages if this is set in the
# syslog config. The messages should be equal for this.
2020-08-15 06:15:18 +00:00
handler_jd.setFormatter(Formatter('%(name)s - %(levelname)s - %(message)s'))
logging.root.addHandler(handler_jd)
2019-10-25 06:50:37 +00:00
else:
2020-08-15 06:15:18 +00:00
handler_rf = RotatingFileHandler(logfile,
maxBytes=1024 * 1024 * 10, # 10Mb
backupCount=10)
handler_rf.setFormatter(Formatter(LOGFORMAT))
logging.root.addHandler(handler_rf)
2020-08-14 12:41:46 +00:00
logging.root.setLevel(logging.INFO if verbosity < 1 else logging.DEBUG)
_set_loggers(verbosity, config.get('api_server', {}).get('verbosity', 'info'))
2020-08-14 12:53:21 +00:00
2019-07-06 22:40:52 +00:00
logger.info('Verbosity set to %s', verbosity)