Bitbucket: set default brnach name method

Author: Gonchik Tsymzhitov

atlassian/bamboo.py

@@ -9,7 +9,6 @@
 
 
 class Bamboo(AtlassianRestAPI):
-
     """ Private methods """
 
     def _get_generator(self, path, elements_key='results', element_key='result', data=None, flags=None,
@@ -45,7 +44,7 @@ def _get_generator(self, path, elements_key='results', element_key='result', dat
                 yield e
         try:
             results = response[elements_key]
-            size = 0 
+            size = 0
             # Check if we still can get results
             if size > max_results or results['size'] == 0:
                 return
@@ -56,9 +55,9 @@ def _get_generator(self, path, elements_key='results', element_key='result', dat
         except TypeError:
             logging.error("Broken response: {}".format(response))
             yield response
-        
 
-    def base_list_call(self, resource, expand, favourite, clover_enabled, max_results, label=None, start_index=0, **kwargs):
+    def base_list_call(self, resource, expand, favourite, clover_enabled, max_results, label=None, start_index=0,
+                       **kwargs):
         flags = []
         params = {'max-results': max_results}
         if expand:
@@ -207,7 +206,8 @@ def get_vcs_branches(self, plan_key, max_results=25):
     """ Build results """
 
     def results(self, project_key=None, plan_key=None, job_key=None, build_number=None, expand=None, favourite=False,
-                clover_enabled=False, issue_key=None, label=None, start_index=0, max_results=25, include_all_states=False):
+                clover_enabled=False, issue_key=None, label=None, start_index=0, max_results=25,
+                include_all_states=False):
         """
         Get results as generic method
         :param project_key:
@@ -258,7 +258,8 @@ def latest_results(self, expand=None, favourite=False, clover_enabled=False, lab
         :return:
         """
         return self.results(expand=expand, favourite=favourite, clover_enabled=clover_enabled,
-                            label=label, issue_key=issue_key, start_index=start_index, max_results=max_results, include_all_states=include_all_states)
+                            label=label, issue_key=issue_key, start_index=start_index, max_results=max_results,
+                            include_all_states=include_all_states)
 
     def project_latest_results(self, project_key, expand=None, favourite=False, clover_enabled=False, label=None,
                                issue_key=None, start_index=0, max_results=25, include_all_states=False):
@@ -276,7 +277,8 @@ def project_latest_results(self, project_key, expand=None, favourite=False, clov
         :return:
         """
         return self.results(project_key, expand=expand, favourite=favourite, clover_enabled=clover_enabled,
-                            label=label, issue_key=issue_key, start_index=start_index, max_results=max_results, include_all_states=include_all_states)
+                            label=label, issue_key=issue_key, start_index=start_index, max_results=max_results,
+                            include_all_states=include_all_states)
 
     def plan_results(self, project_key, plan_key, expand=None, favourite=False, clover_enabled=False, label=None,
                      issue_key=None, start_index=0, max_results=25, include_all_states=False):
@@ -295,7 +297,8 @@ def plan_results(self, project_key, plan_key, expand=None, favourite=False, clov
         :return:
         """
         return self.results(project_key, plan_key, expand=expand, favourite=favourite, clover_enabled=clover_enabled,
-                            label=label, issue_key=issue_key, start_index=start_index, max_results=max_results, include_all_states=include_all_states)
+                            label=label, issue_key=issue_key, start_index=start_index, max_results=max_results,
+                            include_all_states=include_all_states)
 
     def build_result(self, build_key, expand=None, include_all_states=False):
         """
@@ -334,7 +337,7 @@ def build_latest_result(self, plan_key, expand=None, include_all_states=False):
     def delete_build_result(self, build_key):
         """
         Deleting result for specific build
-        :param build_key: Take full build key, example: PROJ-PLAN-8
+        :param build_key: Take full build key, example: PROJECT-PLAN-8
         """
         custom_resource = '/build/admin/deletePlanResults.action'
         build_key = build_key.split('-')

atlassian/bitbucket.py

@@ -19,16 +19,6 @@ def project_list(self, limit=None):
             params['limit'] = limit
         return (self.get('rest/api/1.0/projects', params=params) or {}).get('values')
 
-    def repo_list(self, project_key):
-        """
-        Provide the repository list in a specific BB project
-        :param project_key: The Project Key ID you need to list
-        return:
-        """
-        url = 'rest/api/1.0/projects/{projectKey}/repos?limit=1000'.format(
-            projectKey=project_key)
-        return self.get(url)
-
     def project(self, key):
         """
         Provide project info
@@ -57,11 +47,10 @@ def update_project(self, key, **params):
         """
         Update project
         :param key:
-        :param **params:
         :return:
         """
         data = self.project(key)
-        if not 'errors' in data:
+        if 'errors' not in data:
             data.update(params)
             url = 'rest/api/1.0/projects/{0}'.format(key)
             return self.put(url, data=data)
@@ -72,7 +61,7 @@ def update_project(self, key, **params):
     def project_avatar(self, key, content_type='image/png'):
         """
         Get project avatar
-
+        :param content_type:
         :param key:
         :return:
         """
@@ -86,8 +75,9 @@ def project_avatar(self, key, content_type='image/png'):
     def set_project_avatar(self, key, icon, content_type='image/png'):
         """
         Set project avatar
-
-        :param key:
+        :param key: A Project key
+        :param icon:
+        :param content_type:
         :return:
         """
         headers = {'X-Atlassian-Token': 'no-check'}
@@ -174,18 +164,18 @@ def project_remove_user_permissions(self, project_key, username):
             username=username)
         return self.delete(url)
 
