edit retrieve mime thing

This commit is contained in:
Jack Eilles 2024-01-28 10:24:13 +00:00
parent 41e70bf9c3
commit 359309c830

View file

@ -139,11 +139,11 @@ def getData(id):
with open(os.path.join(Config.fileDir, secure_filename(id)), "rb") as f: with open(os.path.join(Config.fileDir, secure_filename(id)), "rb") as f:
file = f.read() file = f.read()
# Get MIME type from file, if fails then use magic # Get the mimetype from the db
try: try:
mimetype = data["mimetype"] mimetype = data["mimetype"]
except KeyError: except:
mimetype = magic.from_buffer(file, mime=True) mimetype = "text/plain" # This is the default because it seems loads of files are being given the wrong mime
# Return the file with the correct MIME type # Return the file with the correct MIME type
return send_file(io.BytesIO(file), mimetype=mimetype) return send_file(io.BytesIO(file), mimetype=mimetype)