week06

env/lib/python3.12/site-packages/gitlab/v4/objects/custom_attributes.py

@@ -0,0 +1,58 @@
+from typing import Any, cast, Union
+
+from gitlab.base import RESTManager, RESTObject
+from gitlab.mixins import DeleteMixin, ObjectDeleteMixin, RetrieveMixin, SetMixin
+
+__all__ = [
+    "GroupCustomAttribute",
+    "GroupCustomAttributeManager",
+    "ProjectCustomAttribute",
+    "ProjectCustomAttributeManager",
+    "UserCustomAttribute",
+    "UserCustomAttributeManager",
+]
+
+
+class GroupCustomAttribute(ObjectDeleteMixin, RESTObject):
+    _id_attr = "key"
+
+
+class GroupCustomAttributeManager(RetrieveMixin, SetMixin, DeleteMixin, RESTManager):
+    _path = "/groups/{group_id}/custom_attributes"
+    _obj_cls = GroupCustomAttribute
+    _from_parent_attrs = {"group_id": "id"}
+
+    def get(
+        self, id: Union[str, int], lazy: bool = False, **kwargs: Any
+    ) -> GroupCustomAttribute:
+        return cast(GroupCustomAttribute, super().get(id=id, lazy=lazy, **kwargs))
+
+
+class ProjectCustomAttribute(ObjectDeleteMixin, RESTObject):
+    _id_attr = "key"
+
+
+class ProjectCustomAttributeManager(RetrieveMixin, SetMixin, DeleteMixin, RESTManager):
+    _path = "/projects/{project_id}/custom_attributes"
+    _obj_cls = ProjectCustomAttribute
+    _from_parent_attrs = {"project_id": "id"}
+
+    def get(
+        self, id: Union[str, int], lazy: bool = False, **kwargs: Any
+    ) -> ProjectCustomAttribute:
+        return cast(ProjectCustomAttribute, super().get(id=id, lazy=lazy, **kwargs))
+
+
+class UserCustomAttribute(ObjectDeleteMixin, RESTObject):
+    _id_attr = "key"
+
+
+class UserCustomAttributeManager(RetrieveMixin, SetMixin, DeleteMixin, RESTManager):
+    _path = "/users/{user_id}/custom_attributes"
+    _obj_cls = UserCustomAttribute
+    _from_parent_attrs = {"user_id": "id"}
+
+    def get(
+        self, id: Union[str, int], lazy: bool = False, **kwargs: Any
+    ) -> UserCustomAttribute:
+        return cast(UserCustomAttribute, super().get(id=id, lazy=lazy, **kwargs))