Git-Mirrors/mat2-web
1
0
Fork 0
mirror of https://0xacab.org/jvoisin/mat2-web.git synced 2025-06-25 06:40:27 -04:00
Code Issues Projects Releases Packages Wiki Activity

Improve the display of supported formats

Browse source 
- Show the extension instead of the mimetype
- Sort the results
This commit is contained in:
jvoisin 2019-03-06 21:20:39 +01:00
parent 769bafac0a
commit 457d171cd4
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
main.py
View file

@ -1,6 +1,7 @@
import os import os
import hashlib import hashlib
import hmac import hmac
import mimetypes as mtype
from libmat2 import parser_factory from libmat2 import parser_factory
@ -60,7 +61,9 @@ def upload_file():
mimetypes = set() mimetypes = set()
for parser in parser_factory._get_parsers(): for parser in parser_factory._get_parsers():
mimetypes = mimetypes | parser.mimetypes mimetypes |= set(map(mtype.guess_extension, parser.mimetypes))
# since `guess_extension` might return `None`, we need to filter it out
mimetypes = sorted(filter(None, mimetypes))
if request.method == 'POST': if request.method == 'POST':
if 'file' not in request.files: # check if the post request has the file part if 'file' not in request.files: # check if the post request has the file part

4
tests.py
View file

@ -21,8 +21,8 @@ class FlaskrTestCase(unittest.TestCase):
def test_check_mimetypes(self): def test_check_mimetypes(self):
rv = self.app.get('/') rv = self.app.get('/')
self.assertIn(b'application/zip', rv.data) self.assertIn(b'.torrent', rv.data)
self.assertIn(b'audio/x-flac', rv.data) self.assertIn(b'.ods', rv.data)
def test_get_download_dangerous_file(self): def test_get_download_dangerous_file(self):
rv = self.app.get('/download/1337/\..\filename') rv = self.app.get('/download/1337/\..\filename')