summaryrefslogtreecommitdiff
path: root/gitlab/v4/objects/resource_groups.py
blob: 1ca34f6626b2d7f08791c6acb870f6c876bf9ff2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
from typing import Any, cast, Union

from gitlab.base import RESTManager, RESTObject
from gitlab.mixins import ListMixin, RetrieveMixin, SaveMixin, UpdateMixin
from gitlab.types import RequiredOptional

__all__ = [
    "ProjectResourceGroup",
    "ProjectResourceGroupManager",
    "ProjectResourceGroupUpcomingJob",
    "ProjectResourceGroupUpcomingJobManager",
]


class ProjectResourceGroup(SaveMixin, RESTObject):
    _id_attr = "key"

    upcoming_jobs: "ProjectResourceGroupUpcomingJobManager"


class ProjectResourceGroupManager(RetrieveMixin, UpdateMixin, RESTManager):
    _path = "/projects/{project_id}/resource_groups"
    _obj_cls = ProjectResourceGroup
    _from_parent_attrs = {"project_id": "id"}
    _list_filters = (
        "order_by",
        "sort",
        "include_html_description",
    )
    _update_attrs = RequiredOptional(optional=("process_mode",))

    def get(
        self, id: Union[str, int], lazy: bool = False, **kwargs: Any
    ) -> ProjectResourceGroup:
        return cast(ProjectResourceGroup, super().get(id=id, lazy=lazy, **kwargs))


class ProjectResourceGroupUpcomingJob(RESTObject):
    pass


class ProjectResourceGroupUpcomingJobManager(ListMixin, RESTManager):
    _path = "/projects/{project_id}/resource_groups/{resource_group_key}/upcoming_jobs"
    _obj_cls = ProjectResourceGroupUpcomingJob
    _from_parent_attrs = {"project_id": "project_id", "resource_group_key": "key"}