hueso/gajim-plugins
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'httpupload_max_file_size' into 'gtk3'

Browse Source 
[httpupload] Use 'max-file-size' information

See merge request !44
This commit is contained in:
Philipp Hörist
2017-09-01 19:23:36 +02:00
parent 4e3c984145 21f941266f
commit 86aacfc31b
1 changed files with 39 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
httpupload/httpupload.py
View File

@@ -72,13 +72,23 @@ class HTTPUploadPlugin(GajimPlugin):
del IQ_CALLBACK[id_]
def handle_agent_info_received(self, event):
if (NS_HTTPUPLOAD in event.features and
if (NS_HTTPUPLOAD not in event.features or not
app.jid_is_transport(event.jid)):
account = event.conn.name
interface = self.get_interface(account)
interface.enabled = True
interface.component = event.jid
interface.update_button_states(True)
return
account = event.conn.name
interface = self.get_interface(account)
interface.enabled = True
interface.component = event.jid
interface.update_button_states(True)
try:
for form in event.data:
tmp = form.getField("max-file-size").getValue()
interface.max_file_size = int(tmp)
except AttributeError:
interface.max_file_size = None
log.warning("%s does not provide maximum file size" % account)
def handle_outgoing_stanza(self, event):
message = event.msg_iq.getTagData('body')
@@ -183,14 +193,23 @@ class Base(object):
return
invalid_file = False
stat = os.stat(path)
if os.path.isfile(path):
stat = os.stat(path)
if stat[6] == 0:
invalid_file = True
msg = _('File is empty')
else:
invalid_file = True
msg = _('File does not exist')
if self.max_file_size is not None and \
stat.st_size > self.max_file_size:
invalid_file = True
msg = _('File is too large, maximum allowed file size is: %s') % \
GLib.format_size_full(self.max_file_size,
GLib.FormatSizeFlags.IEC_UNITS)
if invalid_file:
ErrorDialog(_('Could not open file'), msg,
transient_for=chat_control.parent_win.window)
@@ -248,12 +267,24 @@ class Base(object):
IQ_CALLBACK[id_] = lambda stanza: self.received_slot(stanza, file)
app.connections[self.account].connection.send(iq)
@staticmethod
def get_slot_error_message(stanza):
tmp = stanza.getTag('error').getTag('file-too-large')
if tmp is not None:
max_file_size = int(tmp.getTag('max-file-size').getData())
return _('File is too large, maximum allowed file size is: %s') % \
GLib.format_size_full(max_file_size,
GLib.FormatSizeFlags.IEC_UNITS)
return stanza.getErrorMsg()
def received_slot(self, stanza, file):
log.info("Received slot")
if stanza.getType() == 'error':
file.progress.close_dialog()
ErrorDialog(_('Could not request upload slot'),
stanza.getErrorMsg(),
self.get_slot_error_message(stanza),
transient_for=file.control.parent_win.window)
log.error(stanza)
return