aboutsummaryrefslogtreecommitdiffstats
path: root/pysite/__init__.py
blob: 3e11a7765a155518c660f9f957f621cb399311cf (plain) (blame)
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
# coding=utf-8
import logging
import os
import sys
from logging import StreamHandler
from logging.handlers import SysLogHandler

from logmatic import JsonFormatter

from pysite.constants import DATADOG_ADDRESS, DATADOG_PORT, PAPERTRAIL_ADDRESS, PAPERTRAIL_PORT
from pysite.logs import NonPicklingSocketHandler

# region Logging
# Get the log level from environment

log_level = os.environ.get("LOG_LEVEL", "info").upper()

if hasattr(logging, log_level):
    log_level = getattr(logging, log_level)
else:
    raise RuntimeError(f"LOG_LEVEL environment variable has invalid value: {log_level}")

logging_handlers = []

if PAPERTRAIL_ADDRESS:
    logging_handlers.append(SysLogHandler(address=(PAPERTRAIL_ADDRESS, PAPERTRAIL_PORT)))

if DATADOG_ADDRESS:
    datadog_handler = NonPicklingSocketHandler(host=DATADOG_ADDRESS, port=DATADOG_PORT)
    datadog_handler.formatter = JsonFormatter(datefmt="%b %d %H:%M:%S")

    logging_handlers.append(datadog_handler)

logging_handlers.append(StreamHandler(stream=sys.stderr))

logging.basicConfig(
    format="%(asctime)s pd.beardfist.com Site: | %(name)35s | %(levelname)8s | %(message)s",
    datefmt="%b %d %H:%M:%S",
    level=log_level,
    handlers=logging_handlers
)
# endregion