Update applications.py

Author: cadonuno

veracode_api_py/applications.py

@@ -33,18 +33,18 @@ def get_by_name (self,appname: str):
         params = {"name": parse.quote(appname)}
         return APIHelper()._rest_paged_request(uri="appsec/v1/applications",method="GET",element="applications",params=params)
 
-    def create(self,app_name:str ,business_criticality, business_unit: UUID=None, teams=[], policy_guid:UUID=None,
-                custom_fields=[], bus_owner_name=None, bus_owner_email=None, git_repo_url=None, custom_kms_alias: str=None):
+    def create(self,app_name:str ,business_criticality, description: str, business_unit: UUID=None, teams=[], policy_guid:UUID=None,
+                custom_fields=[], bus_owner_name=None, bus_owner_email=None, git_repo_url=None):
         return self._create_or_update("CREATE",app_name=app_name,business_criticality=business_criticality,
-                                      business_unit=business_unit,teams=teams, policy_guid=policy_guid, 
+                                      description=description,business_unit=business_unit,teams=teams, policy_guid=policy_guid, 
                                       custom_fields=custom_fields, bus_owner_name=bus_owner_name, 
-                                      bus_owner_email=bus_owner_email, git_repo_url=git_repo_url, custom_kms_alias=custom_kms_alias)
+                                      bus_owner_email=bus_owner_email, git_repo_url=git_repo_url)
 
-    def update(self,guid: UUID,app_name:str ,business_criticality, business_unit: UUID=None, 
+    def update(self,guid: UUID,app_name:str, description: str, business_criticality, business_unit: UUID=None, 
                teams=[], policy_guid:UUID=None, custom_fields=[],
                bus_owner_name=None,bus_owner_email=None, git_repo_url=None):
         return self._create_or_update("UPDATE",app_name=app_name,business_criticality=business_criticality,
-                                      business_unit=business_unit,teams=teams,guid=guid, 
+                                      description=description,business_unit=business_unit,teams=teams,guid=guid, 
                                       policy_guid=policy_guid, custom_fields=custom_fields, 
                                       bus_owner_name=bus_owner_name,bus_owner_email=bus_owner_email,
                                       git_repo_url=git_repo_url)
@@ -53,9 +53,9 @@ def delete(self,guid: UUID):
         uri = 'appsec/v1/applications/{}'.format(guid)
         return APIHelper()._rest_request(uri,'DELETE')
 
-    def _create_or_update(self,method,app_name: str,business_criticality, business_unit: UUID=None, 
+    def _create_or_update(self,method,app_name: str,description: str,business_criticality, business_unit: UUID=None, 
                           teams=[],guid=None,policy_guid:UUID=None, custom_fields=[], 
-                          bus_owner_name=None,bus_owner_email=None,git_repo_url=None,custom_kms_alias:str=None):
+                          bus_owner_name=None,bus_owner_email=None,git_repo_url=None):
         if method == 'CREATE':
             uri = 'appsec/v1/applications'
             httpmethod = 'POST'
@@ -70,6 +70,10 @@ def _create_or_update(self,method,app_name: str,business_criticality, business_u
 
         app_def = {'name':app_name, 'business_criticality':business_criticality}
 
+        if (description != None):
+            desc = { 'description': description}
+            app_def.update(desc)
+
         if policy_guid:
             app_def.update({"policies": [{'guid': policy_guid}]})
 
@@ -95,10 +99,6 @@ def _create_or_update(self,method,app_name: str,business_criticality, business_u
             gru = { 'git_repo_url': git_repo_url}
             app_def.update(gru)
 
-        if (custom_kms_alias != None) & (method=='CREATE'):
-            # custom_kms_alias currently only supported at creation
-            app_def.update({"custom_kms_alias": custom_kms_alias})
-
         payload = json.dumps({"profile": app_def})
         return APIHelper()._rest_request(uri,httpmethod,body=payload)