diff options
-rwxr-xr-x | bitbake/lib/toaster/toastergui/views.py | 15 | ||||
-rw-r--r-- | bitbake/toaster-requirements.txt | 1 |
2 files changed, 11 insertions, 5 deletions
diff --git a/bitbake/lib/toaster/toastergui/views.py b/bitbake/lib/toaster/toastergui/views.py index db791cfd80..6e5815595f 100755 --- a/bitbake/lib/toaster/toastergui/views.py +++ b/bitbake/lib/toaster/toastergui/views.py | |||
@@ -47,19 +47,26 @@ import json | |||
47 | from os.path import dirname | 47 | from os.path import dirname |
48 | from functools import wraps | 48 | from functools import wraps |
49 | import itertools | 49 | import itertools |
50 | import mimetypes | ||
50 | 51 | ||
51 | import magic | ||
52 | import logging | 52 | import logging |
53 | 53 | ||
54 | logger = logging.getLogger("toaster") | 54 | logger = logging.getLogger("toaster") |
55 | 55 | ||
56 | class MimeTypeFinder(object): | 56 | class MimeTypeFinder(object): |
57 | _magic = magic.Magic(flags = magic.MAGIC_MIME_TYPE) | 57 | # setting this to False enables additional non-standard mimetypes |
58 | # to be included in the guess | ||
59 | _strict = False | ||
58 | 60 | ||
59 | # returns the mimetype for a file path | 61 | # returns the mimetype for a file path as a string, |
62 | # or 'application/octet-stream' if the type couldn't be guessed | ||
60 | @classmethod | 63 | @classmethod |
61 | def get_mimetype(self, path): | 64 | def get_mimetype(self, path): |
62 | return self._magic.id_filename(path) | 65 | guess = mimetypes.guess_type(path, self._strict) |
66 | guessed_type = guess[0] | ||
67 | if guessed_type == None: | ||
68 | guessed_type = 'application/octet-stream' | ||
69 | return guessed_type | ||
63 | 70 | ||
64 | # all new sessions should come through the landing page; | 71 | # all new sessions should come through the landing page; |
65 | # determine in which mode we are running in, and redirect appropriately | 72 | # determine in which mode we are running in, and redirect appropriately |
diff --git a/bitbake/toaster-requirements.txt b/bitbake/toaster-requirements.txt index c4a2221553..1d7d21b331 100644 --- a/bitbake/toaster-requirements.txt +++ b/bitbake/toaster-requirements.txt | |||
@@ -2,5 +2,4 @@ Django==1.6 | |||
2 | South==0.8.4 | 2 | South==0.8.4 |
3 | argparse==1.2.1 | 3 | argparse==1.2.1 |
4 | wsgiref==0.1.2 | 4 | wsgiref==0.1.2 |
5 | filemagic==1.6 | ||
6 | beautifulsoup4>=4.4.0 | 5 | beautifulsoup4>=4.4.0 |