wetransfer.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# Download WeTransfer files
#
# VERSION       :1.0
# DATE          :2014-08-01
# AUTHOR        :Alejandro Alonso <https://github.com/superalex>
# URL           :https://github.com/superalex/py-wetransfer
# DEPENDS       :pip install requests

import getopt
import json
import os
import requests
import sys
if (sys.version_info > (3, 0)):
    from urllib.parse import urlparse, parse_qs
else:
    from urlparse import urlparse, parse_qs

DOWNLOAD_URL_PARAMS_PREFIX = "downloads/"
CHUNK_SIZE = 1024


def download(file_id, recipient_id, security_hash, outdir=None):
    if outdir is None:
        outdir = os.getcwd()

    if recipient_id != None:
       url = "https://wetransfer.com/api/ui/transfers/{0}/{2}/download?recipient_id={1}".format(file_id, recipient_id, security_hash)
    else:
       url = "https://wetransfer.com/api/ui/transfers/{0}/{1}/download".format(file_id, security_hash)
    r = requests.post(url)
    download_data = r.json()
    if 'error' in download_data.keys():
        print(download_data)
        raise KeyError("URL didn't work.")

    print("Downloading {0}...".format(url))
    if 'direct_link' in download_data:
        direct_link_path = urlparse(download_data['direct_link']).path
        direct_link_path = direct_link_path.split('/')
        file_name = direct_link_path[-1]

        r = requests.get(download_data['direct_link'], stream=True)
    else:
        file_name = download_data['fields']['filename']
        r = requests.post(
            download_data['formdata']['action'],
            data=download_data["fields"],
            stream=True
            )

    file_name = os.path.join(outdir, file_name)
    file_size = int(r.headers["Content-Length"])
    output_file = open(file_name, 'wb')
    counter = 0
    for chunk in r.iter_content(chunk_size=CHUNK_SIZE):
        if chunk:
            output_file.write(chunk)
            output_file.flush()
            sys.stdout.write(
                '\r{0}% {1}/{2}'.format((counter * CHUNK_SIZE) * 100/file_size,
                                        counter * CHUNK_SIZE,
                                        file_size))
            counter += 1

    sys.stdout.write('\r100% {0}/{1}\n'.format(file_size, file_size))
    output_file.close()
    return file_name


def extract_params(url):
    """
        Extracts params from url
    """
    params = url.split(DOWNLOAD_URL_PARAMS_PREFIX)[1].split('/')
    [file_id, recipient_id, security_hash] = ['', '', '']
    if len(params) > 2:
        # The url is similar to
        # https://www.wetransfer.com/downloads/XXXXXXXXXX/YYYYYYYYY/ZZZZZZZZ
        [file_id, recipient_id, security_hash] = params
    else:
        # The url is similar to https://www.wetransfer.com/downloads/XXXXXXXXXX/ZZZZZZZZ
        # In this case we have no recipient_id
        [file_id, security_hash] = params
        recipient_id = None

    return [file_id, recipient_id, security_hash]


def extract_url_redirection(url):
    """
        Follow the url redirection if necesary
    """
    return requests.get(url).url


def usage():
    print("""
You should have a we transfer address similar to https://www.wetransfer.com/downloads/XXXXXXXXXX/YYYYYYYYY/ZZZZZZZZ
So execute:
    python wetransfer.py -u https://www.wetransfer.com/downloads/XXXXXXXXXXXXXXXXXXXXXXXXX/YYYYYYYYYYYYYYYYYYYYYYYYYYYYYYY/ZZZZZ
And download it! :)
""")
    sys.exit()


def main(argv):
    try:
        opts, args = getopt.getopt(argv, "u:h", ['url', 'help'])
        url = None
        for opt, arg in opts:
            if opt in ('-u', '--url'):
                url = arg
            if opt in ('-h', '--help'):
                usage()

        if len(argv) == 0:
            usage()

        if argv[0].find('http') == 0:
            url = argv[0]

        if not url:
            usage()

        url = extract_url_redirection(url)
        [file_id, recipient_id, security_hash] = extract_params(url)
        download(file_id, recipient_id, security_hash)

    except getopt.GetoptError:
        usage()
        sys.exit(2)


if __name__ == "__main__":
    main(sys.argv[1:])

#  vim: set nowrap sw=3 sts=3 et fdm=marker: