Python: CouchDB example

cdb_example.py

"""
cdb_example.py
CouchDB example script

API documentation: http://packages.python.org/CouchDB/getting-started.html
"""

import sys
import couchdb

def couchdb_connect(username, password, host="127.0.0.1", session=None):
    """
    Connect to CouchDB server and return an instance object
    of the database connection or 'False' on error.
    """
    
    try:
        # Connect and send credentials.
        server = couchdb.Server(host)
        server.resource.credentials = (username, password)
        return server
    except Exception, e:
        print "Fatal error: %s" % str(e)
        return False

def main(argv=None):

    if argv is None:
        argv = sys.argv

    if argv[:1] in ['h', 'help']:
        print "This is a help text, someone write it, please."
        return 1

    connection = couchdb_connect("admin", "secret")

    if not connection:
        print "Error connecting to server, exiting."
        return 2
    else:
        # Select database 'test'.
        db = connection['test']

        # Construct data to send.
        document = {
            'name'  :   "J. James",
            'age'   :   75,
            'gender':   "Male",
            'height':   175,
            'weight':   68,
        }

        # Fetch document ID and revision.
        doc_id, doc_rev = db.save(document)

        # Fetch row with given id (=primary key)
        row = db[doc_id]

        # Print it out
        for key in row:
            print "%s: %s" % (key, row[key])    

if __name__ == "__main__":
    sys.exit(main())