diff options
author | Walter Bender <walter@sugarlabs.org> | 2014-02-10 03:13:52 (GMT) |
---|---|---|
committer | Walter Bender <walter@sugarlabs.org> | 2014-02-10 03:13:52 (GMT) |
commit | 835e26f53a7dc15433c545fe3833b4c9c739ec9f (patch) | |
tree | 7ffe719e9e0dae9456f6cd419bf34dd67bff44f7 | |
parent | de27f30eb2e0a13dc93c975e1b236d5d9b75bcb7 (diff) |
use mimetypes library
-rw-r--r-- | TurtleArt/tautils.py | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/TurtleArt/tautils.py b/TurtleArt/tautils.py index c53d8cc..2ae9714 100644 --- a/TurtleArt/tautils.py +++ b/TurtleArt/tautils.py @@ -33,6 +33,7 @@ import pickle import subprocess import os import string +import mimetypes from gettext import gettext as _ try: @@ -483,24 +484,26 @@ def base64_to_image(data, path_name): def movie_media_type(name): ''' Is it movie media? ''' - return name.lower().endswith(('.ogv', '.vob', '.mp4', '.wmv', '.mov', - '.mpeg', '.ogg', '.webm')) + guess = mimetypes.guess_type(name) + return guess[0][0:5] == 'video' def audio_media_type(name): ''' Is it audio media? ''' - return name.lower().endswith(('.oga', '.m4a')) + guess = mimetypes.guess_type(name) + return guess[0][0:5] == 'audio' def image_media_type(name): ''' Is it image media? ''' - return name.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.tiff', - '.tif', '.svg')) + guess = mimetypes.guess_type(name) + return guess[0][0:5] == 'image' def text_media_type(name): ''' Is it text media? ''' - return name.lower().endswith(('.txt', '.py', '.lg', '.rtf')) + guess = mimetypes.guess_type(name) + return guess[0][0:4] == 'text' def round_int(num): |