def create_database(self):
cmd_executor = CmdExecutor()
db_create_cmd = ''
if self.driver == 'mysql':
# mysql
db_create_cmd = 'mysql -u %s -p%s -h %s -P %s -e "create database %s;"' % \
(self.username, self.password, self.host, self.port, self.dbname)
elif self.driver == 'postgres':
# postgres
env = os.environ.copy()
env['PGPASSWORD'] = self.password
cmd = 'psql -p %s postgres %s -c "create database %s with owner = %s;"'\
% (self.port, self.username, self.dbname, self.username)
else:
raise Exception, "unknown database driver"
out, err, ret = cmd_executor.run_cmd(db_create_cmd)
if ret:
raise Exception, "database create error %s, %s" %(err)
return ret
def remove_database(self, database='lpt'):
cmd_executor = CmdExecutor()
db_remove_cmd = ''
if self.driver == 'mysql':
# mysql
db_remove_cmd = 'mysql -u %s -p%s -h %s -P %s -e "drop database %s;"' \
%(self.username, self.password, self.host, self.port, self.dbname)
elif self.driver == 'postgres':
# postgres
password = self.password
env = os.environ.copy()
env['PGPASSWORD'] = password
db_remove_cmd = 'psql -p %s postgres %s -c "drop database %s;"'\
%(self.port, self.username, self.dbname)
else:
raise Exception, "unknown database driver"
out, err, ret = cmd_executor.run_cmd(db_remove_cmd)
if ret:
raise Exception, "database remove error: %s" %(err)
return ret
