tahoe-lafs/src/allmydata/scripts/tahoe_put-web2ish.py

#!/usr/bin/env python

import re, sys

from twisted import web2 
from twisted.web2 import client, http, stream
import twisted.web2.client.http

from twisted.internet import defer, reactor, protocol

SERVERURL_RE=re.compile("http://([^:]*)(:([1-9][0-9]*))?")

def _put(serverurl, vdrive, vdrive_fname, local_fname, verbosity):
    """
    @param verbosity: 0, 1, or 2, meaning quiet, verbose, or very verbose

    @return: a Deferred which eventually fires with the exit code
    """
    mo = SERVERURL_RE.match(serverurl)
    if not mo:
        raise ValueError("serverurl is required to look like \"http://HOSTNAMEORADDR:PORT\"")
    host = mo.group(1)
    port = int(mo.group(3))

    d = defer.Deferred()
    
    url = "/vdrive/" + vdrive + "/"
    if vdrive_fname:
        url += vdrive_fname

    if local_fname is None or local_fname == "-":
        infileobj = sys.stdin
    else:
        infileobj = open(local_fname, "rb")
    instream = web2.stream.FileStream(infileobj)
    
    d2 = protocol.ClientCreator(reactor, web2.client.http.HTTPClientProtocol).connectTCP(host, port)

    def got_resp(resp):
        # If this isn't a 200 or 201, then write out the response data and
        # exit with resp.code as our exit value.
        if resp.code not in (200, 201,):
            def writeit(data):
                sys.stdout.write(data)

            def exit(dummy):
                d.errback(resp.code)

            return web2.http.stream.readStream(resp.stream, writeit).addCallback(exit)

        # If we are in quiet mode, then just exit with the resp.code.
        if verbosity == 0:
            d.callback(resp.code)
            return

        # Else, this is a successful request and we are not in quiet mode:
        uribuffer = []
        def gather_uri(data):
            uribuffer.append(data)

        def output_result(thingie):
            uri = ''.join(uribuffer)
            outbuf = []
            if resp.code == 200:
                outbuf.append("200 (OK); ")
            elif resp.code == 201:
                outbuf.append("201 (Created); ")
             
            if verbosity == 2:
                if resp.code == 200:
                    outbuf.append("modified existing mapping in vdrive %s of name %s to point to " % (vdrive, vdrive_fname,))
                elif resp.code == 201:
                    outbuf.append("created new mapping in vdrive %s of name %s to point to " % (vdrive, vdrive_fname,))

            outbuf.append("URI: %s" % (uri,))
        
            sys.stdout.write(''.join(outbuf))

            d.callback(resp.code)

        web2.http.stream.readStream(resp.stream, gather_uri).addCallback(output_result)
                
    def send_req(proto):
        proto.submitRequest(web2.client.http.ClientRequest('PUT', url, {}, instream)).addCallback(got_resp)

    d2.addCallback(send_req)

    return d

def put(server, vdrive, vdrive_fname, local_fname, verbosity):
    """
    This starts the reactor, does the PUT command, waits for the result, stops
    the reactor, and returns the exit code. 

    @param verbosity: 0, 1, or 2, meaning quiet, verbose, or very verbose

    @return: the exit code
    """
    d = _put(server, vdrive, vdrive_fname, local_fname, verbosity)
    exitcode = [ None ]
    def exit(result):
        exitcode[0] = result
        reactor.stop()
        return result
        
    d.addCallbacks(exit, exit)
    reactor.run()
    return exitcode[0]