def interaction(cur):
d = defer.succeed(None)
for i, body in enumerate(generate_galaxy_bodies()):
then(d, cur.execute, 'INSERT INTO body VALUES (%s)', body)
if i % 1000:
then(d, log.debug, 'Inserted %d galactic bodies.', i)
return d
def main(conn, log, opts):
sqltransaction = pkg_resources.resource_string('sgg', 'sql/schema.sql')
log.debug('Loaded SQL schema:\n%s', sqltransaction)
log.info('Creating tables.')
d = conn.runOperation(sqltransaction)
then(d, log.info, 'Finished.')
return d
def main(log, args = sys.argv[1:]):
opts = DBArgumentParser.parse_args_simple(description, args)
if dbadmin:
(dbname, dbuser, dbpw) = ('postgres', None, None)
else:
(dbname, dbuser, dbpw) = (opts.dbname, opts.dbuser, opts.dbpw)
d = sql.connect(dbname, dbuser, dbpw)
d.addCallback(submain, log, opts)
d.addErrback(lambda v: log.error('%s', v))
then(d, reactor.stop)
reactor.run()
def main(conn, log, opts):
log.info("Creating user %r with password <redacted>.", opts.dbuser)
d = conn.runOperation("CREATE USER %s WITH UNENCRYPTED PASSWORD %s", [AsIs(opts.dbuser), opts.dbpw])
then(d, log.info, "Creating databse %r owned by %r.", opts.dbname, opts.dbuser)
then(d, conn.runOperation, "CREATE DATABASE %s OWNER %s", [AsIs(opts.dbname), AsIs(opts.dbuser)])
then(d, log.info, "Finished.")
then(d, println, PostscriptTemplate.format(vars(opts)))
return d
