Add logic for Announcments

Author: ErnestoLoma

Diff for: mypy.ini

@@ -0,0 +1 @@
+[mypy]

Diff for: pyatlan/generator/templates/entity.jinja2

@@ -2,9 +2,9 @@ from __future__ import annotations
 from typing import Optional, Dict, Any, List, Literal
 from pydantic import Field
 from datetime import date
-from pyatlan.model.core import AtlanObject, Classification
+from pyatlan.model.core import AtlanObject, Classification, Announcement
 from pyatlan.model.enums import CertificateStatus, EntityStatus, google_datastudio_asset_type, powerbi_endorsement, \
-    icon_type
+    icon_type, AnnouncementType
 
 class Internal(AtlanObject):
     """For internal usage"""
@@ -35,7 +35,7 @@ class {{ entity_def.name }}({{super_classes[0]}}):
         {%- set type = attribute_def.typeName | get_type %}
         {{attribute_def.name | to_snake_case }}: {% if attribute_def.isOptional %}Optional[{% endif %}{{type}}{% if attribute_def.isOptional %}]{% endif %} = Field(None, description='',  alias='{{attribute_def.name}}') # relationship
         {%- endfor %}
-    attributes: Optional['{{entity_def.name}}.Attributes'] = Field(
+    attributes: '{{entity_def.name}}.Attributes' = Field(
         None,
         description='Map of attributes in the instance and their values. The specific keys of this map will vary '
                     'by type, so are described in the sub-types of this schema.\n',
@@ -135,11 +135,41 @@ class {{ entity_def.name }}({{super_classes[0]}}):
         {%- set type = attribute_def.typeName | get_type %}
         {{attribute_def.name | to_snake_case }}: {% if attribute_def.isOptional %}Optional[{% endif %}{{type}}{% if attribute_def.isOptional %}]{% endif %} = Field(None, description='',  alias='{{attribute_def.name}}') # relationship
         {%- endfor %}
-    attributes: Optional['{{entity_def.name}}.Attributes'] = Field(
+    attributes: '{{entity_def.name}}.Attributes' = Field(
         None,
         description='Map of attributes in the instance and their values. The specific keys of this map will vary by '
                     'type, so are described in the sub-types of this schema.\n',
     )
+    {% if entity_def.name == "Asset" %}
+    def has_announcement(self) -> bool:
+        if self.attributes and (
+            self.attributes.announcement_title or self.attributes.announcement_type
+        ):
+            return True
+        else:
+            return False
+
+    def set_announcement(self, announcement: Announcement) -> None:
+        self.attributes.announcement_type = announcement.announcement_type.value
+        self.attributes.announcement_title = announcement.announcement_title
+        self.attributes.announcement_message = announcement.announcement_message
+
+    def get_announcment(self) -> Optional[Announcement]:
+        if self.attributes.announcement_type and self.attributes.announcement_title:
+            return Announcement(
+                announcement_type=AnnouncementType[
+                    self.attributes.announcement_type.upper()
+                ],
+                announcement_title=self.attributes.announcement_title,
+                announcement_message=self.attributes.announcement_message,
+            )
+        return None
+
+    def clear_announcment(self):
+        self.attributes.announcement_message = None
+        self.attributes.announcement_title = None
+        self.attributes.announcement_type = None
+    {% endif %}
     {% endif %}
 {%- endif %}
 {% endfor %}