123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- """Console script for banana."""
- import argparse
- import logging
- import random
- import sys
- import banana
- def get_codec(args):
- if args.banana:
- return banana.BananaCodec()
- if args.ananas:
- return banana.AnanasCodec()
- if args.ribes:
- return banana.RibesCodec()
- if args.avocado:
- return banana.AvocadoCodec()
- kwargs = {}
- if args.dictionary:
- kwargs["dictionary"] = args.dictionary
- if args.dictstart:
- kwargs["dictstart"] = args.dictstart
- if args.shiftend:
- kwargs["shiftend"] = args.shiftend
- return banana.Codec(**kwargs)
- def main_encode(args):
- print(get_codec(args).encode(args.num))
- def main_decode(args):
- print(get_codec(args).decode(args.word))
- def main_check(args):
- if get_codec(args).is_valid(args.word):
- if not args.quiet:
- print("yes")
- sys.exit(0)
- else:
- if not args.quiet:
- print("no")
- sys.exit(1)
- def main_random(args):
- codec = get_codec(args)
- kwargs = dict(minlength=args.minlength)
- if args.seed:
- kwargs["prng"] = random.Random(args.seed)
- print(codec.random(**kwargs))
- def colon_separated_list(s):
- return s.split(":")
- def main():
- parser = argparse.ArgumentParser(description="Convert dec number to banana")
- parser.add_argument(
- "--log-level", choices=["DEBUG", "INFO", "WARN", "ERROR"], default="WARN"
- )
- parser.add_argument("--ananas", action="store_true")
- parser.add_argument("--avocado", action="store_true")
- parser.add_argument("--banana", action="store_true")
- parser.add_argument("--ribes", action="store_true")
- parser.add_argument(
- "--dictionary",
- help="Set dictionary in colon-separated list",
- type=colon_separated_list,
- )
- parser.add_argument(
- "--dictstart", help="Set starting dictionary", type=int, default=0
- )
- parser.add_argument(
- "--shiftend", help="Set shift for ending dictionary", type=int, default=0
- )
- sub = parser.add_subparsers()
- encode = sub.add_parser("encode", help="Convert numbers to words")
- encode.add_argument("num", type=int)
- encode.set_defaults(func=main_encode)
- decode = sub.add_parser("decode", help="Convert words to numbers")
- decode.add_argument("word")
- decode.set_defaults(func=main_decode)
- check = sub.add_parser("check", help="Convert words to numbers")
- check.add_argument("word")
- check.add_argument("--quiet", "-q", action="store_true")
- check.set_defaults(func=main_check)
- rand = sub.add_parser("random", help="Generate random banana")
- rand.add_argument("--minlength", help="Set minimum length", type=int, default=6)
- rand.add_argument("--seed", type=int, default=None)
- rand.set_defaults(func=main_random)
- args = parser.parse_args()
- if not hasattr(args, "func"):
- print("You need to select one subcommand. \nUse --help", file=sys.stderr)
- # parser.print_help()
- sys.exit(1)
- logging.basicConfig(level=args.log_level)
- args.func(args)
- if __name__ == "__main__":
- # pragma: no cover
- main()
|