diff options
Diffstat (limited to 'Omni/Log.py')
-rw-r--r-- | Omni/Log.py | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/Omni/Log.py b/Omni/Log.py new file mode 100644 index 0000000..2fbd007 --- /dev/null +++ b/Omni/Log.py @@ -0,0 +1,35 @@ +"""Setup logging like Omni/Log.hs.""" + +# noqa: builtin-attribute-shadowing +import logging +import typing + + +class LowerFormatter(logging.Formatter): + """A logging formatter that formats logs how I like.""" + + def format(self: "LowerFormatter", record: typing.Any) -> typing.Any: + """Use the format I like for logging.""" + record.levelname = record.levelname.lower() + return super(logging.Formatter, self).format(record) # type: ignore[misc] + + +def setup() -> None: + """Run this in your __main__ function.""" + logging.basicConfig( + level=logging.DEBUG, + format="%(levelname)s: %(name)s: %(message)s", + ) + logging.addLevelName(logging.DEBUG, "dbug") + logging.addLevelName(logging.ERROR, "fail") + logging.addLevelName(logging.INFO, "info") + logger = logging.getLogger(__name__) + formatter = LowerFormatter() + handler = logging.StreamHandler() + handler.setFormatter(formatter) + logger.addHandler(handler) + + +if __name__ == "__main__": + setup() + logging.debug("i am doing testing") |