Add patch from Milosz to handle LAVA authentication credential error.

1 files changed, 9 insertions, 3 deletions

diff --git a/report_automation.py b/report_automation.py
index 4e84c47..b3311d8 100755
--- a/report_automation.py
+++ b/report_automation.py
@@ -5,6 +5,7 @@ import json
 import re
 import os
 import ast
+import sys
 import zipfile
 import base64
 import shutil
@@ -74,13 +75,18 @@ class LAVA(object):
             exit(-1)
 
     def get_job_info(self, job_no):
-        job_info = {}
+        job_status = None
+        job_info   = {}
         try:
             job_status      = self.server.scheduler.job_status(job_no)
-        except Exception as e:
+        except xmlrpclib.ProtocolError as err:
+            if err.errcode == 401:
+                logger.error("Invalid LAVA credentials. Please check your .netrc")
+                sys.exit(1)
+        except:
             logger.warning("Can not get any information for test job: %s" % job_no)
 
-        if job_status['job_status'] != 'Complete':
+        if job_status == None or job_status['job_status'] != 'Complete':
             logger.warning("!!Job #%s is not completed.\n\tJob status: %s!! " % (job_no, job_status['job_status']))
         else:
             job_detail                  = self.server.scheduler.job_details(job_no)