import logging
import tempfile
from contextlib import contextmanager

from image_prediction.config import CONFIG


@contextmanager
def temporary_pdf_file(pdf: bytes):
    with tempfile.NamedTemporaryFile() as f:
        f.write(pdf)
        yield f.name


def make_logger_getter():

    logger = logging.getLogger("imclf")
    logger.propagate = False

    handler = logging.StreamHandler()
    handler.setLevel(CONFIG.service.logging_level)

    log_format = "[%(levelname)s]: %(message)s"
    formatter = logging.Formatter(log_format)

    handler.setFormatter(formatter)
    logger.addHandler(handler)

    def get_logger():
        return logger

    return get_logger


get_logger = make_logger_getter()