Added exception handling for prettytable and serial modules - gracefuly handle missing imports.

Author: PrzemekWirkus

workspace_tools/host_tests/host_test.py

@@ -15,8 +15,14 @@
 limitations under the License.
 """
 
+# Check if 'serial' module is installed
+try:
+    from serial import Serial
+except ImportError, e:
+    print "Error: Can't import 'serial' module: %s"% e
+    exit(-1)
+
 from optparse import OptionParser
-from serial import Serial
 from time import sleep
 from sys import stdout
 

workspace_tools/singletest.py

@@ -70,14 +70,27 @@
 
 """
 
+# Check if 'prettytable' module is installed
+try:
+    from prettytable import PrettyTable
+except ImportError, e:
+    print "Error: Can't import 'prettytable' module: %s"% e
+    exit(-1)
+
+# Check if 'serial' module is installed
+try:
+    from serial import Serial
+except ImportError, e:
+    print "Error: Can't import 'serial' module: %s"% e
+    exit(-1)
+
 import sys
 import json
 import optparse
 import pprint
 import re
 import os
 from types import ListType
-from prettytable import PrettyTable
 
 from os.path import join, abspath, dirname, exists, basename
 from shutil import copy