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

migrate to magic for mime detection

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

6
app/worker.py
View file

@ -4,13 +4,13 @@ from app import bcrypt
import secrets
import datetime
import random
import time
import magic
import os
def uploadFile(file, ip, userid, filename, id, retention):
# Is the MIME and file size good?
if file.content_type not in disallowedMimeTypes:
if magic.from_buffer(file, 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 +52,7 @@ def uploadFile(file, ip, userid, filename, id, retention):
'id': id,
'filename': filename,
'filesize': fileSize,
'mimetype': file.content_type if file.content_type != None else "text/plain",
'mimetype': magic.from_buffer(file, mime=True) if magic.from_buffer(file, mime=True) != None else "text/plain",
'retention': retention,
'userid': userid,
'ip': ip,