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