get direct resource link from masv download link

get_masv.py

import requests
import re
import sys

root_url = "https://api.massive.app/v1"


def get_masv(dl_link, password=None):
    print('Retrieving direct link(s)...')
    split_link = re.split('[/?=]', dl_link)
    link_id = split_link[3]
    secret = split_link[-1]
    header = {}
    if password:
        header = {'X-Link-Password': password}
    r = requests.get(
        f'{root_url}/links/{link_id}',
        headers=header,
        params={'secret': secret}
    )
    if not r.ok:
        return False, r.json()
    package_token = r.json()["package_token"]
    package_id = r.json()["package_id"]
    r2 = requests.get(
        f'{root_url}/packages/{package_id}/files',
        headers={'X-Package-Token': package_token}
    )
    if not r2.ok:
        return False, r2.json()
    files_url = {}
    for item in r2.json():
        r3 = requests.get(
            f'{root_url}/packages/{package_id}/files/{item["id"]}/download',
            headers={'X-Package-Token': package_token}
        )
        files_url[item["name"]] = r3.json()['url']
    return True, files_url


if __name__ == "__main__":
    if len(sys.argv) < 2 or len(sys.argv) > 3:
        print(
            'Correct usage is python get_masv.py [download link] [password (optional)]'
        )
        sys.exit()
    status, resp = get_masv(*sys.argv[1::])
    if status:
        print('Success! The links are: \n')
        print(resp)
    else:
        print('Something went wrong: \n')
        print(resp)