logger.py

""" Logging utility module. """

import logging

# import logstash
from logstash_async.handler import AsynchronousLogstashHandler
from elasticecslogging.handlers import ElasticECSHandler

from laudio import config


logging.basicConfig(level=config["logging"]["level"], format=config["logging"]["format"])


def get_logger(name=None, **kwargs):
    """ Get logger instance. """
    logger = logging.getLogger(name)

    logger.addHandler(
        ElasticECSHandler(
            hosts=[{"host": "localhost", "port": 9200}],
            auth_type=ElasticECSHandler.AuthType.NO_AUTH,
            es_index_name=f"laudio-{kwargs.get('env')}",
            es_additional_fields={**kwargs}
        )
    )

    return logger