36 lines
710 B
Python
36 lines
710 B
Python
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("imgclf")
|
|
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()
|