Source code for opencadd.utils

"""
Useful utilities used across the library
"""

import logging
import os
import shutil
import tempfile
import contextlib

_logger = logging.getLogger(__name__)


@contextlib.contextmanager
def enter_temp_directory(remove=True):
    """Create and enter a temporary directory; used as context manager."""
    temp_dir = tempfile.mkdtemp()
    cwd = os.getcwd()
    os.chdir(temp_dir)
    _logger.debug("Entered %s", temp_dir)
    yield cwd, temp_dir
    os.chdir(cwd)
    _logger.debug("Left %s", temp_dir)
    if remove:
        _logger.debug("Deleting %s", temp_dir)
        shutil.rmtree(temp_dir)


class PerLevelFormatter(logging.Formatter):
    """
    Adapted from https://stackoverflow.com/a/14859558
    """

    FORMATS = {
        logging.ERROR: "ERROR! %(message)s",
        logging.WARNING: "WARNING: %(message)s",
        logging.INFO: "%(message)s",
        logging.DEBUG: "Debug: %(message)s",
        101: "%(message)s",
        25: "%(message)s",
    }

    def __init__(self, fmt="%(levelname)d: %(message)s", datefmt=None, style="%", **kwargs):
        super().__init__(fmt=fmt, datefmt=datefmt, style=style, **kwargs)

    def format(self, record):
        # Save the original format configured by the user
        # when the logger formatter was instantiated
        format_orig = self._style._fmt
        self._style._fmt = self.FORMATS.get(record.levelno, self._style._fmt)
        # Call the original formatter class to do the grunt work
        result = super().format(record)
        # Restore the original format configured by the user
        self._style._fmt = format_orig

        return result


class EmojiPerLevelFormatter(PerLevelFormatter):
    FORMATS = {
        logging.ERROR: "💥 ERROR! %(message)s",
        logging.WARNING: "⚠️ WARNING: %(message)s",
        logging.INFO: "ℹ️ %(message)s",
        logging.DEBUG: "⚙️ Debug: %(message)s",
        101: "%(message)s",
        25: "☑️ %(message)s",
    }