Adding AMP body support to injectionAPI

Author: ryan-lydz

socketlabs/injectionapi/core/injectionrequestfactory.py

@@ -142,6 +142,7 @@ def generate_base_message(message: MessageBase):
     message_json.subject = message.subject
     message_json.plain_text_body = message.plain_text_body
     message_json.html_body = message.html_body
+    message_json.amp_body = message.amp_body
     message_json.mailing_id = message.mailing_id
     message_json.message_id = message.message_id
     message_json.charset = message.charset

socketlabs/injectionapi/core/serialization/messagejson.py

@@ -14,6 +14,7 @@ def __init__(self):
         self._subject = None
         self._plain_text_body = None
         self._html_body = None
+        self._amp_body = None
         self._api_template = None
         self._mailing_id = None
         self._message_id = None
@@ -147,6 +148,24 @@ def html_body(self, val: str):
         """
         self._html_body = val
 
+    @property
+    def amp_body(self):
+        """
+        Get the AMP portion of the message body.
+        :return the AMP body
+        :rtype str
+        """
+        return self._amp_body
+
+    @amp_body.setter
+    def amp_body(self, val: str):
+        """
+        Set the AMP portion of the message body.
+        :param val: the AMP body
+        :type val: str
+        """
+        self._amp_body = val
+        
     @property
     def api_template(self):
         """
@@ -353,6 +372,9 @@ def to_json(self):
         if self.html_body is not None and self.html_body.strip() != '':
             json["htmlBody"] = self.html_body
 
+        if self.amp_body is not None and self._amp_body.strip() != '':
+            json["ampBody"] = self.amp_body
+
         if self.plain_text_body is not None and self.plain_text_body.strip() != '':
             json["textBody"] = self.plain_text_body
 

socketlabs/injectionapi/message/basicmessage.py

@@ -35,6 +35,7 @@ def __init__(self):
         self._subject = None
         self._plain_text_body = None
         self._html_body = None
+        self._amp_body = None
         self._api_template = None
         self._mailing_id = None
         self._message_id = None
@@ -213,6 +214,24 @@ def html_body(self, val: str):
         """
         self._html_body = val
 
+    @property
+    def amp_body(self):
+        """
+        Get the AMP portion of the message body.
+        :return the AMP body
+        :rtype str
+        """
+        return self._amp_body
+
+    @amp_body.setter
+    def amp_body(self, val: str):
+        """
+        Set the AMP portion of the message body.
+        :param val: the AMP body
+        :type val: str
+        """
+        self._amp_body = val
+
     @property
     def api_template(self):
         """

socketlabs/injectionapi/message/bulkmessage.py

@@ -39,6 +39,7 @@ def __init__(self):
         self._subject = None
         self._plain_text_body = None
         self._html_body = None
+        self._amp_body = None
         self._api_template = None
         self._mailing_id = None
         self._message_id = None
@@ -174,6 +175,24 @@ def html_body(self, val: str):
         """
         self._html_body = val
 
+    @property
+    def amp_body(self):
+        """
+        Get the AMP portion of the message body.
+        :return the AMP body
+        :rtype str
+        """
+        return self._amp_body
+
+    @amp_body.setter
+    def amp_body(self, val: str):
+        """
+        Set the AMP portion of the message body.
+        :param val: the AMP body
+        :type val: str
+        """
+        self._amp_body = val
+
     @property
     def api_template(self):
         """

socketlabs/injectionapi/message/messagebase.py

@@ -70,6 +70,26 @@ def html_body(self, val: str):
         """
         pass
 
+    @property
+    @abstractmethod
+    def amp_body(self):
+        """
+        Get the AMP portion of the message body.
+        :return the AMP body
+        :rtype str
+        """
+        pass
+
+    @amp_body.setter
+    @abstractmethod
+    def amp_body(self, val: str):
+        """
+        Set the AMP portion of the message body.
+        :param val: the AMP body
+        :type val: str
+        """
+        pass
+
     @property
     @abstractmethod
     def api_template(self):