pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add max-file-size support to HTTP File Upload example.

Browse source
This commit is contained in:
Emmanuel Gil Peyrot 2018-03-08 12:11:26 +01:00
parent 94ea8151d4
commit 29faf114a7
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
examples/http_upload.py
View file

@ -53,7 +53,17 @@ class HttpUpload(slixmpp.ClientXMPP):
self.disconnect() self.disconnect()
return return
# TODO: check the maximum size from info_iq here. for form in info_iq['disco_info'].iterables:
values = form['values']
if values['FORM_TYPE'] == ['urn:xmpp:http:upload:0']:
max_file_size = int(values['max-file-size'])
if self.size > max_file_size:
log.error('File size bigger than max allowed')
self.disconnect()
return
break
else:
log.warn('Impossible to find max-file-size, assuming infinite storage space')
log.info('Using service %s', info_iq['from']) log.info('Using service %s', info_iq['from'])
slot_iq = yield from self['xep_0363'].request_slot( slot_iq = yield from self['xep_0363'].request_slot(
@ -127,6 +137,7 @@ if __name__ == '__main__':
xmpp = HttpUpload(args.jid, args.password, args.recipient, args.file) xmpp = HttpUpload(args.jid, args.password, args.recipient, args.file)
xmpp.register_plugin('xep_0071') xmpp.register_plugin('xep_0071')
xmpp.register_plugin('xep_0128')
xmpp.register_plugin('xep_0363') xmpp.register_plugin('xep_0363')
# Connect to the XMPP server and start processing XMPP stanzas. # Connect to the XMPP server and start processing XMPP stanzas.