summaryrefslogtreecommitdiff
path: root/sleekxmpp/plugins/xep_0297/stanza.py
diff options
context:
space:
mode:
authorLance Stout <lancestout@gmail.com>2012-09-25 12:25:45 -0700
committerLance Stout <lancestout@gmail.com>2012-09-25 12:25:45 -0700
commite449dce65c0d267c4d3206b6f9ac2e89807192bc (patch)
tree1e436290a05883e9ff15e50580657a3baef0880d /sleekxmpp/plugins/xep_0297/stanza.py
parent671f680bb39f366ad13bf937c7b611f667343314 (diff)
downloadslixmpp-e449dce65c0d267c4d3206b6f9ac2e89807192bc.tar.gz
slixmpp-e449dce65c0d267c4d3206b6f9ac2e89807192bc.tar.bz2
slixmpp-e449dce65c0d267c4d3206b6f9ac2e89807192bc.tar.xz
slixmpp-e449dce65c0d267c4d3206b6f9ac2e89807192bc.zip
Fix handling forwarded stanzas to do proper lookups and deletions.
Diffstat (limited to 'sleekxmpp/plugins/xep_0297/stanza.py')
-rw-r--r--sleekxmpp/plugins/xep_0297/stanza.py20
1 files changed, 11 insertions, 9 deletions
diff --git a/sleekxmpp/plugins/xep_0297/stanza.py b/sleekxmpp/plugins/xep_0297/stanza.py
index 1cf02f74..8b97accc 100644
--- a/sleekxmpp/plugins/xep_0297/stanza.py
+++ b/sleekxmpp/plugins/xep_0297/stanza.py
@@ -6,6 +6,7 @@
See the file LICENSE for copying permission.
"""
+from sleekxmpp.stanza import Message, Presence, Iq
from sleekxmpp.xmlstream import ElementBase
@@ -16,12 +17,9 @@ class Forwarded(ElementBase):
interfaces = set(['stanza'])
def get_stanza(self):
- if self.xml.find('{jabber:client}message') is not None:
- return self['message']
- elif self.xml.find('{jabber:client}presence') is not None:
- return self['presence']
- elif self.xml.find('{jabber:client}iq') is not None:
- return self['iq']
+ for stanza in self:
+ if isinstance(stanza, (Message, Presence, Iq)):
+ return stanza
return ''
def set_stanza(self, value):
@@ -29,6 +27,10 @@ class Forwarded(ElementBase):
self.append(value)
def del_stanza(self):
- del self['message']
- del self['presence']
- del self['iq']
+ found_stanzas = []
+ for stanza in self:
+ if isinstance(stanza, (Message, Presence, Iq)):
+ found_stanzas.append(stanza)
+ for stanza in found_stanzas:
+ self.iterables.remove(stanza)
+ self.xml.remove(stanza.xml)