Aria2 RPC interface

aria2rpc.py

#!/usr/bin/env python

import json, urllib2, sys, os
from argparse import ArgumentParser

parser = ArgumentParser()
parser.add_argument('-c', '--cookie', help='use cookies', type=str, 
					default='', metavar='COOKIES', dest='cookies')
parser.add_argument('-o', '--output', help='output name', type=str, 
					default='', metavar='NAME', dest='output')
parser.add_argument('-d', '--dir', help='dest dir (server side)', type=str, 
					default='', metavar='DIR', dest='dir')
parser.add_argument('-R', '--rpc', 
					help='aria2 rpc (http://localhost:6800/jsonroc)',
					type=str, default='http://127.0.0.1:6800/jsonrpc',
					metavar='URL', dest='rpc')
parser.add_argument('-r', '--referer', help='referer', default='', type=str, 
					metavar='URL', dest='referer')
# parser.add_argument('-s', '--session-dir',
# 					help='session file dir (server side)', default='', 
# 					type=str, metavar='DIR', dest='session')
parser.add_argument('URIs', nargs='+', help='URIs', type=str, 
					default='', metavar='URI')
opts = parser.parse_args()

jsondict = {'jsonrpc':'2.0', 'id':'qwer',
		'method':'aria2.addUri','params':[opts.URIs]}

aria2optsDefault={
		'continue'                  :'true',
		'max-connection-per-server' :15,
		'split'                     :15,
		'min-split-size'            :'10M',
		'user-agent':'Mozilla/5.0 (X11; Linux; rv:5.0) Gecko/5.0 Firefox/5.0'}

aria2opts = {}
aria2opts.update(aria2optsDefault)

if opts.output:
	aria2opts['out'] = opts.output
if opts.dir:
	aria2opts['dir'] = opts.dir
	# if opts.session:
	# 	aria2opts['save-session'] = \
	# 			os.path.join(opts.session, opts.output + '.session')
if opts.referer:
	aria2opts['referer'] = opts.referer
if opts.cookies:
	aria2opts['header'] = ['Cookie: {0}'.format(opts.cookies)]

jsondict['params'].append(aria2opts)

jsonreq = json.dumps(jsondict)
# print jsonreq
urllib2.urlopen(opts.rpc, jsonreq)