diff --git a/src/cli.py b/src/cli.py --- a/src/cli.py +++ b/src/cli.py @@ -9,8 +9,8 @@ def run(): parser = ArgumentParser() subparsers = parser.add_subparsers() - buildSplitParser(subparsers.add_parser("split")) - buildJoinParser(subparsers.add_parser("join")) + build_split_parser(subparsers.add_parser("split")) + build_join_parser(subparsers.add_parser("join")) parser.set_defaults(func=lambda _: parser.error("missing command")) @@ -18,11 +18,11 @@ def run(): args.func(args) -def buildSplitParser(parser): +def build_split_parser(parser): parser.add_argument("-k", type=int, required=True, help="number of shares necessary for recovering the secret") parser.add_argument("-n", type=int, required=True, help="number of generated shares") - encoding=parser.add_mutually_exclusive_group() + encoding = parser.add_mutually_exclusive_group() encoding.add_argument("--hex", action="store_true", help="encode shares' bytes as a hexadecimal string") encoding.add_argument("--b32", action="store_true", help="encode shares' bytes as a base32 string") encoding.add_argument("--b64", action="store_true", help="encode shares' bytes as a base64 string") @@ -31,8 +31,8 @@ def buildSplitParser(parser): parser.set_defaults(func=_generate) -def buildJoinParser(parser): - encoding=parser.add_mutually_exclusive_group() +def build_join_parser(parser): + encoding = parser.add_mutually_exclusive_group() encoding.add_argument("--hex", action="store_true", help="decode shares' bytes from a hexadecimal string") encoding.add_argument("--b32", action="store_true", help="decode shares' bytes from a base32 string") encoding.add_argument("--b64", action="store_true", help="decode shares' bytes from a base64 string") @@ -43,7 +43,7 @@ def buildJoinParser(parser): def _generate(args): - encoding = getEncoding(args) or "b32" + encoding = get_encoding(args) or "b32" try: shares = generate(args.secret, args.k, args.n, encoding) @@ -54,14 +54,14 @@ def _generate(args): def _reconstruct(args): - encoding = getEncoding(args) + encoding = get_encoding(args) try: print(reconstruct(*args.share, encoding=encoding, raw=args.raw)) except SException as e: print(e) -def getEncoding(args): +def get_encoding(args): if args.hex: return "hex" elif args.b32: return "b32" elif args.b64: return "b64"