From c51c52ce4843e68d11f79205845d93dafd9549f5 Mon Sep 17 00:00:00 2001 From: Attila Farkas <x394097@fi.muni.cz> Date: Tue, 22 Oct 2019 09:49:54 +0200 Subject: [PATCH] remove traceback --- create.py | 4 +++- modules/device_creator.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/create.py b/create.py index 40a6116..b3ee2c7 100644 --- a/create.py +++ b/create.py @@ -7,6 +7,9 @@ import jinja2 from modules.file_generator import generate_vagrantfile, generate_ansible_files from modules.device_creator import open_file +""" Supressing error traceback. """ +sys.tracebacklimit = 0 + if len(sys.argv) != 2: print("Error: Expecting 1 argument (yml file).") sys.exit() @@ -15,4 +18,3 @@ device_definitions = open_file(str(sys.argv[1])) generate_vagrantfile(device_definitions) generate_ansible_files(device_definitions) - diff --git a/modules/device_creator.py b/modules/device_creator.py index d91bf53..c8e2906 100644 --- a/modules/device_creator.py +++ b/modules/device_creator.py @@ -17,7 +17,7 @@ def open_file(file_name): input_file = open(str(file_name)) return yaml.safe_load(input_file) except IOError: - print("Error: Cannot find a required file.") + print("Error: Cannot find a required file: " + str(file_name)) def _add_provisioning(hostname, host_definitions): """ Adds provisioning to the device if the file exists. """ -- GitLab