python-gitlab/gitlab/v4/objects/invitations.py at feat/graphql · python-gitlab/python-gitlab · GitHub

Repository navigation

History

91 lines (75 loc) · 2.67 KB

Raw

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"}