xep_0363: pass more information to FileTooBig exception

Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2020-05-12 01:49:56 +02:00
parent e57289358f
commit c7cd2fcf33
Signed by: pep
GPG key ID: DEDA74AEECA9D0F2

View file

@ -28,7 +28,9 @@ class UploadServiceNotFound(FileUploadError):
pass
class FileTooBig(FileUploadError):
pass
def __str__(self):
return 'File size too large: {} (max: {} bytes)' \
.format(self.args[0], self.args[1])
class HTTPError(FileUploadError):
def __str__(self):
@ -126,7 +128,7 @@ class XEP_0363(BasePlugin):
input_file.seek(0)
if size > self.max_file_size:
raise FileTooBig()
raise FileTooBig(size, self.max_file_size)
if content_type is None:
content_type = guess_type(filename)[0]