コード例 #1
0
    def __init__(self):
        usage = "--user <user-name> [--database <database-name>] [--host <host-name>] [--port <port>]"
        parser = OptionParser()
        parser.set_usage(usage)
        parser.add_option("-u", "--user", dest="user", help="DataHub username")
        parser.add_option("-H",
                          "--host",
                          dest="host",
                          help="DataHub server hostname",
                          default="datahub-experimental.csail.mit.edu")
        parser.add_option("-p",
                          "--port",
                          dest="port",
                          help="DataHub server port",
                          type="int",
                          default=9000)
        parser.add_option("-d",
                          "--database",
                          dest="database",
                          help="DataHub database")
        (self.options, self.arg_values) = parser.parse_args()

        if not self.options.user:
            parser.print_usage()
            sys.exit(1)

        parser.destroy()
        self.password = getpass.getpass('DataHub password for %s: ' %
                                        (self.options.user))
        cmd2.Cmd.__init__(self, completekey='tab')
        self.client = DataHubClient(host=self.options.host,
                                    port=self.options.port)
        try:
            database = None
            if self.options.database:
                database = DHDatabase(name=self.options.database)
            con_params = DHConnectionParams(user=self.options.user,
                                            password=self.password,
                                            database=database)
            self.con = self.client.connect(con_params)
            self.prompt = "datahub> "
        except Exception, e:
            self.print_line('error: %s' % (e.message))
            sys.exit(1)
コード例 #2
0
ファイル: sample_load.py プロジェクト: sciencewiz1/datahub
#!/usr/bin/python

from client.python.dh_client import DataHubClient
from datahub.constants import *
'''
@author: anant bhardwaj
@date: Oct 11, 2013

Sample python code accesing DataHub APIs
'''

# conect to DataHub
client = DataHubClient()
con_params = DHConnectionParams(user=None,
                                password=None,
                                database=DHDatabase(name='postgres'))
con = client.connect(con_params=con_params)

result = client.load(con=con, url='demo.person')

for row in result.data.table.rows:
    print '\t'.join([c.value for c in row.cells])
    print '\n'