add method for updating teams

Author: tjarrettveracode

README.md

@@ -69,6 +69,7 @@ The following methods call Veracode REST APIs and return JSON.
 
 - `get_teams(all_for_org)`: get the list of teams for the user, or (if `all_for_org` is `True`) all teams in the organization.
 - `create_team(team_name,business_unit,members)`: create a team named `team_name`. Optionally pass the business unit guid and/or a list of user names to add to the team.
+- `update_team(team_guid,team_name(opt),business_unit(opt),members(opt))`: update the team identified by `team_guid` with the provided information.
 - `delete_team(team_guid)`: delete the team identified by `team_guid`.
 
 #### Business Units

setup.py

@@ -3,13 +3,13 @@
 setup(
   name = 'veracode_api_py',         
   packages = ['veracode_api_py'],   
-  version = '0.9.3',      
+  version = '0.9.4',      
   license='MIT',        
   description = 'Python helper library for working with the Veracode APIs. Handles retries, pagination, and other features of the modern Veracode REST APIs.',   
   author = 'Tim Jarrett',                  
   author_email = 'tjarrett@veracode.com',      
   url = 'https://github.com/tjarrettveracode',   
-  download_url = 'https://github.com/tjarrettveracode/veracode-api-py/archive/v_092.tar.gz',    
+  download_url = 'https://github.com/tjarrettveracode/veracode-api-py/archive/v_094.tar.gz',    
   keywords = ['veracode', 'veracode-api'],   
   install_requires=[            
           'veracode-api-signing'

veracode_api_py/api.py

@@ -339,6 +339,29 @@ def create_team (self, team_name, business_unit=None, members=[]):
         payload = json.dumps(team_def)
         return self._rest_request('api/authn/v2/teams','POST',body=payload)
 
+    def update_team (self, team_guid, team_name="", business_unit=None, members=[]):
+        requestbody = {}
+        
+        if team_name != "":
+            requestbody.update({"team_name": team_name})
+
+        if business_unit != None:
+            requestbody.update({"business_unit": {"bu_id": business_unit}})
+
+        if len(members) > 0:
+            users = []
+            for member in members:
+                users.append({"user_name": member})
+            requestbody.update({"users": users})
+
+        if requestbody == {}:
+            logging.error("No update specified for team {}".format(team_guid))
+
+        payload = json.dumps(requestbody)
+        params = {"partial":True, "incremental":True}
+        uri = 'api/authn/v2/teams/{}'.format(team_guid)
+        return self._rest_request(uri,'PUT',body=payload,params=params)
+
     def delete_team (self, team_guid):
         uri = 'api/authn/v2/teams/{}'.format(team_guid)
         return self._rest_request(uri,"DELETE")
@@ -361,8 +384,11 @@ def get_workspace_by_name(self,name):
         return self._rest_paged_request(self.sca_base_url,"GET",params=request_params,element="workspaces")
 
     def create_workspace(self,name):
-        r = self._rest_request(self.sca_base_url,"POST",body=name,fullresponse=True)
-        return r.headers.get('location','')
+        #pass payload with name, return guid to workspace
+        payload = json.dumps({"name": name})
+        r = self._rest_request(self.sca_base_url,"POST",body=payload,fullresponse=True)
+        loc = r.headers.get('location','')
+        return loc.split("/")[-1]
 
     def add_workspace_team(self,workspace_guid,team_id):
         return self._rest_request(self.sca_base_url + "/{}/teams/{}".format(workspace_guid,team_id),"PUT")