subtitle_downloader.py

import urllib
import os
import hashlib
import sys


def get_hash(name):
    readsize = 64 * 1024
    with open(name, "rb") as f:
        size = os.path.getsize(name)
        data = f.read(readsize)
        f.seek(-readsize, os.SEEK_END)
        data += f.read(readsize)
    return hashlib.md5(data).hexdigest()


def main():
    userAgent = (
        "SubDB1.0 (seema1711/0.1; https://github.com/seema1711/subtitle-downloader)"
    )
    moviePath = sys.argv[1]
    movieName = os.path.join(os.getcwd(), moviePath)
    language = "en"
    action = "download"
    baseURL = "http://api.thesubdb.com/?"
    hashed = get_hash(movieName)

    content = {
        "action": action,
        "hash": hashed,
        "language": language,
    }

    url = baseURL + urllib.urlencode(content)
    request = urllib.Request(url)
    request.add_header("User-Agent", userAgent)
    response = urllib.urlopen(request)
    subtitles = response.read()

    index = movieName.rfind(".")
    fileName = movieName[0:index] + ".srt"
    with open(fileName, "w") as f:
        f.write(subtitles)
    print("Subtitle Downloaded!!")