python实现批量修改服务器密码的方法

#_*_ coding: utf-8 _*_'''@author liaogs'''import jsonimport xlrdimport xlwtimport timeimport datetimeimport base64import randomfrom xlutils.copy import copyclass excelhandler():  def __init__(self,path):    self.path = path    self.workbook = None    self.rows = 0    self.cols = 0    self.serverlist = []  def read_excel(self):    self.workbook = xlrd.open_workbook(self.path)    sh1 = self.workbook.sheet_by_index(0)    self.rows = sh1.nrows    self.cols = sh1.ncols    for row in range(1,sh1.nrows):      server = []      for col in [0,1,2,sh1.ncols-2,sh1.ncols-1]:        server.append(sh1.cell(row,col).value)      self.serverlist.append(server)  def gen_new_password_excel(self):    old_excel = xlrd.open_workbook(self.path)    new_excel = copy(old_excel)    ws = new_excel.get_sheet(0)    coldt = "pass"+ str(datetime.date.today())    ws.write(0,self.cols,coldt)    for row in range(1,self.rows):      ws.write(row,self.cols,self.gen_key())    dt = time.strftime("%Y%m%d%H%M%S",time.localtime())    new_excel.save(dt+self.path)  def write_excel(self,serverlist):    wb = xlwt.Workbook()    ws = wb.add_sheet(u'sheet1',cell_overwrite_ok=True)    header = ["IP","PORT","USERNAME","OLDPASS","NEWPASS","FLAG"]    for col in range(0,6):      ws.write(0,col,header[col])    for row in range(len(serverlist)):      for col in range(0,6):        ws.write(row+1,col,serverlist[row][col])    dt = time.strftime("%Y%m%d%H%M%S", time.localtime())    wb.save(dt+".xlsx")  def get_server_list(self):    return self.serverlist  def get_rows(self):    return self.rows  def get_cols(self):    return self.cols  def gen_key(self):    pool = "1234567890abcdefghijklmnopqrstuvwxyzQWERTYUIOPASDFGHJKLZXCVBNM"    length = len(pool)    key = ""    for i in range(28):      c = random.randint(0,length)      key += pool[c:c+1]    return key

#_*_ coding: utf-8 _*_'''@author liaogs'''import paramikoimport sysclass ChangePassword():  def __init__(self,hostip,port,username,oldpass,newpass):    self.hostip = hostip    self.port = port    self.username = username    self.oldpass = oldpass    self.newpass = newpass    self.updateflag = False  def run_change(self):    s = paramiko.SSHClient()    s.set_missing_host_key_policy(paramiko.AutoAddPolicy())    tasklist = []    try:      s.connect(hostname=self.hostip, port=self.port, username=self.username, password=self.oldpass)      print ('"%s" is updating password' % self.hostip)      stdin, stdout, stderr = s.exec_command('echo %s |passwd --stdin root' % self.newpass)      r_message = stdout.read()      if "successfully" in r_message:        self.updateflag = True        print('%s is successful' %self.hostip)      else:        print('%s is failed' %self.hostip)        self.updateflag = False      s.close()    except Exception:      self.updateflag = False      print("connection error")    tasklist = [self.hostip, self.port, self.username, self.oldpass, self.newpass, self.updateflag]    return tasklist

#_*_ coding: utf-8 _*_'''@author liaogs'''import reimport sysfrom excelhandler import excelhandlerfrom changepassword import ChangePasswordif __name__ == '__main__':  if len(sys.argv) == 1:    eh = excelhandler("pass.xlsx")  else:    eh = excelhandler(sys.argv[1])  eh.read_excel()  def updatepassword():    ret = eh.get_server_list()    tasklist = []    for i in range(len(ret)):      print(ret[i][0],ret[i][2],ret[i][1],ret[i][3],ret[i][4])      cp = ChangePassword(hostip=ret[i][0],port=int(ret[i][2]),username=ret[i][1],oldpass=ret[i][3],newpass=ret[i][4])      task = cp.run_change()      tasklist.append(task)    print(tasklist)    eh.write_excel(tasklist)  while True:    inp = input("1、生成密码 2、更新密码>>")    if str(inp) == "1":      eh.gen_new_password_excel()    elif str(inp) == "2":      updatepassword()    elif inp == "exit":      exit()    else:      continue