import section for logzero and switch to logging but setting formatter like logzero

test_logger.py

#!/usr/bin/env python3

try:
    from logzero import logger
except ImportError:
    from logging import DEBUG
    from logging import Formatter
    from logging import StreamHandler
    from logging import getLogger

    formatter = Formatter(
        "[%(levelname)1.1s %(asctime)s %(module)s:%(lineno)d] %(message)s",
        datefmt="%y%m%d %H:%M:%S",
    )
    logger = getLogger(__name__)
    handler = StreamHandler()
    handler.setLevel(DEBUG)
    handler.setFormatter(formatter)
    logger.setLevel(DEBUG)
    logger.addHandler(handler)
    logger.propagate = False


if __name__ == "__main__":

    logger.info("info")
    logger.warning("warning")
    logger.debug("debug")
    logger.error("error")
    logger.critical("critical")