week06

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

@@ -0,0 +1,40 @@
+from typing import Any, cast, Union
+
+from gitlab.base import RESTManager, RESTObject
+from gitlab.mixins import CRUDMixin, ObjectDeleteMixin, SaveMixin
+from gitlab.types import ArrayAttribute, RequiredOptional
+
+__all__ = [
+    "BroadcastMessage",
+    "BroadcastMessageManager",
+]
+
+
+class BroadcastMessage(SaveMixin, ObjectDeleteMixin, RESTObject):
+    pass
+
+
+class BroadcastMessageManager(CRUDMixin, RESTManager):
+    _path = "/broadcast_messages"
+    _obj_cls = BroadcastMessage
+
+    _create_attrs = RequiredOptional(
+        required=("message",),
+        optional=("starts_at", "ends_at", "color", "font", "target_access_levels"),
+    )
+    _update_attrs = RequiredOptional(
+        optional=(
+            "message",
+            "starts_at",
+            "ends_at",
+            "color",
+            "font",
+            "target_access_levels",
+        )
+    )
+    _types = {"target_access_levels": ArrayAttribute}
+
+    def get(
+        self, id: Union[str, int], lazy: bool = False, **kwargs: Any
+    ) -> BroadcastMessage:
+        return cast(BroadcastMessage, super().get(id=id, lazy=lazy, **kwargs))