summaryrefslogtreecommitdiff
path: root/gitlab/v4/objects/bulk_imports.py
blob: e8ef74f22a3a843f57c4826f670f08d2698c592a (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
46
47
48
49
50
51
52
53
54
from typing import Any, cast, Union

from gitlab.base import RESTManager, RESTObject
from gitlab.mixins import CreateMixin, ListMixin, RefreshMixin, RetrieveMixin
from gitlab.types import RequiredOptional

__all__ = [
    "BulkImport",
    "BulkImportManager",
    "BulkImportAllEntity",
    "BulkImportAllEntityManager",
    "BulkImportEntity",
    "BulkImportEntityManager",
]


class BulkImport(RefreshMixin, RESTObject):
    entities: "BulkImportEntityManager"


class BulkImportManager(CreateMixin, RetrieveMixin, RESTManager):
    _path = "/bulk_imports"
    _obj_cls = BulkImport
    _create_attrs = RequiredOptional(required=("configuration", "entities"))
    _list_filters = ("sort", "status")

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


class BulkImportEntity(RefreshMixin, RESTObject):
    pass


class BulkImportEntityManager(RetrieveMixin, RESTManager):
    _path = "/bulk_imports/{bulk_import_id}/entities"
    _obj_cls = BulkImportEntity
    _from_parent_attrs = {"bulk_import_id": "id"}
    _list_filters = ("sort", "status")

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


class BulkImportAllEntity(RESTObject):
    pass


class BulkImportAllEntityManager(ListMixin, RESTManager):
    _path = "/bulk_imports/entities"
    _obj_cls = BulkImportAllEntity
    _list_filters = ("sort", "status")