Displays a message when the generator is unknown.

This commit is contained in:
Cédric Bonhomme 2016-05-22 15:28:13 +02:00
parent f8668a978e
commit 4c2fb50c68
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
2 changed files with 4 additions and 2 deletions

View file

@ -72,7 +72,8 @@ parser.set_defaults(input_image_file = './pictures/Lenna.png',
try: try:
generator = getattr(generators, options.generator_function)() generator = getattr(generators, options.generator_function)()
except AttributeError as e: except AttributeError as e:
raise e print("Unknown generator: {}".format(options.generator_function))
exit(1)
if options.hide: if options.hide:
if options.secret_message != "" and options.secret_file == "": if options.secret_message != "" and options.secret_file == "":

View file

@ -174,7 +174,8 @@ if __name__ == '__main__':
try: try:
generator = getattr(generators, options.generator_function)() generator = getattr(generators, options.generator_function)()
except AttributeError as e: except AttributeError as e:
raise e print("Unknown generator: {}".format(options.generator_function))
exit(1)
if options.hide: if options.hide:
if options.secret_message != "" and options.secret_file == "": if options.secret_message != "" and options.secret_file == "":