from datetime import datetime

import colorama


class Logger(object):
    def __init__(self, module: str, datefmt: str = '%m/%d/%Y %I:%M:%S %p'):
        colorama.init()
        self.module = module
        self.datefmt = datefmt

    def info(self, msg: str, levelname: str = 'INFO'):
        asctime = datetime.now().strftime(self.datefmt)

        print(f'\033[92m[{asctime}] - {levelname} - {self.module} - {msg}\033[39m')

    def debug(self, msg: str, levelname: str = 'DEBUG'):
        asctime = datetime.now().strftime(self.datefmt)

        print(f'\033[93m[{asctime}] - {levelname} - {self.module} - {msg}\033[39m')

    def verbose(self, msg: str, levelname: str = 'VERBOSE'):
        asctime = datetime.now().strftime(self.datefmt)

        print(f'\033[96m[{asctime}] - {levelname} - {self.module} - {msg}\033[39m')

    def error(self, msg: str, levelname: str = 'ERROR'):
        asctime = datetime.now().strftime(self.datefmt)

        print(f'\033[91m[{asctime}] - {levelname} - {self.module} - {msg}\033[39m')