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

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]