From 38e0ee81e9db0984540d42034fb8308959d4e1e3 Mon Sep 17 00:00:00 2001 From: Alyssa Rosenzweig Date: Mon, 25 Jun 2018 15:32:31 -0400 Subject: [PATCH] Graceful server --- libremanage.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/libremanage.py b/libremanage.py index 34a8a47..06f904b 100644 --- a/libremanage.py +++ b/libremanage.py @@ -55,8 +55,14 @@ if len(sys.argv) != 3: die_with_usage("Incorrect number of arguments") def get_server_handle(name): - server = CONFIG["servers"][name] + try: + server = CONFIG["servers"][name] + except KeyError: + die_with_usage("Unknown server, please configure") + + # Associate manager configuration server["ssh"] = CONFIG["managers"][server["manager"]] + return server def set_server_power(state, server): -- 2.25.1