def dump(self):
fun = inspect.stack()[0][3]
dbs = conf.dbs
tables = conf.table
cols = conf.columns
'''
if (len(dbs) > 1 and len(tables)>0) or (len(tables)>1 and len(cols>0)):
logger.info("Please -D db -T table -C col --dump")
'''
for db in dbs:
if not tables:
tables = self.do("get_tables", db)
logger.info(tables)
for table in tables:
if (not conf.table
and not conf.columns) or (conf.table
and not conf.columns):
cols = self.do("get_columns", db, table)
logger.info(cols)
out = PrettyTable()
info = "{0}.{1}\n".format(db, table)
logger.info(info)
put_file_contents(DUMP_FILE, info)
for col in cols:
values = self.do(fun, db, table, col)
out.add_column(col, values)
logger.info(out)
put_file_contents(DUMP_FILE, str(out))
return
def get_dbs(self):
fun = inspect.stack()[0][3]
dbs = self.do(fun)
out = PrettyTable()
out.add_column("DATABASE:", dbs)
logger.info(out)
put_file_contents(DUMP_FILE, str(out))
return
def get_columns(self):
fun = inspect.stack()[0][3]
db = conf.dbs[0]
tables = conf.table
for table in tables:
cols = self.do(fun, db, table)
out = PrettyTable()
out.add_column(table, cols)
logger.info(out)
put_file_contents(DUMP_FILE, "table:{0}.{1}".format(db, table))
put_file_contents(DUMP_FILE, str(out))
return
def get_tables(self):
fun = inspect.stack()[0][3]
dbs = conf.dbs
tables = []
for db in dbs:
tables = self.do(fun, db)
logger.success(db)
out = PrettyTable()
out.add_column("TABLES:", tables)
logger.info(out)
put_file_contents(DUMP_FILE, "db:{0}".format(db))
put_file_contents(DUMP_FILE, str(out))
return
