def toProtocolTreeNode(self): node = super(MediaMessageProtocolEntity, self).toProtocolTreeNode() mediaNode = ProtocolTreeNode("media", {"type": self.mediaType}, None, None) node.addChild(mediaNode) if self.preview: mediaNode.setData(self.preview) return node