From 835e26f53a7dc15433c545fe3833b4c9c739ec9f Mon Sep 17 00:00:00 2001 From: Walter Bender Date: Mon, 10 Feb 2014 03:13:52 +0000 Subject: use mimetypes library --- 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): -- cgit v0.9.1