From e3525874fd733397cc3e24dcf1c7dbc0fcbb1dbc Mon Sep 17 00:00:00 2001 From: Hossin Asaadi Date: Thu, 10 Sep 2020 09:12:52 +0430 Subject: [PATCH] add getJid method add getJid method to recognize who's ResultGetPictureIqProtocolEntity response from. --- .../protocolentities/iq_picture_get_result.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/yowsup/layers/protocol_profiles/protocolentities/iq_picture_get_result.py b/yowsup/layers/protocol_profiles/protocolentities/iq_picture_get_result.py index 676adbb9d..7732ad68c 100644 --- a/yowsup/layers/protocol_profiles/protocolentities/iq_picture_get_result.py +++ b/yowsup/layers/protocol_profiles/protocolentities/iq_picture_get_result.py @@ -10,12 +10,15 @@ class ResultGetPictureIqProtocolEntity(PictureIqProtocolEntity): ''' def __init__(self, jid, pictureData, pictureId, preview = True, _id = None): super(ResultGetPictureIqProtocolEntity, self).__init__(jid, _id, "result") - self.setResultPictureProps(pictureData, pictureId, preview) + self.setResultPictureProps(jid, pictureData, pictureId, preview) + - def setResultPictureProps(self, pictureData, pictureId, preview = True): + + def setResultPictureProps(self,jid , pictureData, pictureId, preview = True ): self.preview = preview self.pictureData = pictureData self.pictureId = pictureId + self.jid = jid def isPreview(self): return self.preview @@ -26,6 +29,9 @@ def getPictureData(self): def getPictureId(self): return self.pictureId + def getJid(self): + return self.jid + def writeToFile(self, path): with open(path, "wb") as outFile: outFile.write(self.getPictureData()) @@ -41,5 +47,5 @@ def fromProtocolTreeNode(node): entity = PictureIqProtocolEntity.fromProtocolTreeNode(node) entity.__class__ = ResultGetPictureIqProtocolEntity pictureNode = node.getChild("picture") - entity.setResultPictureProps(pictureNode.getData(), pictureNode.getAttributeValue("id"), pictureNode.getAttributeValue("type") == "preview") - return entity \ No newline at end of file + entity.setResultPictureProps(node.getAttributeValue("from"),pictureNode.getData(), pictureNode.getAttributeValue("id"), pictureNode.getAttributeValue("type") == "preview") + return entity