New exception for ProjectKey.enable_deploy_key and disable_deploy_key

Asher256@users.noreply.github.com · Asher256@users.noreply.github.com · commit 61d4cac5c711 · 2016-10-30T22:18:42.000-04:00
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,13 +1253,13 @@ def enable_deploy_key(self, project_id, key_id):
         """Enable a deploy key for a project."""
         url = '/projects/%s/deploy_keys/%s/enable' % (project_id, key_id)
         r = self.gitlab._raw_post(url)
-        raise_error_from_response(r, GitlabBuildRetryError, 201)
+        raise_error_from_response(r, GitlabProjectDeployKeyError, 201)
 
     def disable_deploy_key(self, project_id, key_id):
         """Disable a deploy key for a project."""
         url = '/projects/%s/deploy_keys/%s/disable' % (project_id, key_id)
         r = self.gitlab._raw_delete(url)
-        raise_error_from_response(r, GitlabBuildRetryError, 201)
+        raise_error_from_response(r, GitlabProjectDeployKeyError, 201)
 
 
 class ProjectKeyManager(BaseManager):
