MR: add support for cancel_merge_when_build_succeeds

Gauvain Pocentek · Gauvain Pocentek · commit f6a51d67c38c · 2016-03-22T18:32:31.000+01:00
diff --git a/gitlab/cli.py b/gitlab/cli.py
@@ -42,6 +42,7 @@
                                                  'filepath']},
                            'builds': {'required': ['id', 'project-id']}},
     gitlab.ProjectMergeRequest: {
+        'cancel': {'required': ['id', 'project-id']},
         'merge': {'required': ['id', 'project-id'],
                   'optional': ['merge-commit-message',
                                'should-remove-source-branch',
@@ -223,6 +224,13 @@ def do_project_build_retry(self, cls, gl, what, args):
         except Exception as e:
             _die("Impossible to retry project build (%s)" % str(e))
 
+    def do_project_merge_request_cancel(self, cls, gl, what, args):
+        try:
+            o = self.do_get(cls, gl, what, args)
+            return o.cancel_merge_when_build_succeeds()
+        except Exception as e:
+            _die("Impossible to cancel merge request (%s)" % str(e))
+
     def do_project_merge_request_merge(self, cls, gl, what, args):
         try:
             o = self.do_get(cls, gl, what, args)
diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -1085,6 +1085,18 @@ def Note(self, id=None, **kwargs):
             self.gitlab, id, project_id=self.project_id,
             merge_request_id=self.id, **kwargs)
 
+    def cancel_merge_when_build_succeeds(self, **kwargs):
+        """Cancel merge when build succeeds."""
+
+        u = ('/projects/%s/merge_requests/%s/cancel_merge_when_build_succeeds'
+             % (self.project_id, self.id))
+        r = self.gitlab._raw_put(u, **kwargs)
+        errors = {401: GitlabMRForbiddenError,
+                  405: GitlabMRClosedError,
+                  406: GitlabMROnBuildSuccessError}
+        raise_error_from_response(r, errors)
+        return ProjectMergeRequest(self, r.json())
+
     def merge(self, merge_commit_message=None,
               should_remove_source_branch=False,
               merged_when_build_succeeds=False,
