Fixing the response and project_id argument

Asher256 · Asher256 · commit 72d982b70b00 · 2016-11-01T20:03:50.000-04:00
diff --git a/docs/gl_objects/deploy_keys.py b/docs/gl_objects/deploy_keys.py
@@ -36,13 +36,9 @@
 # end delete
 
 # enable
-key = gl.project_keys.enable(key_id, project_id=1)
-# or
 key = project.keys.enable(key_id)
 # end enable
 
 # disable
-key = gl.project_keys.disable(key_id, project_id=1)
-# or
 key = project.keys.disable(key_id)
 # end disable
diff --git a/gitlab/objects.py b/gitlab/objects.py
@@ -1249,17 +1249,17 @@ class ProjectKey(GitlabObject):
     requiredUrlAttrs = ['project_id']
     requiredCreateAttrs = ['title', 'key']
 
-    def enable(self, key_id, project_id):
+    def enable(self, key_id):
         """Enable a deploy key for a project."""
-        url = '/projects/%s/deploy_keys/%s/enable' % (project_id, key_id)
+        url = '/projects/%s/deploy_keys/%s/enable' % (self.project_id, key_id)
         r = self.gitlab._raw_post(url)
-        raise_error_from_response(r, GitlabProjectDeployKeyError, 201)
+        raise_error_from_response(r, GitlabProjectDeployKeyError, 200)
 
-    def disable(self, key_id, project_id):
+    def disable(self, key_id):
         """Disable a deploy key for a project."""
-        url = '/projects/%s/deploy_keys/%s/disable' % (project_id, key_id)
+        url = '/projects/%s/deploy_keys/%s/disable' % (self.project_id, key_id)
         r = self.gitlab._raw_delete(url)
-        raise_error_from_response(r, GitlabProjectDeployKeyError, 201)
+        raise_error_from_response(r, GitlabProjectDeployKeyError, 200)
 
 
 class ProjectKeyManager(BaseManager):
