import logging
from datetime import datetime

from utils.pretty import Pretty

from conf import config

class Logger:
    def __init__(self, message, pretty=False):
        log_level=logging.INFO
        es_logger = logging.getLogger('elasticsearch')
        es_logger.setLevel(logging.WARNING)

        format = '%(asctime)s: %(levelname)s - %(message)s'
        logging.basicConfig(
            handlers=[
                logging.FileHandler(f'./logs/ingest-{config.START_DATE}-{config.END_DATE}.log'),
                logging.StreamHandler()
            ],
            format=format,
            level=log_level,
            datefmt='%m/%d/%Y %I:%M:%S %p',
        )

        self.message = message

        if pretty:
            self.message = Pretty(message).print()

    def log_exception(self, error):
        logging.error(self.message)
        logging.error(error)

    def log_info(self):
        logging.info(self.message)