~chris-atlee/blobs/main

#!/usr/bin/env python2.5

"""

Blob File Server client

"""

import hashlib, os, email

import simplejson

from blobs.client.httpclient import HTTPClient, FileProxy

 

class BlobClient(object):

    def __init__(self, url):

        """

        Creates a new client object

        You must pass in the url to the Blob server

        """

        self.client = HTTPClient(url)

 

    def getFile(self, hash, output):

        url = "/files/%s" % hash

 

        res = self.client.get(url)

 

        if res.status != 200:

            raise ValueError("File not found")

 

        if hasattr(output, "write"):

            fp = output

        else:

            if os.path.isdir(output):

                contentDisp = res.getheader("content-disposition", None)

                msg = email.message_from_string(str(res.msg))

                filename = msg.get_filename()

                fp = open(filename, "w")

            else:

                fp = open(output, "w")

 

        while True:

            data = res.read(4096)

            if not data:

                break

            fp.write(data)

 

    def putFile(self, filename, formparam="file", size = None,

                inputFileObj = None):

 

        if inputFileObj is None:

            inputFileObj = open(filename)

 

        if size is None:

            size = os.path.getsize(filename)

 

        res = self.client.post("/files",

                {formparam: FileProxy(filename = filename,

                    file_obj = inputFileObj, size = size),

                 'filename': filename})

 

        return res.status, res.reason

 

    def listFiles(self):

        res = self.client.get("/files.json")

 

        return simplejson.load(res)

 

if __name__ == "__main__":

    from optparse import OptionParser

    import os, sys

 

    parser = OptionParser()

    parser.set_defaults(action=None,

            server=os.environ.get("BLOB_SERVER"),

            output=".")

 

    parser.add_option("-s", "--server", dest="server",

            help="Specify the server to connect to.  Alternatively set the BLOB_SERVER environment variable")

    parser.add_option("-l", "--list", dest="action", action="store_const",

            const="list", help="Print file listing")

    parser.add_option("-u", "--upload", dest="action", action="store_const",

            const="upload",

            help="Upload the specified file(s) to the server")

    parser.add_option("-d", "--download", dest="action", action="store_const",

            const="download",

            help="Download the specified list of files given as a list of hashes")

    parser.add_option("-o", "--output", dest="output",

            help="""Where to save the downloaded file(s).  If multiple files

are to be downloaded, then this must be a directory.

 

Otherwise, if a single file is to be downloaded, this can be the name of a file

to write to.  It can also be the directory to save the file into, in which case the original filename will be preserved.  It can also be '-' then output will be written to stdout""")

 

    options, args = parser.parse_args()

 

    if not options.server:

        parser.error("Need to specify a server")

 

    c = BlobClient(options.server)

 

    if options.action == "list":

        for h, size, filename in c.listFiles():

            print h, filename

 

    elif options.action == "upload":

        for fn in args:

            print c.putFile(fn)

 

    elif options.action == "download":

        if len(args) == 1:

            if options.output == "-":

                c.getFile(args[0], sys.stdout)

            elif os.path.isdir(options.output):

                c.getFile(args[0], options.output)

            else:

                c.getFile(args[0], open(options.output, "w"))

        else:

            if options.output == "-":

                parser.error("You cannot output to stdout when multiple files are downloaded")

            elif not os.path.isdir(options.output):

                parser.error("You must output to a directory when multiple files are downloaded")

 

            for h in args:

                c.getFile(h, options.output)