Error handling added

Author: Murat Kumykov

blackduck/HubRestApi.py

@@ -140,7 +140,12 @@ def get_auth_token(self):
                 verify=not self.config['insecure']
             )
             csrf_token = response.headers['X-CSRF-TOKEN']
-            bearer_token = json.loads(response.content.decode('utf-8'))['bearerToken']
+            try:
+              bearer_token = json.loads(response.content.decode('utf-8'))['bearerToken']
+            except json.decoder.JSONDecodeError as e:
+              import traceback
+              traceback.print_exc()
+              raise Exception("Failed to obtain bearer token, check for valid authentucation token")
             return (bearer_token, csrf_token, None)
         else:
             authendpoint="/j_spring_security_check"
@@ -667,6 +672,7 @@ def get_projects(self, limit=100, parameters={}):
         url = self._get_projects_url() + self._get_parameter_string(parameters)
         headers['Accept'] = 'application/vnd.blackducksoftware.project-detail-4+json'
         response = requests.get(url, headers=headers, verify = not self.config['insecure'])
+        print (response)
         jsondata = response.json()
         return jsondata
 

examples/custom_signatures_export.py

@@ -146,6 +146,8 @@ def main(argv=None): # IGNORE:C0111
         ### handle keyboard interrupt ###
         return 0
     except Exception as e:
+        import traceback
+        traceback.print_exc()
         if DEBUG or TESTRUN:
             raise(e)
         indent = len(program_name) * " "
@@ -171,4 +173,4 @@ def main(argv=None): # IGNORE:C0111
         stats.print_stats()
         statsfile.close()
         sys.exit(0)
-    sys.exit(main())
+    sys.exit(main())

examples/custom_signatures_import.py

@@ -60,15 +60,21 @@ def read_metadata_file(metadata_path):
 def create_projects(hub, metadata):
     for project_name in metadata['content']:
         for version_name in metadata['content'][project_name]:
-            version = hub.get_or_create_project_version(project_name, version_name)
-        project = hub.get_project_by_name(project_name)
-        if not project['customSignatureEnabled']:
-            print ("Enabling custom signature for {}".format(project['name']))
-            project['customSignatureEnabled'] = True
-            response = hub.update_project_settings(project, new_settings=project)
-            print (response)
-        else:
-            print ("Custom signature for {} already enabled".format(project_name))
+            try:
+                version = hub.get_or_create_project_version(project_name, version_name)
+            except:
+                print ("Failed to create project {} version {}".format(project_name, version_name))
+        try:
+            project = hub.get_project_by_name(project_name)
+            if not project['customSignatureEnabled']:
+                print ("Enabling custom signature for {}".format(project['name']))
+                project['customSignatureEnabled'] = True
+                response = hub.update_project_settings(project, new_settings=project)
+                print (response)
+            else:
+                print ("Custom signature for {} already enabled".format(project_name))
+        except:
+            print ("Failed to set custom signature flag for {}".format(project_name))
 
 def upload_scan_data(hub, workdir):
     with (os.scandir(workdir)) as entries:
@@ -130,6 +136,8 @@ def main(argv=None): # IGNORE:C0111
 
         return 0
     except Exception as e:
+        import traceback
+        traceback.print_stack()
         if DEBUG or TESTRUN:
             raise(e)
         indent = len(program_name) * " "
@@ -155,4 +163,4 @@ def main(argv=None): # IGNORE:C0111
         stats.print_stats()
         statsfile.close()
         sys.exit(0)
-    sys.exit(main())
+    sys.exit(main())