pia-deluge-portforward.py

#!/usr/bin/env python
from argparse import ArgumentParser
import json
import hashlib
import subprocess
import os


def main():
    parser = ArgumentParser()
    parser.add_argument('-u', '--username', default=os.environ.get('DELUGE_USERNAME', 'localclient'))
    parser.add_argument('-p', '--password', default=os.environ.get('DELUGE_PASSWORD'))
    parser.add_argument('-I', '--interface', default='tun0')
    parser.add_argument('-b', '--base-url', default='http://209.222.18.222:2000')
    parser.add_argument('-H', '--deluge-host', default='localhost')
    parser.add_argument('-P', '--deluge-port', default=58846, type=int)

    args = parser.parse_args()

    client_id = hashlib.sha256(os.urandom(32)).hexdigest()

    out = subprocess.check_output([
        'curl',
        '-s',
        '--interface', args.interface,
        '{}/?client_id={}'.format(args.base_url, client_id),
    ])
    result = json.loads(out)
    port = result['port']

    print('Configuring Deluge to use port {}'.format(port))

    cmds = [
        'connect {}:{} {} {}'.format(args.deluge_host, args.deluge_port, args.username, args.password),
        'config --set listen_ports ({port},{port})'.format(port=port),
    ]
    subprocess.check_call([
        'deluge-console',
        ';'.join(cmds),
    ])


if __name__ == '__main__':
    main()