branches_permissions (#376)

made repository optional added 'all_branches_permissions'
atlassian-api · Oct 28, 2019 · bc1cb99 · bc1cb99
1 parent c57b96f
commit bc1cb99
Showing 1 changed file with 29 additions and 4 deletions.
diff --git a/atlassian/bitbucket.py b/atlassian/bitbucket.py
@@ -1065,22 +1065,47 @@ def get_content_of_file(self, project, repository, filename, at=None, markup=Non
             params['markup'] = markup
         return self.get(url, params=params, not_json_response=True, headers=headers)
 
-    def get_branches_permissions(self, project, repository, limit=25):
+    def get_branches_permissions(self, project, repository=None, start=0, limit=25):
         """
         Get branches permissions from a given repo
         :param project:
         :param repository:
+        :param start:
         :param limit:
         :return:
         """
-        url = 'rest/branch-permissions/2.0/projects/{project}/repos/{repository}/restrictions'.format(
-            project=project,
-            repository=repository)
+        if repository != None:
+            url = 'rest/branch-permissions/2.0/projects/{project}/repos/{repository}/restrictions'.format(
+                project=project,
+                repository=repository)
+        else:
+            url = 'rest/branch-permissions/2.0/projects/{project}/restrictions'.format(
+                project=project)
+
         params = {}
         if limit:
             params['limit'] = limit
+        if start:
+            params['start'] = start
         return self.get(url, params=params)
 
+    def all_branches_permissions(self, project, repository=None):
+        """
+        Get branches permissions from a given repo
+        :param project:
+        :param repository:
+        :return:
+        """
+        start = 0
+        branches_permissions = []
+        response = self.get_branches_permissions(project=project, repository=repository,  start=start)
+        branches_permissions  += response.get('values')
+        while not response.get('isLastPage'):
+            start = response.get('nextPageStart')
+            response = self.get_branches_permissions(project=project, repository=repository,  start=start)
+            branches_permissions  += response.get('values')
+        return branches_permissions
+
     def reindex(self):
         """
         Rebuild the bundled Elasticsearch indexes for Bitbucket Server