92 lines
2.7 KiB
Python
92 lines
2.7 KiB
Python
from typing import Any, cast, Union
|
|
|
|
from gitlab.base import RESTManager, RESTObject
|
|
from gitlab.exceptions import GitlabInvitationError
|
|
from gitlab.mixins import CRUDMixin, ObjectDeleteMixin, SaveMixin
|
|
from gitlab.types import ArrayAttribute, CommaSeparatedListAttribute, RequiredOptional
|
|
|
|
__all__ = [
|
|
"ProjectInvitation",
|
|
"ProjectInvitationManager",
|
|
"GroupInvitation",
|
|
"GroupInvitationManager",
|
|
]
|
|
|
|
|
|
class InvitationMixin(CRUDMixin):
|
|
def create(self, *args: Any, **kwargs: Any) -> RESTObject:
|
|
invitation = super().create(*args, **kwargs)
|
|
|
|
if invitation.status == "error":
|
|
raise GitlabInvitationError(invitation.message)
|
|
|
|
return invitation
|
|
|
|
|
|
class ProjectInvitation(SaveMixin, ObjectDeleteMixin, RESTObject):
|
|
_id_attr = "email"
|
|
|
|
|
|
class ProjectInvitationManager(InvitationMixin, RESTManager):
|
|
_path = "/projects/{project_id}/invitations"
|
|
_obj_cls = ProjectInvitation
|
|
_from_parent_attrs = {"project_id": "id"}
|
|
_create_attrs = RequiredOptional(
|
|
required=("access_level",),
|
|
optional=(
|
|
"expires_at",
|
|
"invite_source",
|
|
"tasks_to_be_done",
|
|
"tasks_project_id",
|
|
),
|
|
exclusive=("email", "user_id"),
|
|
)
|
|
_update_attrs = RequiredOptional(
|
|
optional=("access_level", "expires_at"),
|
|
)
|
|
_list_filters = ("query",)
|
|
_types = {
|
|
"email": CommaSeparatedListAttribute,
|
|
"user_id": CommaSeparatedListAttribute,
|
|
"tasks_to_be_done": ArrayAttribute,
|
|
}
|
|
|
|
def get(
|
|
self, id: Union[str, int], lazy: bool = False, **kwargs: Any
|
|
) -> ProjectInvitation:
|
|
return cast(ProjectInvitation, super().get(id=id, lazy=lazy, **kwargs))
|
|
|
|
|
|
class GroupInvitation(SaveMixin, ObjectDeleteMixin, RESTObject):
|
|
_id_attr = "email"
|
|
|
|
|
|
class GroupInvitationManager(InvitationMixin, RESTManager):
|
|
_path = "/groups/{group_id}/invitations"
|
|
_obj_cls = GroupInvitation
|
|
_from_parent_attrs = {"group_id": "id"}
|
|
_create_attrs = RequiredOptional(
|
|
required=("access_level",),
|
|
optional=(
|
|
"expires_at",
|
|
"invite_source",
|
|
"tasks_to_be_done",
|
|
"tasks_project_id",
|
|
),
|
|
exclusive=("email", "user_id"),
|
|
)
|
|
_update_attrs = RequiredOptional(
|
|
optional=("access_level", "expires_at"),
|
|
)
|
|
_list_filters = ("query",)
|
|
_types = {
|
|
"email": CommaSeparatedListAttribute,
|
|
"user_id": CommaSeparatedListAttribute,
|
|
"tasks_to_be_done": ArrayAttribute,
|
|
}
|
|
|
|
def get(
|
|
self, id: Union[str, int], lazy: bool = False, **kwargs: Any
|
|
) -> GroupInvitation:
|
|
return cast(GroupInvitation, super().get(id=id, lazy=lazy, **kwargs))
|