Cleanly abort startup if RTSLibError happens.

1 files changed, 7 insertions, 2 deletions

diff --git a/scripts/rtsadmin b/scripts/rtsadmin
index a365b94..3601cc5 100755
--- a/scripts/rtsadmin
+++ b/scripts/rtsadmin
@@ -22,6 +22,7 @@ All rights reserved.
 
 from os import getuid
 from rtsadmin import UIRoot
+from rtslib import RTSLibError
 from configshell import ConfigShell
 
 class RTSAdmin(ConfigShell):
@@ -62,8 +63,12 @@ def main():
         shell.con.display('You are not root, disabling privileged commands.')
         shell.con.display('')
 
-    root_node.refresh()
-    shell.run_interactive()
+    try:
+        root_node.refresh()
+    except RTSLibError, error:
+        shell.con.display(shell.con.render_text(str(error), 'red'))
+    else:
+        shell.run_interactive()
 
 if __name__ == "__main__":
     main()