64 lines
1.8 KiB
Python
64 lines
1.8 KiB
Python
from typing import Any, cast, Union
|
|
|
|
from gitlab.base import RESTManager, RESTObject
|
|
from gitlab.mixins import (
|
|
CRUDMixin,
|
|
NoUpdateMixin,
|
|
ObjectDeleteMixin,
|
|
SaveMixin,
|
|
UpdateMethod,
|
|
)
|
|
from gitlab.types import RequiredOptional
|
|
|
|
__all__ = [
|
|
"ProjectBranch",
|
|
"ProjectBranchManager",
|
|
"ProjectProtectedBranch",
|
|
"ProjectProtectedBranchManager",
|
|
]
|
|
|
|
|
|
class ProjectBranch(ObjectDeleteMixin, RESTObject):
|
|
_id_attr = "name"
|
|
|
|
|
|
class ProjectBranchManager(NoUpdateMixin, RESTManager):
|
|
_path = "/projects/{project_id}/repository/branches"
|
|
_obj_cls = ProjectBranch
|
|
_from_parent_attrs = {"project_id": "id"}
|
|
_create_attrs = RequiredOptional(required=("branch", "ref"))
|
|
|
|
def get(
|
|
self, id: Union[str, int], lazy: bool = False, **kwargs: Any
|
|
) -> ProjectBranch:
|
|
return cast(ProjectBranch, super().get(id=id, lazy=lazy, **kwargs))
|
|
|
|
|
|
class ProjectProtectedBranch(SaveMixin, ObjectDeleteMixin, RESTObject):
|
|
_id_attr = "name"
|
|
|
|
|
|
class ProjectProtectedBranchManager(CRUDMixin, RESTManager):
|
|
_path = "/projects/{project_id}/protected_branches"
|
|
_obj_cls = ProjectProtectedBranch
|
|
_from_parent_attrs = {"project_id": "id"}
|
|
_create_attrs = RequiredOptional(
|
|
required=("name",),
|
|
optional=(
|
|
"push_access_level",
|
|
"merge_access_level",
|
|
"unprotect_access_level",
|
|
"allow_force_push",
|
|
"allowed_to_push",
|
|
"allowed_to_merge",
|
|
"allowed_to_unprotect",
|
|
"code_owner_approval_required",
|
|
),
|
|
)
|
|
_update_method = UpdateMethod.PATCH
|
|
|
|
def get(
|
|
self, id: Union[str, int], lazy: bool = False, **kwargs: Any
|
|
) -> ProjectProtectedBranch:
|
|
return cast(ProjectProtectedBranch, super().get(id=id, lazy=lazy, **kwargs))
|