velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

messages: fix attachments for latest quart

This commit is contained in:
Luna 2021-08-31 22:22:17 -03:00
parent 33f79c019a
commit c4e7155f7b
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
litecord/common/messages.py
View File

@ -32,9 +32,13 @@ async def msg_create_request() -> tuple:
files = await request.files files = await request.files
for form_key, given_file in files.items():
if given_file.content_length is None:
raise BadRequest("Given file does not have content length.")
# we don't really care about the given fields on the files dict, so # we don't really care about the given fields on the files dict, so
# we only extract the values # we only extract the values
return json_from_form, [v for k, v in files.items()] return json_from_form, [x for x in files.values()]
def msg_create_check_content(payload: dict, files: list, *, use_embeds=False): def msg_create_check_content(payload: dict, files: list, *, use_embeds=False):
@ -78,8 +82,10 @@ async def msg_add_attachment(message_id: int, channel_id: int, attachment_file)
img_width, img_height = None, None img_width, img_height = None, None
# extract file size # extract file size
# TODO: this is probably inneficient # it's possible a part does not contain content length.
file_size = attachment_file.stream.getbuffer().nbytes # do not let that pass on.
file_size = attachment_file.content_length
assert file_size is not None
if is_image: if is_image:
# open with pillow, extract image size # open with pillow, extract image size