command-line downloader for shooter.cn subtitles

shooter-subdl.py

#!/usr/bin/env python
import os
import re
import md5
import sys
import json
import urllib
import urllib2

try:
    from blessings import Terminal
except ImportError, e:
    class Terminal(object):
        def __getattr__(self, name):
            def _missing(*args, **kwargs):
                return ''.join(args)
            return _missing


# globals
t = Terminal()


def compute_video_hash (filename):
    file_size = os.path.getsize(filename)
    offsets = [4096, file_size / 3 * 2, file_size / 3, file_size - 8192]
    with open(filename, "rb") as infile:
        chunks = map(lambda offset: infile.seek(offset) or infile.read(4096), offsets)
        hashes = map(lambda chunk: md5.new(chunk).hexdigest(), chunks)
    return ";".join(hashes)
        

def query (filename, vhash):
    url = "http://www.shooter.cn/api/subapi.php"
    params = urllib.urlencode({"format": "json", "lang": "Chn", "pathinfo": filename, "filehash": vhash})
    request = "{url}?{params}".format(**locals())
    response = urllib2.urlopen(request).read()
    # response = urllib2.build_opener(urllib2.ProxyHandler({'http': 'http://proxy.hinet.net:80'})).open(request).read()
    items = json.loads(response)
    return map(lambda item: item["Files"][0]["Link"], items)


def ask (choices):
    results = [] # (filename, content)
    for index, link in enumerate(choices):
        response = urllib2.urlopen(link)
        # response = urllib2.build_opener(urllib2.ProxyHandler({'http': 'http://proxy.hinet.net:80'})).open(link)
        content_disposition = response.info().getheaders("content-disposition")[0]
        filename = re.findall(r'filename=("?)(?P<filename>.+)\1', content_disposition)[0][1]
        content = response.read()
        print "{index}. {filename} ({file_size} KiB)".format(
            index=t.red(str(index+1)), filename=t.yellow(filename), file_size=t.green("%.1f"%(len(content)/1024.0)))
        print "\n".join(map(lambda line: "%8s%s"%("", line), content.split("\n")[1:4]))
        results.append((filename, content))
        # write to temp
        with open(os.path.join("/tmp", filename), "wb") as outfile:
            outfile.write(content)
    user_input = raw_input("What items do you like? (seperated by commas) [1] ")
    if user_input: answers = map(lambda x: int(x)-1, user_input.split(r","))
    else: answers = [0]
    return map(lambda index: results[index], answers)


def save(chosens):
    for filename, content in chosens:
        with open(filename, "wb") as outfile:
            outfile.write(content)
        print "Saved {filename}".format(filename=filename)


if __name__ == '__main__':
    if len(sys.argv) < 2:
        print 'Usage: %s <filename>' % sys.argv[0]
        print "Example: %s 'The.Big.Bang.Theory.S07E01.Web-DL.720p.mkv'" % sys.argv[0]
        sys.exit(1)
    filename = sys.argv[1]
    vhash = compute_video_hash(filename)
    choices = query(filename, vhash)
    chosens = ask(choices)
    save(chosens)