From 4c2fb50c681a4223e45e3a4e3b1a85022e02cfa8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?C=C3=A9dric=20Bonhomme?= Date: Sun, 22 May 2016 15:28:13 +0200 Subject: [PATCH] Displays a message when the generator is unknown. --- bin/lsb-set | 3 ++- stegano/lsbset/lsbset.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/bin/lsb-set b/bin/lsb-set index 5340399..6823a55 100755 --- a/bin/lsb-set +++ b/bin/lsb-set @@ -72,7 +72,8 @@ parser.set_defaults(input_image_file = './pictures/Lenna.png', try: generator = getattr(generators, options.generator_function)() except AttributeError as e: - raise e + print("Unknown generator: {}".format(options.generator_function)) + exit(1) if options.hide: if options.secret_message != "" and options.secret_file == "": diff --git a/stegano/lsbset/lsbset.py b/stegano/lsbset/lsbset.py index e39388a..f7f24dc 100644 --- a/stegano/lsbset/lsbset.py +++ b/stegano/lsbset/lsbset.py @@ -174,7 +174,8 @@ if __name__ == '__main__': try: generator = getattr(generators, options.generator_function)() except AttributeError as e: - raise e + print("Unknown generator: {}".format(options.generator_function)) + exit(1) if options.hide: if options.secret_message != "" and options.secret_file == "":