get_subtitles.py

import struct, os, sys, subprocess, urllib
from xmlrpclib import ServerProxy,ProtocolError

MOVIE_FORMATS = ['avi','mkv']

def getMovieHash(movie):
    longlongformat = 'Q'
    bytesize = struct.calcsize(longlongformat)
    format = "<%d%s" % (65536//bytesize, longlongformat)
    f = open(movie,'rb')
    filesize = os.fstat(f.fileno()).st_size
    hash = filesize

    #if filesize < 65536 * 2:
    #    return "SizeError"

    buffer= f.read(65536)
    longlongs= struct.unpack(format, buffer)
    hash+= sum(longlongs)
    f.seek(-65536, os.SEEK_END)
    buffer= f.read(65536)
    longlongs= struct.unpack(format, buffer)
    hash+= sum(longlongs)
    hash&= 0xFFFFFFFFFFFFFFFF
    f.close()
    returnedhash =  "%016x" % hash
    return (returnedhash,filesize)

if len(sys.argv) < 2:
    print "|ERROR| ----- You need to set a base directory to search for media"
    sys.exit()

directory = sys.argv[1]

videos = []
for f in MOVIE_FORMATS:
    res = subprocess.Popen(['find',directory,'-name','*%s'%f],stdout=subprocess.PIPE).communicate()[0]
    if res != '':
        videos.extend(res.split('\n'))

#remove empty strings from list
videos = filter(lambda x: x != '', videos)

print "|INFO| ----- Found the following movies:"
for vid in videos:
    print "             %s"%vid

s = ServerProxy('http://api.opensubtitles.org/xml-rpc', allow_none=True)
try:
    res = s.LogIn('peddromcaraujo','58TRmLf6Tccl','en','VLSub')
    print "|INFO| ----- Connected Opensubtitles.org"
except ProtocolError:
    print "|ERROR| ----- Opensubtitles.org is probably overloaded. Give it some time....."
    sys.exit()

# [{'moviebytesize': '366786266', 'sublanguageid': 'eng', 'moviehash': 'dcb7dbae558c63f6'}, {'moviebytesize': '576700632', 'sublanguageid': 'eng', 'moviehash': 'f2671cfa1d9ece6f'}]
videos_info = []
query = []
for video in videos:
    returnedhash,filesize = getMovieHash(video)
    videos_info.append({'file':video,'moviehash':returnedhash,'sublanguageid':'eng','moviebytesize':str(filesize)})
    query.append({'moviehash':returnedhash,'sublanguageid':'eng','moviebytesize':str(filesize)})

subtitles = s.SearchSubtitles(res['token'],query)

print "|INFO| ----- Search complete, will begin to process the findings....."

for i in subtitles['data']:
    movie_hash = i['MovieHash']
    for h in videos_info:
        if movie_hash == h['moviehash'] and h.get('subdownloadlink',None) is None:
            h.update({'subdownloadlink':i['SubDownloadLink'],'subformat':i['SubFormat']})

#lets download the subtitles, unpack them, change the name and put them together with their respective file
for video in videos_info:
    filename = video['file'].split('/')[-1]
    path = '/'.join(video['file'].split('/')[:-1])
    for format in MOVIE_FORMATS:
        filename = filename.rstrip(format)
    try:
        url = video['subdownloadlink']
        webFile = urllib.urlopen(url)
        subtitle_file = "%s/%s"%(path,url.split('/')[-1])
        subtitle_filename = url.split('/')[-1].rstrip('.gz')
        localFile = open(subtitle_file, 'w')
        localFile.write(webFile.read())
        webFile.close()
        localFile.close()
        subprocess.Popen(['gzip','-d',subtitle_file]).wait()
        os.rename('%s/%s'%(path,subtitle_filename),'%s/%s%s'%(path,filename,video['subformat']))
        print "|INFO| ----- Downloaded subtitles for %savi"%filename
    except:
        print "|ERROR| ----- Did not find subtitle for %s"%filename