Implement ProjectBuild.keep_artifacts

Gauvain Pocentek · Gauvain Pocentek · commit e0cf1c276d16 · 2016-07-17T13:11:48.000+02:00
diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -858,7 +858,8 @@ class ProjectBranchManager(BaseManager):
 class ProjectBuild(GitlabObject):
     _url = '/projects/%(project_id)s/builds'
     _constructorTypes = {'user': 'User',
-                         'commit': 'ProjectCommit'}
+                         'commit': 'ProjectCommit',
+                         'runner': 'Runner'}
     requiredUrlAttrs = ['project_id']
     canDelete = False
     canUpdate = False
@@ -876,6 +877,18 @@ def retry(self, **kwargs):
         r = self.gitlab._raw_post(url)
         raise_error_from_response(r, GitlabBuildRetryError, 201)
 
+    def keep_artifacts(self, **kwargs):
+        """Prevent artifacts from being delete when expiration is set.
+
+        Raises:
+            GitlabConnectionError: If the server cannot be reached.
+            GitlabCreateError: If the request failed.
+        """
+        url = ('/projects/%s/builds/%s/artifacts/keep' %
+               (self.project_id, self.id))
+        r = self.gitlab._raw_post(url)
+        raise_error_from_response(r, GitlabGetError, 200)
+
     def artifacts(self, **kwargs):
         """Get the build artifacts.
 
