Napiprojekt API. Pobieranie napisów do filmu

np.py

import sys
import hashlib
import base64
import os.path
import urllib.request, urllib.parse
import xml.etree.ElementTree as ET

URL='http://www.napiprojekt.pl/api/api-napiprojekt3.php'


def hash(fileName):
    h = hashlib.md5()
    c = 2**10
    n = 0
    with open(fileName, 'rb') as f:
        while n < 2**10 * 10:
            data = f.read(c)
            if not data:
                break
            h.update(data)
            n += 1
        f.close()
    return h.hexdigest()


def get(hash, language='PL'):
    params = {
        'downloaded_subtitles_lang': language,
        'downloaded_subtitles_txt': '1',
        'downloaded_subtitles_id': hash,
        'client': 'NapiProjekt',
        'mode': '1'
    }
    params = urllib.parse.urlencode(params).encode('ascii')
    request = urllib.request.Request(url=URL, data=params, method='POST')
    with urllib.request.urlopen(request) as f:
        root = ET.fromstring(f.read().decode('utf8'))
        text = root.find('subtitles').find('content').text
        return base64.b64decode(text).decode('utf8')

if __name__ == "__main__":
    if len(sys.argv) < 2:
        print("Please provide file name")
        exit(1)

    fileName = sys.argv[1]

    if not os.path.isfile(fileName):
        print("File not found")
        exit(1)

    outputFileName = os.path.splitext(fileName)[0] + ".txt"

    with open(outputFileName, 'w', encoding='utf8') as f:
        f.write(get(hash(fileName)))
        f.close()