-    def project_grant_group_permissions(self, project_key, groupname, permission):
+    def project_grant_group_permissions(self, project_key, group_name, permission):
         """
         Grant the specified project permission to an specific group
         :param project_key: project key involved
-        :param groupname: group to be granted
+        :param group_name: group to be granted
         :param permission: the project permissions available are 'PROJECT_ADMIN', 'PROJECT_WRITE' and 'PROJECT_READ'
         :return:
         """
-        url = 'rest/api/1.0/projects/{project_key}/permissions/groups?permission={permission}&name={groupname}'.format(
+        url = 'rest/api/1.0/projects/{project_key}/permissions/groups?permission={permission}&name={group_name}'.format(
             project_key=project_key,
             permission=permission,
-            groupname=groupname)
+            group_name=group_name)
         return self.put(url)
 
     def project_remove_group_permissions(self, project_key, groupname):
@@ -199,9 +189,9 @@ def project_remove_group_permissions(self, project_key, groupname):
         :param groupname: group to be granted
         :return:
         """
-        url = 'rest/api/1.0/projects/{project_key}/permissions/groups?name={groupname}'.format(
+        url = 'rest/api/1.0/projects/{project_key}/permissions/groups?name={group_name}'.format(
             project_key=project_key,
-            groupname=groupname)
+            group_name=groupname)
         return self.delete(url)
 
     def repo_grant_user_permissions(self, project_key, repo_key, username, permission):
@@ -273,6 +263,8 @@ def repo_remove_group_permissions(self, project_key, repo_key, groupname, permis
         :return:
         """
         params = {'name': groupname}
+        if permission:
+            params['permission'] = permission
         url = 'rest/api/1.0/projects/{project_key}/repos/{repo_key}/permissions/groups'.format(project_key=project_key,
                                                                                                repo_key=repo_key)
         return self.delete(url, params=params)
@@ -506,6 +498,20 @@ def get_default_branch(self, project, repository):
                                                                                            repository=repository)
         return self.get(url)
 
