|  | 
| 1 |  | -from gitlab import cli | 
| 2 |  | -from gitlab import exceptions as exc | 
| 3 | 1 | from gitlab.base import RequiredOptional, RESTManager, RESTObject | 
| 4 | 2 | from gitlab.mixins import NoUpdateMixin, ObjectDeleteMixin | 
| 5 | 3 | 
 | 
|  | 
| 14 | 12 | class ProjectBranch(ObjectDeleteMixin, RESTObject): | 
| 15 | 13 |     _id_attr = "name" | 
| 16 | 14 | 
 | 
| 17 |  | -    @cli.register_custom_action( | 
| 18 |  | -        "ProjectBranch", tuple(), ("developers_can_push", "developers_can_merge") | 
| 19 |  | -    ) | 
| 20 |  | -    @exc.on_http_error(exc.GitlabProtectError) | 
| 21 |  | -    def protect(self, developers_can_push=False, developers_can_merge=False, **kwargs): | 
| 22 |  | -        """Protect the branch. | 
| 23 |  | -
 | 
| 24 |  | -        Args: | 
| 25 |  | -            developers_can_push (bool): Set to True if developers are allowed | 
| 26 |  | -                                        to push to the branch | 
| 27 |  | -            developers_can_merge (bool): Set to True if developers are allowed | 
| 28 |  | -                                         to merge to the branch | 
| 29 |  | -            **kwargs: Extra options to send to the server (e.g. sudo) | 
| 30 |  | -
 | 
| 31 |  | -        Raises: | 
| 32 |  | -            GitlabAuthenticationError: If authentication is not correct | 
| 33 |  | -            GitlabProtectError: If the branch could not be protected | 
| 34 |  | -        """ | 
| 35 |  | -        id = self.get_id().replace("/", "%2F") | 
| 36 |  | -        path = "%s/%s/protect" % (self.manager.path, id) | 
| 37 |  | -        post_data = { | 
| 38 |  | -            "developers_can_push": developers_can_push, | 
| 39 |  | -            "developers_can_merge": developers_can_merge, | 
| 40 |  | -        } | 
| 41 |  | -        self.manager.gitlab.http_put(path, post_data=post_data, **kwargs) | 
| 42 |  | -        self._attrs["protected"] = True | 
| 43 |  | - | 
| 44 |  | -    @cli.register_custom_action("ProjectBranch") | 
| 45 |  | -    @exc.on_http_error(exc.GitlabProtectError) | 
| 46 |  | -    def unprotect(self, **kwargs): | 
| 47 |  | -        """Unprotect the branch. | 
| 48 |  | -
 | 
| 49 |  | -        Args: | 
| 50 |  | -            **kwargs: Extra options to send to the server (e.g. sudo) | 
| 51 |  | -
 | 
| 52 |  | -        Raises: | 
| 53 |  | -            GitlabAuthenticationError: If authentication is not correct | 
| 54 |  | -            GitlabProtectError: If the branch could not be unprotected | 
| 55 |  | -        """ | 
| 56 |  | -        id = self.get_id().replace("/", "%2F") | 
| 57 |  | -        path = "%s/%s/unprotect" % (self.manager.path, id) | 
| 58 |  | -        self.manager.gitlab.http_put(path, **kwargs) | 
| 59 |  | -        self._attrs["protected"] = False | 
| 60 |  | - | 
| 61 | 15 | 
 | 
| 62 | 16 | class ProjectBranchManager(NoUpdateMixin, RESTManager): | 
| 63 | 17 |     _path = "/projects/%(project_id)s/repository/branches" | 
|  | 
0 commit comments