revi-archive/xygt
1
0
Fork 0
mirror of https://github.com/jackeilles/xygt.git synced 2024-11-08 07:52:38 +01:00
Code Issues Projects Releases Packages Wiki Activity

fixed issue with bad mimetypes, deploying.

Browse source
This commit is contained in:
Jack Eilles 2024-01-28 10:38:04 +00:00
parent 27fa621fd9
commit 33fab474fa
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
app/worker.py
View file

@ -10,7 +10,8 @@ import os
def uploadFile(file, ip, userid, filename, id, retention):
# Is the MIME and file size good?
if magic.from_buffer(file, mime=True) not in disallowedMimeTypes:
file_content = file.read()
if magic.from_buffer(file_content, mime=True) not in disallowedMimeTypes:
if file.content_length <= Config.maxFileSize:
# We're going to check whether the id variable has been filled
@ -52,7 +53,7 @@ def uploadFile(file, ip, userid, filename, id, retention):
'id': id,
'filename': filename,
'filesize': fileSize,
'mimetype': magic.from_buffer(file, mime=True) if magic.from_buffer(file, mime=True) != None else "text/plain",
'mimetype': magic.from_buffer(file_content, mime=True) if magic.from_buffer(file_content, mime=True) != None else "text/plain",
'retention': retention,
'userid': userid,
'ip': ip,