chainlib

commit c8357c1fd8f9d27b9eef67f7160e431b4dce5590
parent b340c6b0b014bad1b53d802dd855cf025d62b9b9
Author: lash <dev@holbrook.no>
Date:   Wed,  9 Aug 2023 10:35:32 +0100

Add flag generation script

Diffstat:
A
scripts/chainlib-flags.py
 | 
51
+++++++++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 51 insertions(+), 0 deletions(-)
diff --git a/scripts/chainlib-flags.py b/scripts/chainlib-flags.py
@@ -0,0 +1,51 @@
+# standard imports
+import argparse
+import sys
+import math
+
+# local imports
+from chainlib.cli.arg import ArgFlag
+
+
+def value_int(v):
+    return str(v)
+
+def value_hex(v):
+    sz = math.log(v, 2)
+    if v == 0:
+        return "00"
+    sz = math.ceil(sz)
+    sz = (sz / 8) + 1
+    return '0x' + v.to_bytes(int(sz), byteorder='big').hex()
+
+argflag = ArgFlag()
+
+argparser = argparse.ArgumentParser()
+argparser.add_argument('-l', '--list', action='store_true', help='List all flag names and values')
+argparser.add_argument('-x', '--hex', action='store_true', help='Output all values in hex')
+argparser.add_argument('flag', type=str, nargs='*', help='One or more flag name to calculate flag value for')
+args = argparser.parse_args(sys.argv[1:])
+
+m = value_int
+if args.hex:
+    m = value_hex
+
+if args.list:
+    o = dict(argflag)
+    for k in o.keys():
+        r = k.split("=")
+        v = m(o[k])
+        if len(r) == 1:
+            print(k + ' ' + v)
+        else:
+            print(r[0] + ' ' + v + ' (' + r[1] + ')')
+    sys.exit(0)
+
+if len(args.flag) == 0:
+    print(m(argflag.all))
+    sys.exit(0)
+
+r = 0
+for k in args.flag:
+    r += argflag.val(k)
+print(m(r))