add support for project builds

Gauvain Pocentek · Gauvain Pocentek · commit ebf36b81f122 · 2016-01-30T19:04:30.000+01:00
diff --git a/gitlab/exceptions.py b/gitlab/exceptions.py
@@ -75,6 +75,14 @@ class GitlabTransferProjectError(GitlabOperationError):
     pass
 
 
+class GitlabBuildCancelError(GitlabOperationError):
+    pass
+
+
+class GitlabBuildRetryError(GitlabOperationError):
+    pass
+
+
 def raise_error_from_response(response, error, expected_code=200):
     """Tries to parse gitlab error message from response and raises error.
 
diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -691,6 +691,30 @@ class ProjectBranchManager(BaseManager):
     obj_cls = ProjectBranch
 
 
+class ProjectBuild(GitlabObject):
+    _url = '/projects/%(project_id)s/builds'
+    _constructorTypes = {'user': 'User',
+                         'commit': 'ProjectCommit'}
+    requiredUrlAttrs = ['project_id']
+    canDelete = False
+    canUpdate = False
+    canCreate = False
+
+    def cancel(self):
+        url = '/projects/%s/builds/%s/cancel' % (self.project_id, self.id)
+        r = self.gitlab._raw_post(url)
+        raise_error_from_response(r, GitlabBuildCancelError, 201)
+
+    def retry(self):
+        url = '/projects/%s/builds/%s/retry' % (self.project_id, self.id)
+        r = self.gitlab._raw_post(url)
+        raise_error_from_response(r, GitlabBuildRetryError, 201)
+
+
+class ProjectBuildManager(BaseManager):
+    obj_cls = ProjectBuild
+
+
 class ProjectCommit(GitlabObject):
     _url = '/projects/%(project_id)s/repository/commits'
     canDelete = False
@@ -716,6 +740,20 @@ def blob(self, filepath, **kwargs):
 
         return r.content
 
+    def builds(self, **kwargs):
+        url = '/projects/%s/repository/commits/%s/builds' % (self.project_id,
+                                                             self.id)
+        r = self.gitlab._raw_get(url, **kwargs)
+        raise_error_from_response(r, GitlabListError)
+
+        l = []
+        for j in r.json():
+            o = ProjectBuild(self, j)
+            o._from_api = True
+            l.append(o)
+
+        return l
+
 
 class ProjectCommitManager(BaseManager):
     obj_cls = ProjectCommit
@@ -1088,6 +1126,7 @@ class Project(GitlabObject):
     shortPrintAttr = 'path'
     managers = [
         ('branches', ProjectBranchManager, [('project_id', 'id')]),
+        ('builds', ProjectBuildManager, [('project_id', 'id')]),
         ('commits', ProjectCommitManager, [('project_id', 'id')]),
         ('commitstatuses', ProjectCommitStatusManager, [('project_id', 'id')]),
         ('events', ProjectEventManager, [('project_id', 'id')]),
