Libraries for Subliminal

2011-12-23 16:02:48 +01:00
parent a03c33c1c3
commit 8efc88e691
109 changed files with 26894 additions and 0 deletions
@@ -0,0 +1,58 @@
+#!/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 import Guess
+import hashlib, os.path
+
+def hash_file(filename):
+    """Returns the ed2k hash of a given file.
+
+    >>> hash_file('test/dummy.srt')
+    'ed2k://|file|dummy.srt|44|1CA0B9DED3473B926AA93A0A546138BB|/'
+    """
+    return 'ed2k://|file|%s|%d|%s|/' % (os.path.basename(filename),
+                                        os.path.getsize(filename),
+                                        hash_filehash(filename).upper())
+
+def hash_filehash(filename):
+    """Returns the ed2k hash of a given file.
+
+    This function is taken from:
+    http://www.radicand.org/blog/orz/2010/2/21/edonkey2000-hash-in-python/
+    """
+    md4 = hashlib.new('md4').copy
+
+    def gen(f):
+        while True:
+            x = f.read(9728000)
+            if x: yield x
+            else: return
+
+    def md4_hash(data):
+        m = md4()
+        m.update(data)
+        return m
+
+    with open(filename, 'rb') as f:
+        a = gen(f)
+        hashes = [md4_hash(data).digest() for data in a]
+        if len(hashes) == 1:
+            return hashes[0].encode("hex")
+        else: return md4_hash(reduce(lambda a,d: a + d, hashes, "")).hexd