week06

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

@@ -0,0 +1,44 @@
+from typing import Any, cast, Union
+
+from gitlab.base import RESTManager, RESTObject
+from gitlab.mixins import BadgeRenderMixin, CRUDMixin, ObjectDeleteMixin, SaveMixin
+from gitlab.types import RequiredOptional
+
+__all__ = [
+    "GroupBadge",
+    "GroupBadgeManager",
+    "ProjectBadge",
+    "ProjectBadgeManager",
+]
+
+
+class GroupBadge(SaveMixin, ObjectDeleteMixin, RESTObject):
+    pass
+
+
+class GroupBadgeManager(BadgeRenderMixin, CRUDMixin, RESTManager):
+    _path = "/groups/{group_id}/badges"
+    _obj_cls = GroupBadge
+    _from_parent_attrs = {"group_id": "id"}
+    _create_attrs = RequiredOptional(required=("link_url", "image_url"))
+    _update_attrs = RequiredOptional(optional=("link_url", "image_url"))
+
+    def get(self, id: Union[str, int], lazy: bool = False, **kwargs: Any) -> GroupBadge:
+        return cast(GroupBadge, super().get(id=id, lazy=lazy, **kwargs))
+
+
+class ProjectBadge(SaveMixin, ObjectDeleteMixin, RESTObject):
+    pass
+
+
+class ProjectBadgeManager(BadgeRenderMixin, CRUDMixin, RESTManager):
+    _path = "/projects/{project_id}/badges"
+    _obj_cls = ProjectBadge
+    _from_parent_attrs = {"project_id": "id"}
+    _create_attrs = RequiredOptional(required=("link_url", "image_url"))
+    _update_attrs = RequiredOptional(optional=("link_url", "image_url"))
+
+    def get(
+        self, id: Union[str, int], lazy: bool = False, **kwargs: Any
+    ) -> ProjectBadge:
+        return cast(ProjectBadge, super().get(id=id, lazy=lazy, **kwargs))