From 8bd555fc73e014ff0ba762ab47f8ca5a5ce3cdde Mon Sep 17 00:00:00 2001 From: Yuval Nativ Date: Mon, 3 Feb 2014 09:58:57 +0200 Subject: [PATCH] Error Handeling Fixed issues with malformed 'set' and 'use' methods. --- imports/terminal_handler.py | 32 +++++++++++++++++++------------- 1 file changed, 19 insertions(+), 13 deletions(-) diff --git a/imports/terminal_handler.py b/imports/terminal_handler.py index 7bccc4f..7e32744 100644 --- a/imports/terminal_handler.py +++ b/imports/terminal_handler.py @@ -82,16 +82,19 @@ class Controller: self.MainMenu() if re.match('^set', cmd): - cmd = re.split('\s+', cmd) - print cmd[1] + ' => ' + cmd[2] - if cmd[1] == 'arch': - self.arch = cmd[2] - if cmd[1] == 'plat': - self.plat = cmd[2] - if cmd[1] == 'lang': - self.lang = cmd[2] - if cmd[1] == 'type': - self.type = cmd[2] + try: + cmd = re.split('\s+', cmd) + print cmd[1] + ' => ' + cmd[2] + if cmd[1] == 'arch': + self.arch = cmd[2] + if cmd[1] == 'plat': + self.plat = cmd[2] + if cmd[1] == 'lang': + self.lang = cmd[2] + if cmd[1] == 'type': + self.type = cmd[2] + except: + print 'Need to use the set method with two arguments.' cmd = '' self.MainMenu() @@ -125,9 +128,12 @@ class Controller: self.MainMenu() if re.match('^use', cmd): - cmd = re.split('\s+', cmd) - self.currentmodule = cmd[1] - cmd = '' + try: + cmd = re.split('\s+', cmd) + self.currentmodule = cmd[1] + cmd = '' + except: + print 'The use method needs an argument.' self.MainMenu() if cmd == 'back':