Project pipeline schedules

mlq · mlq · commit 34e32a0944b6 · 2017-11-13T15:12:36.000+01:00
diff --git a/gitlab/v3/objects.py b/gitlab/v3/objects.py
@@ -1496,6 +1496,18 @@ class ProjectFileManager(BaseManager):
     obj_cls = ProjectFile
 
 
+class ProjectPipelineSchedule(GitlabObject):
+    _url = '/projects/%(project_id)s/pipeline_schedules'
+    _create_url = '/projects/%(project_id)s/pipeline_schedules'
+
+    requiredUrlAttrs = ['project_id']
+    requiredCreateAttrs = ['description', 'ref', 'cron']
+
+
+class ProjectPipelineSchedulesManager(BaseManager):
+    obj_cls = ProjectPipelineSchedule
+
+
 class ProjectPipeline(GitlabObject):
     _url = '/projects/%(project_id)s/pipelines'
     _create_url = '/projects/%(project_id)s/pipeline'
@@ -1803,6 +1815,7 @@ class Project(GitlabObject):
         ('notificationsettings', 'ProjectNotificationSettingsManager',
          [('project_id', 'id')]),
         ('pipelines', 'ProjectPipelineManager', [('project_id', 'id')]),
+        ('pipeline_schedules', 'ProjectPipelineSchedulesManager', [('project_id', 'id')]),
         ('runners', 'ProjectRunnerManager', [('project_id', 'id')]),
         ('services', 'ProjectServiceManager', [('project_id', 'id')]),
         ('snippets', 'ProjectSnippetManager', [('project_id', 'id')]),
diff --git a/gitlab/v4/objects.py b/gitlab/v4/objects.py
@@ -1764,6 +1764,65 @@ def create(self, data, **kwargs):
         return CreateMixin.create(self, data, path=path, **kwargs)
 
 
+class ProjectPipelineScheduleVariable(SaveMixin, ObjectDeleteMixin, RESTObject):
+    _id_attr = 'key'
+
+
+class ProjectPipelineScheduleVariableManager(CRUDMixin, RESTManager):
+    _path = '/projects/%(project_id)s/pipeline_schedules/%(pipeline_schedule_id)s/variables'
+    _obj_cls = ProjectPipelineScheduleVariable
+    _from_parent_attrs = {'project_id': 'project_id',
+                          'pipeline_schedule_id' : 'id'}
+    _create_attrs = (('pipeline_schedule_id', 'key', 'value'), tuple())
+    _create_attrs = (('key', 'value'), tuple())
+
+    def list(self):
+        array = []
+        if 'variables' in self._parent._attrs:
+            for variable in self._parent._attrs['variables']:
+                schedule_variable = self._obj_cls(self, variable)
+                array.append(schedule_variable)
+        else:
+            obj = self._parent.manager.get(self._parent.id)
+            for variable in obj._attrs['variables']:
+                schedule_variable = self._obj_cls(self, variable)
+                array.append(schedule_variable)
+
+        return array
+
+
+class ProjectPipelineSchedule(RESTObject):
+    _managers = (
+        ('variables', 'ProjectPipelineScheduleVariableManager'),
+    )
+
+
+class ProjectPipelineSchedulesManager(RetrieveMixin, CreateMixin, RESTManager):
+    _path = '/projects/%(project_id)s/pipeline_schedules'
+    _obj_cls = ProjectPipelineSchedule
+    _from_parent_attrs = {'project_id': 'id'}
+    _create_attrs = (('description', 'ref', 'cron'),
+                     ('cron_timezone', 'active'))
+
+    def create(self, data, **kwargs):
+        """Creates a new object.
+
+        Args:
+            data (dict): Parameters to send to the server to create the
+                         resource
+            **kwargs: Extra options to send to the server (e.g. sudo)
+
+        Raises:
+            GitlabAuthenticationError: If authentication is not correct
+            GitlabCreateError: If the server cannot perform the request
+
+        Returns:
+            RESTObject: A new instance of the managed object class build with
+                the data sent by the server
+        """
+        return CreateMixin.create(self, data, path=self.path, **kwargs)
+
+
 class ProjectSnippetNote(SaveMixin, ObjectDeleteMixin, RESTObject):
     pass
 
@@ -2035,6 +2094,7 @@ class Project(SaveMixin, ObjectDeleteMixin, RESTObject):
         ('notificationsettings', 'ProjectNotificationSettingsManager'),
         ('pipelines', 'ProjectPipelineManager'),
         ('protectedbranches', 'ProjectProtectedBranchManager'),
+        ('pipeline_schedules', 'ProjectPipelineSchedulesManager'),
         ('runners', 'ProjectRunnerManager'),
         ('services', 'ProjectServiceManager'),
         ('snippets', 'ProjectSnippetManager'),
