New metadata scanner

2012-01-27 18:54:12 +01:00
parent fe7ac58bf5
commit 228f7e9e9e
192 changed files with 193 additions and 44918 deletions
@@ -0,0 +1,76 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# GuessIt - A library for guessing information from filenames
+# Copyright (c) 2011 Nicolas Wack <wackou@gmail.com>
+#
+# GuessIt is free software; you can redistribute it and/or modify it under
+# the terms of the Lesser GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# GuessIt is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# Lesser GNU General Public License for more details.
+#
+# You should have received a copy of the Lesser GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+from guessit.patterns import subtitle_exts, video_exts, episode_rexps, find_properties, canonical_form
+import os.path
+import re
+import logging
+
+log = logging.getLogger("guessit.filetype")
+
+
+def guess_filetype(filename, filetype = 'autodetect'):
+    other = {}
+
+    # look at the extension first
+    fileext = os.path.splitext(filename)[1][1:].lower()
+    if fileext in subtitle_exts:
+        if 'movie' in filetype:
+            filetype = 'moviesubtitle'
+        elif 'episode' in filetype:
+            filetype = 'episodesubtitle'
+        else:
+            filetype = 'subtitle'
+        other = { 'container': fileext }
+    elif fileext in video_exts:
+        if filetype == 'autodetect':
+            filetype = 'video'
+        other = { 'container': fileext }
+    else:
+        if filetype == 'autodetect':
+            filetype = 'unknown'
+        other = { 'extension': fileext }
+
+    # now look whether there are some specific hints for episode vs movie
+    if filetype in ('video', 'subtitle'):
+        for rexp, confidence, span_adjust in episode_rexps:
+            match = re.search(rexp, filename, re.IGNORECASE)
+            if match:
+                if filetype == 'video':
+                    filetype = 'episode'
+                elif filetype == 'subtitle':
+                    filetype = 'episodesubtitle'
+                break
+
+        for prop, value, start, end in find_properties(filename):
+            if canonical_form(value) == 'DVB':
+                if filetype == 'video':
+                    filetype = 'episode'
+                elif filetype == 'subtitle':
+                    filetype = 'episodesubtitle'
+                break
+
+        # if no episode info found, assume it's a movie
+        if filetype == 'video':
+            filetype = 'movie'
+        elif filetype == 'subtitle':
+            filetype = 'moviesubtitle'
+
+    return filetype, other