plugins/upload: handle more exceptions

Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2020-05-12 01:58:37 +02:00
parent ac3fb69574
commit 8b385a0c99
Signed by: pep
GPG key ID: DEDA74AEECA9D0F2

View file

@ -24,7 +24,7 @@ import traceback
from os.path import expanduser from os.path import expanduser
from glob import glob from glob import glob
from slixmpp.plugins.xep_0363.http_upload import UploadServiceNotFound from slixmpp.plugins.xep_0363.http_upload import FileTooBig, HTTPError, UploadServiceNotFound
from poezio.plugin import BasePlugin from poezio.plugin import BasePlugin
from poezio.core.structs import Completion from poezio.core.structs import Completion
@ -56,6 +56,9 @@ class Plugin(BasePlugin):
except UploadServiceNotFound: except UploadServiceNotFound:
self.api.information('HTTP Upload service not found.', 'Error') self.api.information('HTTP Upload service not found.', 'Error')
return None return None
except (FileTooBig, HTTPError) as exn:
self.api.information(str(exn), 'Error')
return None
except Exception: except Exception:
exception = traceback.format_exc() exception = traceback.format_exc()
self.api.information('Failed to upload file: %s' % exception, self.api.information('Failed to upload file: %s' % exception,