Merge branch 'master-project-deploy-keys' of https://github.com/Asher256/python-gitlab into Asher256-master-project-deploy-keys

Gauvain Pocentek · Gauvain Pocentek · commit 0c1817f8be11 · 2016-11-05T07:54:09.000+01:00
diff --git a/docs/gl_objects/deploy_keys.py b/docs/gl_objects/deploy_keys.py
@@ -34,3 +34,11 @@
 # or
 key.delete()
 # end delete
+
+# enable
+key = project.keys.enable(key_id)
+# end enable
+
+# disable
+key = project.keys.disable(key_id)
+# end disable
diff --git a/docs/gl_objects/deploy_keys.rst b/docs/gl_objects/deploy_keys.rst
@@ -56,3 +56,15 @@ Delete a deploy key for a project:
 .. literalinclude:: deploy_keys.py
    :start-after: # delete
    :end-before: # end delete
+
+Enable a deploy key for a project:
+
+.. literalinclude:: deploy_keys.py
+   :start-after: # enable
+   :end-before: # end enable
+
+Disable a deploy key for a project:
+
+.. literalinclude:: deploy_keys.py
+   :start-after: # disable
+   :end-before: # end disable
diff --git a/gitlab/exceptions.py b/gitlab/exceptions.py
@@ -75,6 +75,10 @@ class GitlabTransferProjectError(GitlabOperationError):
     pass
 
 
+class GitlabProjectDeployKeyError(GitlabOperationError):
+    pass
+
+
 class GitlabCancelError(GitlabOperationError):
     pass
 
diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -1253,6 +1253,18 @@ class ProjectKey(GitlabObject):
 class ProjectKeyManager(BaseManager):
     obj_cls = ProjectKey
 
+    def enable(self, key_id):
+        """Enable a deploy key for a project."""
+        url = '/projects/%s/deploy_keys/%s/enable' % (self.parent.id, key_id)
+        r = self.gitlab._raw_post(url)
+        raise_error_from_response(r, GitlabProjectDeployKeyError, 201)
+
+    def disable(self, key_id):
+        """Disable a deploy key for a project."""
+        url = '/projects/%s/deploy_keys/%s/disable' % (self.parent.id, key_id)
+        r = self.gitlab._raw_delete(url)
+        raise_error_from_response(r, GitlabProjectDeployKeyError, 201)
+
 
 class ProjectEvent(GitlabObject):
     _url = '/projects/%(project_id)s/events'
