From a4304fd8b93777047047b8fea63ac2975e37764e Mon Sep 17 00:00:00 2001
From: j <0x006A@0x2620.org>
Date: Thu, 16 Aug 2012 18:57:57 +0200
Subject: [PATCH] add more extensions, make extension accessible

---
 ox/movie.py | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/ox/movie.py b/ox/movie.py
index 203dbb7..4d3f8e5 100644
--- a/ox/movie.py
+++ b/ox/movie.py
@@ -13,6 +13,19 @@ from text import get_sort_name, findRe
 
 __all__ = ['parse_movie_path', 'create_movie_path', 'get_oxid']
 
+extensions = {
+    'audio': [
+        'aac', 'flac', 'm4a', 'mp3', 'oga', 'ogg', 'wav', 'wma'
+    ],
+    'subtitle': [
+        'idx', 'srt', 'sub'
+    ],
+    'video': [
+        'avi', 'divx', 'dv', 'flv', 'm4v', 'mkv', 'mov', 'mp4',
+        'mpeg', 'mpg', 'mts', 'ogm', 'ogv', 'rm', 'webm', 'wmv'
+    ],
+}
+
 '''
 Naming scheme:
 X/[Group, The; Lastname, Firstname/]The Title[ (YEAR[-YEAR])]/
@@ -60,15 +73,10 @@ def parse_path(path):
     def parse_title(string):
         return title, year
     def parse_type(string):
-        if string in ['aac', 'm4a', 'mp3', 'ogg']:
-            type = 'audio'
-        elif string in ['idx', 'srt', 'sub']:
-            type = 'subtitle'
-        elif string in ['avi', 'divx', 'm4v', 'mkv', 'mov', 'mpg', 'ogv', 'rm']:
-            type = 'video'
-        else:
-            type = None
-        return type
+        for type in extensions:
+            if string in extensions[type]:
+                return type
+        return None
     def parse_underscores(string):
         string = re.sub('^_', '.', string)
         string = re.sub('_$', '.', string)