#!/usr/bin/env python
# NukeFS2: simplest shortener-based file cloud storage for Python 2.6+
# Chain storage version
# 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'
	chunks = []
	with open(localpath, 'rb') as fileobj:
		while 1:
			data = fileobj.read(CHUNKLEN)
			if not data: break
			chunks.append(''.join(data.encode('base64').splitlines()))
		fileobj.close()
	baseurl = 'http://tinyurl.com/api-create.php?url='
	key = None
	for i in xrange(len(chunks),0,-1):
		chunkobj = {'data' : chunks[i-1], 'next' : key}
		encoded = quote_plus(dumps(chunkobj))
		key = get(baseurl+encoded).text[19:]
	return key
    
def download(key, localpath):
	'Download a file to localpath by its access key'
	with open(localpath, 'wb') as fileobj:
		while 1:
			data = get('http://tinyurl.com/'+key, allow_redirects=False).headers['location']
			if not data: break
			chunkobj = loads(unquote_plus(data))
			if not chunkobj['data']: break
			fileobj.write(chunkobj['data'].decode('base64'))
			if not chunkobj['next']: break
			key = chunkobj['next']
		fileobj.close()

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