jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

XEP-0231: do not crash if max-age is None

Browse Source 
it is only RECOMMENDED
This commit is contained in:
mathieui 2021-02-26 00:07:08 +01:00
parent 8eee559d39
commit 8e388df8e0
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
slixmpp/plugins/xep_0231/bob.py
View File

@ -89,7 +89,8 @@ class XEP_0231(BasePlugin):
bob['data'] = data bob['data'] = data
bob['type'] = mtype bob['type'] = mtype
bob['cid'] = cid bob['cid'] = cid
bob['max_age'] = max_age if max_age is not None:
bob['max_age'] = max_age
await self.api['set_bob'](args=bob) await self.api['set_bob'](args=bob)
# Schedule destruction of the data # Schedule destruction of the data

8
slixmpp/plugins/xep_0231/stanza.py
View File

@ -18,10 +18,14 @@ class BitsOfBinary(ElementBase):
interfaces = {'cid', 'max_age', 'type', 'data'} interfaces = {'cid', 'max_age', 'type', 'data'}
def get_max_age(self): def get_max_age(self):
return int(self._get_attr('max-age')) try:
return int(self._get_attr('max-age'))
except ValueError:
return None
def set_max_age(self, value): def set_max_age(self, value):
self._set_attr('max-age', str(value)) if value is not None:
self._set_attr('max-age', str(value))
def get_data(self): def get_data(self):
return base64.b64decode(bytes(self.xml.text)) return base64.b64decode(bytes(self.xml.text))