Add support for protected branches

This feature appeared in gitlab 9.5.

Fixes #299

1 files changed, 12 insertions, 0 deletions

diff --git a/gitlab/v4/objects.py b/gitlab/v4/objects.py
index 92c4543..bf79deb 100644
--- a/gitlab/v4/objects.py
+++ b/gitlab/v4/objects.py
@@ -1732,6 +1732,17 @@ class ProjectDeploymentManager(RetrieveMixin, RESTManager):
     _from_parent_attrs = {'project_id': 'id'}
 
 
+class ProjectProtectedBranch(ObjectDeleteMixin, RESTObject):
+    _id_attr = 'name'
+
+
+class ProjectProtectedBranchManager(NoUpdateMixin, RESTManager):
+    _path = '/projects/%(project_id)s/protected_branches'
+    _obj_cls = ProjectProtectedBranch
+    _from_parent_attrs = {'project_id': 'id'}
+    _create_attrs = (('name', ), ('push_access_level', 'merge_access_level'))
+
+
 class ProjectRunner(ObjectDeleteMixin, RESTObject):
     pass
 
@@ -1767,6 +1778,7 @@ class Project(SaveMixin, ObjectDeleteMixin, RESTObject):
         ('notes', 'ProjectNoteManager'),
         ('notificationsettings', 'ProjectNotificationSettingsManager'),
         ('pipelines', 'ProjectPipelineManager'),
+        ('protectedbranches', 'ProjectProtectedBranchManager'),
         ('runners', 'ProjectRunnerManager'),
         ('services', 'ProjectServiceManager'),
         ('snippets', 'ProjectSnippetManager'),