class BaseMessage(object): msg_type = None def render(self): """Use to build message""" raise NotImplementedError("Implement this method before using.") class TextMessage(BaseMessage): msg_type = "text" def __init__(self, text): self.text = text def render(self,email) -> dict: return { "email" : email, "msg_type": self.msg_type, "content": {"text": self.text}, }