#!/usr/bin/env python
# NukeFS: simplest shortener-based file cloud storage for Python 2.6+
# by Multiversum
# Note: it gets damn slow on large files so plz don't upload >1M files

from requests import get
from json import dumps, loads
from urllib import quote_plus, unquote_plus

CHUNKLEN=22000 #current TinyURL effective chunk length limitation

def upload(localpath):
	'Upload a file from localpath to cloud and get its access key'
	filekeys = []
	baseurl = 'http://tinyurl.com/api-create.php?url='
	with open(localpath, 'rb') as fileobj:
		while 1:
			data = fileobj.read(CHUNKLEN)
			if not data: break
			encoded = quote_plus(''.join(data.encode('base64').splitlines()))
			key = get(baseurl+encoded).text[19:]
			filekeys.append(key)
		fileobj.close()
	return get(baseurl+quote_plus(dumps(filekeys))).text[19:]
    
def download(filekey, localpath):
	'Download a file to localpath by its access key'
	filekeys = loads(unquote_plus(get('http://tinyurl.com/'+filekey, allow_redirects=False).headers['location']))
	with open(localpath, 'wb') as fileobj:
		for key in filekeys:
			data = get('http://tinyurl.com/'+key, allow_redirects=False).headers['location']
			if not data: break
			fileobj.write(unquote_plus(data).decode('base64'))
		fileobj.close()

if __name__ == '__main__': #usage example
	megakey = upload('uploaded.zip')
	print 'File access key: ', megakey
	download(megakey, 'downloaded.zip')
	print 'Downloaded.'