add util functions for making main()s, update zenthread to use

ClassLogger over loggify

Signed-off-by: Zen <[email protected]>

Author: desultory

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "zenlib"
-version = "1.5.1"
+version = "1.6.0"
 authors = [
   { name="Desultory", email="[email protected]" },
 ]

src/zenlib/threading/zenthread.py

@@ -2,11 +2,10 @@
 __version__ = '0.3.0'
 
 from threading import Thread, Event
-from zenlib.logging import loggify
+from zenlib.logging import ClassLogger
 
 
-@loggify
-class ZenThread(Thread):
+class ZenThread(ClassLogger, Thread):
     """ A thread that stores the exception and return value of the function it runs. """
     def __init__(self, looping=False, *args, **kwargs):
         super().__init__(*args, **kwargs)

src/zenlib/util/__init__.py

@@ -6,5 +6,8 @@
 from .walk_dict import walk_dict
 from .check_dict import check_dict
 from .clear_widgets import clear_widgets
+from .init_logger import init_logger
+from .process_args import process_args
+from .init_argparser import init_argparser
 
-__all__ = ['handle_plural', 'NoDupFlatList', 'pretty_print', 'replace_file_line', 'update_init', 'walk_dict', 'check_dict', 'clear_widgets']
+__all__ = ['handle_plural', 'NoDupFlatList', 'pretty_print', 'replace_file_line', 'update_init', 'walk_dict', 'check_dict', 'clear_widgets', 'init_logger', 'process_args', 'init_argparser']

src/zenlib/util/init_argparser.py

@@ -0,0 +1,11 @@
+from argparse import ArgumentParser
+
+
+def init_argparser(prog=None, description=None):
+    argparser = ArgumentParser(prog=prog, description=description)
+    argparser.add_argument('-d', '--debug', action='store_true', help='Debug mode.')
+    argparser.add_argument('-dd', '--verbose', action='store_true', help='Verbose debug mode.')
+    argparser.add_argument('-v', '--version', action='store_true', help='Print the version and exit.')
+
+    return argparser
+

src/zenlib/util/init_logger.py

@@ -0,0 +1,14 @@
+
+from logging import getLogger, StreamHandler
+from zenlib.logging import ColorLognameFormatter
+
+
+def init_logger(name=None, args=None):
+    name = name or __name__
+    logger = getLogger(name)
+    handler = StreamHandler()
+    formatter = ColorLognameFormatter()
+    handler.setFormatter(formatter)
+    logger.addHandler(handler)
+    return logger
+

src/zenlib/util/process_args.py

@@ -0,0 +1,32 @@
+from zenlib.logging import ColorLognameFormatter
+
+
+def process_args(argparser, logger=None):
+    args = argparser.parse_args()
+    if args.version:
+        from importlib.metadata import version
+        print(f"{__package__} {version(__package__)}")
+        exit(0)
+
+    if logger:
+        if args.verbose:
+            logger.setLevel(5)
+            formatter = ColorLognameFormatter('%(levelname)s | %(name)-42s | %(message)s')
+        elif args.debug:
+            logger.setLevel(10)
+            formatter = ColorLognameFormatter('%(levelname)s | %(name)-42s | %(message)s')
+        else:
+            logger.setLevel(20)
+            formatter = ColorLognameFormatter()
+
+        # Add the formatter to the first handler, or add a new handler
+        for handler in logger.handlers:
+            handler.setFormatter(formatter)
+            break
+        else:
+            from logging import StreamHandler
+            handler = StreamHandler()
+            handler.setFormatter(formatter)
+            logger.addHandler(handler)
+
+    return args