chainlib

commit b5ffc4cb57fe08e9aecfbc5c9c1a611176eada04
parent 63f2694494cf61a423a80b7570ecfbca7658f89e
Author: lash <dev@holbrook.no>
Date:   Tue, 14 Feb 2023 06:02:48 +0000

Enable chainlib module listings

Diffstat:
M
chainlib/runnable/gen.py
 | 
60
+++++++++++++++++++++++++++++++++++++++++++++++++++---------

1 file changed, 51 insertions(+), 9 deletions(-)
diff --git a/chainlib/runnable/gen.py b/chainlib/runnable/gen.py
@@ -10,12 +10,16 @@ from chainlib.cli.gen import find_chainlib_modules
 logging.basicConfig(level=logging.WARNING)
 logg = logging.getLogger()
 
+MODE_GEN = 0
+MODE_KEYS = 1
+MODE_LS = 2
 
 ap = argparse.ArgumentParser()
 ap.add_argument('-v', action='store_true', help='verbose logging')
 ap.add_argument('-vv', action='store_true', help='very verbose logging')
 ap.add_argument('-k', action='store_true', help='return keys for command')
-ap.add_argument('module', type=str, help='module to locate and execute')
+ap.add_argument('-l', action='store_true', help='list chainlib modules')
+ap.add_argument('module', type=str, nargs='?', help='module to locate and execute')
 ap.add_argument('command', type=str, nargs='?', help='command to execute on module. default command will be executed if not specified')
 ap.add_argument('arg', type=str, nargs=argparse.REMAINDER)
 args = ap.parse_args()
@@ -25,15 +29,34 @@ if args.vv:
 elif args.v:
     logg.setLevel(logging.INFO)
 
+mode = MODE_GEN
+if args.l:
+    mode = MODE_LS
+elif args.k:
+    mode = MODE_KEYS
+
 def parse_remaining(rargs):
     rargsr = []
+    debug_lvl = logging.WARNING
     for v in rargs:
+        if v == '--help':
+            global ap
+            ap.print_help()
+            sys.exit(0)
         if v == '-vv':
-            logg.setLevel(logging.DEBUG)
+            if debug_lvl > logging.DEBUG:
+                debug_lvl = logging.DEBUG
+            logg.setLevel(debug_lvl)
+            continue
         elif v == '-v':
-            logg.setLevel(logging.INFO)
+            if debug_lvl > logging.INFO:
+                debug_lvl = logging.INFO
+            logg.setLevel(debug_lvl)
+            continue
+        if v == '-l':
+            return (MODE_LS, [],)
         elif v == '-k':
-            return None
+            return (MODE_KEYS, [],)
         else:
             rargsr.append(v)
     r = {}
@@ -51,7 +74,11 @@ def parse_remaining(rargs):
         except IndexError:
             raise ValueError('missing value for attribute: {}'.format(pfx))
         r[k] = v
-    return r
+    return (MODE_GEN, r,)
+
+
+def list_handler(m, cmd, args, is_key_query=False):
+    print(m.__name__)
 
 
 def default_handler(m, cmd, args, is_key_query=False):
@@ -72,11 +99,26 @@ def default_handler(m, cmd, args, is_key_query=False):
 
 
 def main():
-    arg = parse_remaining(args.arg)
+    global mode
+    (parsed_mode, arg) = parse_remaining(args.arg)
+    if parsed_mode > mode:
+        mode = parsed_mode
+    logg.debug('running mode: ' + str(mode))
     is_key_query = False
-    if arg == None:
-        is_key_query = True
-    find_chainlib_modules(fltr=[args.module], cmd=args.command, args=arg, handler=default_handler, is_key_query=is_key_query)
+    hndlr = default_handler
+    cmd = None
+    module = None
+    if mode == MODE_LS:
+        hndlr = list_handler
+    else:
+        module = [args.module]
+        cmd = args.command
+        if mode == MODE_KEYS:
+            if cmd == None:
+                raise ValueError('command required for key query')
+            is_key_query = True
+
+    find_chainlib_modules(fltr=module, cmd=args.command, args=arg, handler=hndlr, is_key_query=is_key_query)
 
 
 if __name__ == '__main__':