該腳本針對批量導入數(shù)據(jù)sql文件，data目錄存放所要導入的sql文件，list.txt存放要導入的列表信息。
腳本內容如下：
#!/usr/bin/env python # -*- coding:utf-8 -*- import os, sys, logging, datetime # 日志文件配置 if not os.path.isdir('logs'): os.mkdir('logs') logging.basicConfig(level=logging.INFO, format='%(asctime)s %(name)-12s %(levelname)-8s %(message)s', datefmt='%Y-%m-%d %H:%M', filename='logs/importable.log', filemode='a') console = logging.StreamHandler() console.setLevel(logging.INFO) formatter = logging.Formatter('%(message)s') console.setFormatter(formatter) logging.getLogger('').addHandler(console) logger = logging.getLogger(__name__) # 從本地配置文件中讀取主機列表信息 def readinfo(confile): info_list = [] if os.path.isfile(confile): with open(confile, 'r+') as f: for line in f.readlines(): if not line.startswith("#"): dict = {} dict['id'] = line.split()[0] dict['host'] = line.split()[1] dict['name'] = line.split()[2] info_list.append(dict) return info_list else: logger.error(u"本地配置文件%s不存在！" % confile) sys.exit(1) def readsqlfile(data): sql_list = [] # 判斷本地的sql文件是否存在目錄是否存在。 if os.path.isdir(data): for sql_file in os.listdir(data): if sql_file: sql_list.append(os.path.join(data,sql_file)) else: logger.error(u"本地%s目錄中沒有要導入的sql文件！"%data) else: logger.error(u"%s 不是一個目錄。"%data) return sql_list # 循環(huán)info_list和sql_list將sql導入每一個服務器 def importhost(info_list, sql_list): succ_list = [] fail_list = [] for ser in info_list: for sql_file in sql_list: cmd = "%s/mysql -u%s -p%s -h'%s' %s < %s" % (mysql_path, mysql_user, mysql_passwd, ser['host'], ser['name'] ,sql_file) status = os.system(cmd) if status == 0: msg = "%s -- %s -- %s 導入sql：%s 完畢！"%(ser['id'],ser['host'],ser['name'],sql_file) succ_list.append(msg) logger.info(msg) else: msg = "%s -- %s -- %s 導入sql：%s 失?。? % (ser['id'], ser['host'], ser['name'], sql_file) fail_list.append(msg) logger.error(msg) sys.exit(1) return succ_list,fail_list def yesorno(confile,data): print "數(shù)據(jù)庫列表如下:" info_list = readinfo(confile) for ser in info_list: print "%s --- %s ---- %s"%(ser['id'],ser['host'],ser['name']) iput = raw_input("你是否要導入sql文件 %s，請輸入yes or no:" % (os.listdir(data))) while True: if iput.lower() in ['y','yes']: print "你選擇了繼續(xù)執(zhí)行程序動作！" break elif iput.lower() in ['n','no']: print "你選擇了退出程序動作!" sys.exit(1) else: print "你輸入了非法的字符,腳本并未執(zhí)行導入sql文件的任何動作，程序即將退出。" sys.exit(1) if __name__ == '__main__': # 基本信息配置變量 data = r'data' # 存放本地sql文件的目錄 confile = 'list.txt' # 存放服務器列表信息,格式如： id host dbname ： 1 192.168.2.20 game_name_cn1 # MySQL變量 mysql_path = '/usr/local/mysql/bin' mysql_user = 'root' mysql_passwd = 'xirexrt,mf' yesorno(confile,data) startime = datetime.datetime.now() sql_list = readsqlfile(data) info_list = readinfo(confile) succ,fail= importhost(info_list,sql_list) logger.info(u'總共執(zhí)行成功%s條. \r\n 總共執(zhí)行失敗%s條.'%(len(succ), len(fail))) endtime = datetime.datetime.now() logger.info(u"\r\nTotal time: %s ms" % ((endtime - startime).microseconds / 1000))另外有需要云服務器可以了解下創(chuàng)新互聯(lián)cdcxhl.cn，海內外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。
重慶分公司
重慶分公司
