mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-15 20:53:58 +00:00
101 lines
4.1 KiB
Python
101 lines
4.1 KiB
Python
import logging
|
|
from logging import Formatter
|
|
from logging.handlers import RotatingFileHandler, SysLogHandler
|
|
|
|
from freqtrade.constants import Config
|
|
from freqtrade.exceptions import OperationalException
|
|
from freqtrade.loggers.buffering_handler import FTBufferingHandler
|
|
from freqtrade.loggers.set_log_levels import set_loggers
|
|
from freqtrade.loggers.std_err_stream_handler import FTStdErrStreamHandler
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
LOGFORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
|
|
|
# Initialize bufferhandler - will be used for /log endpoints
|
|
bufferHandler = FTBufferingHandler(1000)
|
|
bufferHandler.setFormatter(Formatter(LOGFORMAT))
|
|
|
|
|
|
def get_existing_handlers(handlertype):
|
|
"""
|
|
Returns Existing handler or None (if the handler has not yet been added to the root handlers).
|
|
"""
|
|
return next((h for h in logging.root.handlers if isinstance(h, handlertype)), None)
|
|
|
|
|
|
def setup_logging_pre() -> None:
|
|
"""
|
|
Early setup for logging.
|
|
Uses INFO loglevel and only the Streamhandler.
|
|
Early messages (before proper logging setup) will therefore only be sent to additional
|
|
logging handlers after the real initialization, because we don't know which
|
|
ones the user desires beforehand.
|
|
"""
|
|
logging.basicConfig(
|
|
level=logging.INFO, format=LOGFORMAT, handlers=[FTStdErrStreamHandler(), bufferHandler]
|
|
)
|
|
|
|
|
|
def setup_logging(config: Config) -> None:
|
|
"""
|
|
Process -v/--verbose, --logfile options
|
|
"""
|
|
# Log level
|
|
verbosity = config["verbosity"]
|
|
logging.root.addHandler(bufferHandler)
|
|
|
|
logfile = config.get("logfile")
|
|
|
|
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_sl = get_existing_handlers(SysLogHandler)
|
|
if handler_sl:
|
|
logging.root.removeHandler(handler_sl)
|
|
handler_sl = 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_sl.setFormatter(Formatter("%(name)s - %(levelname)s - %(message)s"))
|
|
logging.root.addHandler(handler_sl)
|
|
elif s[0] == "journald": # pragma: no cover
|
|
try:
|
|
from cysystemd.journal import JournaldLogHandler
|
|
except ImportError:
|
|
raise OperationalException(
|
|
"You need the cysystemd python package be installed in "
|
|
"order to use logging to journald."
|
|
)
|
|
handler_jd = get_existing_handlers(JournaldLogHandler)
|
|
if handler_jd:
|
|
logging.root.removeHandler(handler_jd)
|
|
handler_jd = JournaldLogHandler()
|
|
# 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.
|
|
handler_jd.setFormatter(Formatter("%(name)s - %(levelname)s - %(message)s"))
|
|
logging.root.addHandler(handler_jd)
|
|
else:
|
|
handler_rf = get_existing_handlers(RotatingFileHandler)
|
|
if handler_rf:
|
|
logging.root.removeHandler(handler_rf)
|
|
handler_rf = RotatingFileHandler(
|
|
logfile,
|
|
maxBytes=1024 * 1024 * 10, # 10Mb
|
|
backupCount=10,
|
|
)
|
|
handler_rf.setFormatter(Formatter(LOGFORMAT))
|
|
logging.root.addHandler(handler_rf)
|
|
|
|
logging.root.setLevel(logging.INFO if verbosity < 1 else logging.DEBUG)
|
|
set_loggers(verbosity, config.get("api_server", {}).get("verbosity", "info"))
|
|
|
|
logger.info("Verbosity set to %s", verbosity)
|