From 5e60bbc230e7545ab84e2b0d1b840957175d8570 Mon Sep 17 00:00:00 2001
From: Walter Bender <walter@walter-laptop.(none)>
Date: Sun, 07 Feb 2010 13:36:18 +0000
Subject: switched to .endswith

---
(limited to 'tautils.py')

diff --git a/tautils.py b/tautils.py
index 77536b0..60c906e 100644
--- a/tautils.py
+++ b/tautils.py
@@ -156,15 +156,14 @@ def get_pixbuf_from_journal(dsobject, w, h):
             pixbuf = None
     return pixbuf
 
-def movie_media_type(suffix):
-    if suffix.replace('.','') in ['ogv','vob','mp4','wmv','mov', 'mpeg']:
-        return True
-    return False
-
-def audio_media_type(suffix):
-    if suffix.replace('.','') in ['ogg', 'oga', 'm4a']:
-        return True
-    return False
+def movie_media_type(name):
+    return name.endswith(('.ogv','.vob','.mp4','.wmv','.mov', '.mpeg'))
+
+def audio_media_type(name): 
+    return name.endswith(('.ogg', '.oga', '.m4a'))
+
+def image_media_type(name):
+    return name.endswith(('.png', '.jpg', '.jpeg', '.gif', '.tiff', '.tif'))
 
 def round_int(n):
     if int(float(n)) == n:
--
cgit v0.9.1