1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
|
from __future__ import annotations
from typing import Any
import requests
from gitlab import cli
from gitlab import exceptions as exc
from gitlab.base import RESTObject
from gitlab.mixins import (
CreateMixin,
CRUDMixin,
ListMixin,
ObjectDeleteMixin,
SaveMixin,
)
from gitlab.types import RequiredOptional
__all__ = ["DeployKey", "DeployKeyManager", "ProjectKey", "ProjectKeyManager"]
class DeployKey(RESTObject):
pass
class DeployKeyManager(CreateMixin[DeployKey], ListMixin[DeployKey]):
_path = "/deploy_keys"
_obj_cls = DeployKey
_create_attrs = RequiredOptional(
required=("title", "key"), optional=("expires_at",)
)
class ProjectKey(SaveMixin, ObjectDeleteMixin, RESTObject):
pass
class ProjectKeyManager(CRUDMixin[ProjectKey]):
_path = "/projects/{project_id}/deploy_keys"
_obj_cls = ProjectKey
_from_parent_attrs = {"project_id": "id"}
_create_attrs = RequiredOptional(
required=("title", "key"), optional=("can_push", "expires_at")
)
_update_attrs = RequiredOptional(optional=("title", "can_push", "expires_at"))
@cli.register_custom_action(
cls_names="ProjectKeyManager",
required=("key_id",),
requires_id=False,
help="Enable a deploy key for the project",
)
@exc.on_http_error(exc.GitlabProjectDeployKeyError)
def enable(self, key_id: int, **kwargs: Any) -> dict[str, Any] | requests.Response:
"""Enable a deploy key for a project.
Args:
key_id: The ID of the key to enable
**kwargs: Extra options to send to the server (e.g. sudo)
Raises:
GitlabAuthenticationError: If authentication is not correct
GitlabProjectDeployKeyError: If the key could not be enabled
Returns:
A dict of the result.
"""
path = f"{self.path}/{key_id}/enable"
return self.gitlab.http_post(path, **kwargs)
|