+    def set_default_branch(self, project, repository, ref_branch_name):
+        """
+        Update the default branch of a repository.
+        The authenticated user must have REPO_ADMIN permission for the specified repository to call this resource.
+        :param project: project key
+        :param repository: repo slug
+        :param ref_branch_name: ref name like refs/heads/master
+        :return:
+        """
+        data = {'id': ref_branch_name}
+        url = 'rest/api/1.0/projects/{project}/repos/{repository}/branches/default'.format(project=project,
+                                                                                           repository=repository)
+        return self.put(url, data=data)
+
     def create_branch(self, project_key, repository, name, start_point, message=""):
         """Creates a branch using the information provided in the request.
 
@@ -587,6 +593,7 @@ def get_pull_requests(self, project, repository, state='OPEN', order='newest', l
                                 (as in: "oldest first") or NEWEST.
         :param limit:
         :param start:
+        :param at:
         :return:
         """
         url = 'rest/api/1.0/projects/{project}/repos/{repository}/pull-requests'.format(project=project,
@@ -613,7 +620,7 @@ def get_pull_requests(self, project, repository, state='OPEN', order='newest', l
             pr_list += (response or {}).get('values')
         return pr_list
 
-    def get_pull_requests_activities(self, project, repository, pull_request_id):
+    def get_pull_requests_activities(self, project, repository, pull_request_id, start=0):
         """
         Get pull requests activities
         :param project:
@@ -625,7 +632,7 @@ def get_pull_requests_activities(self, project, repository, pull_request_id):
             project=project,
             repository=repository,
             pullRequestId=pull_request_id)
-        params = {'start': 0}
+        params = {'start': start}
         response = self.get(url, params=params)
         if 'values' not in response:
             return []
@@ -807,7 +814,7 @@ def update_pull_request_comment(self, project, repository, pull_request_id, comm
         that must match the server's version of the comment
         or the update will fail.
         """
-        url = "/rest/api/1.0/projects/{project}/repos/{repository}/pull-requests/{pull_request_id}/comments/{comment_id}".format(
+        url = "rest/api/1.0/projects/{project}/repos/{repository}/pull-requests/{pull_request_id}/comments/{comment_id}".format(
             project=project, repository=repository, pull_request_id=pull_request_id, comment_id=comment_id
         )
         payload = {
@@ -1321,6 +1328,8 @@ def upload_plugin(self, plugin_path):
     def upload_file(self, project, repository, content, message, branch, filename):
         """
         Upload new file for given branch.
+        :param project:
+        :param repository:
         :param content:
         :param message:
         :param branch:
@@ -1392,8 +1401,10 @@ def delete_code_insights_report(self, project_key, repository_slug, commit_id, r
     def create_code_insights_report(self, project_key, repository_slug, commit_id, report_key, report_title,
                                     **report_params):
         """
-        Create a new insight report, or replace the existing one if a report already exists for the given repository, commit, and report key. 
-        A request to replace an existing report will be rejected if the authenticated user was not the creator of the specified report.
+        Create a new insight report, or replace the existing one
+        if a report already exists for the given repository, commit, and report key.
+        A request to replace an existing report will be rejected
+        if the authenticated user was not the creator of the specified report.
         For further information visit: https://docs.atlassian.com/bitbucket-server/rest/6.6.1/bitbucket-code-insights-rest.html
         :projectKey: str
         :repositorySlug: str

atlassian/jira.py

@@ -384,8 +384,8 @@ def add_version(self, project_key, project_id, version, is_archived=False, is_re
         :param project_key: the project key
         :param project_id: the project id
         :param version: the new project version to add
-        :is_archived:
-        :is_released:
+        :param is_archived:
+        :param is_released:
         :return:
         """
         payload = {'name': version, 'archived': is_archived, 'released': is_released, 'project': project_key,
@@ -579,7 +579,7 @@ def get_custom_fields(self, search=None, start=1, limit=50):
         if limit:
             params['maxResults'] = limit
         return self.get(url, params=params)
-    
+
     def create_custom_field(self, name, type, search_key=None, description=None):
         """
         Creates a custom field with the given name and type
@@ -1771,6 +1771,7 @@ def tempo_timesheets_approval_worklog_report(self, user_key, period_start_date):
             params['periodStartDate'] = period_start_date
         if user_key:
             params['userKey'] = user_key
+        return self.get(url, params=params)
 
     def tempo_timesheets_get_required_times(self, from_date, to_date, user_name):
         """

atlassian/rest_client.py

@@ -201,7 +201,7 @@ def get(self, path, data=None, flags=None, params=None, headers=None, not_json_r
         :return:
         """
         response = self.request('GET', path=path, flags=flags, params=params, data=data, headers=headers,
-                              trailing=trailing)
+                                trailing=trailing)
         if self.advanced_mode:
             return response
         if not_json_response: