diff --git a/src/cli.py b/src/cli.py --- a/src/cli.py +++ b/src/cli.py @@ -2,7 +2,7 @@ from argparse import ArgumentParser -from shamira import generate, reconstruct +from shamira import generate, reconstruct, SException def run(): @@ -49,16 +49,16 @@ def _generate(args): shares=generate(args.secret,args.k,args.n,encoding) for s in shares: print(s) - except ValueError as e: - print("operation failed: ",e) + except SException as e: + print(e) def _reconstruct(args): encoding=getEncoding(args) try: print(reconstruct(*args.share,encoding=encoding,raw=args.raw)) - except ValueError as e: - print("operation failed: ",e) + except SException as e: + print(e) def getEncoding(args):