#!/usr/bin/env python #-*- encoding:utf8 -*- #--------------------------------------------------------------------------------------- # FILE: sshd.py # USAGE: sshd.py # DESCRIPTION: Copy python style guide and coding standard . # The default copy example is the current text . # Dont.t find text on other directories . # OPTIONS: see fuction 'usage' below # BUGS: --- # AUTHOR: Dr.-Vision. Simple(sp) # VERSION: 1.0 # CREATED: 08.18.2014 - 10:15:30 # REVISION: 12.02.2014 # PROJECT: XXX # COPYRITHT: Copyright(c)2002-2014 Python, All Rights Reserved #--------------------------------------------------------------------------------------- # import python module #--------------------------------------------------------------------------------------- # define python import modules . #--------------------------------------------------------------------------------------- import os import sys sys.path.append('../module') from zgtz import idsTempDic,showServerStart,showServerEnd import socket import time import pexpect import paramiko import getpass # Python Function comments #======== Function ==================================================================== # NAME: # DESCRIPTION: Display usage information for this script. # PARAMETER 1: --- #======================================================================================= class SSH(object): user = "用户名" port = "端口" passwd = "明文密码" cmd = "命令行" local_path = "本地路径" remote_path = "远程路径" known_host = "host文件" public_rsa_file = "RSA公钥文件" private_rsa_file = "RSA私钥文件" public_dsa_file = "DSA公钥文件" private_dsa_file = "DSA私钥文件" auth_key_file = "authorized_keys" rsa_cmd = "RSA keygen" dsa_cmd = "DSA keygen" record_log = 'ssh日志文件' t1 = "函数启动时间" t2 = "函数结束时间" def __init__(self): self.user = "root" self.port = 22 self.passwd = "" self.cmd = "ls ./tmp" self.local_path = os.environ['HOME']+'/.ssh/' self.remote_path = os.environ['HOME']+'/.ssh/' self.known_host = "known_hosts" self.public_dsa_file = "id_dsa.pub" self.private_dsa_file = "id_dsa" self.public_rsa_file = "id_rsa.pub" self.private_rsa_file = "id_rsa" self.auth_key_file = "authorized_keys" self.rsa_cmd = "/usr/bin/ssh-keygen -t rsa " self.dsa_cmd = "/usr/bin/ssh-keygen -t dsa " self.record_log = "./tmp/paramiko.log" self.t1 = time.time() def key_rw(self,host): self.host = host if os.path.exists(self.local_path+self.auth_key_file) and os.path.getsize(self.local_path+self.auth_key_file) != 0 : print "%sFileName ===%10s\tCVM_Address === %s\t[ CVM 跳板机 Is Exists ! ] %s" %(rx,self.auth_key_file,self.host,ry) all = [] os.chdir(self.local_path) Fname= open(self.auth_key_file,'r') for line in Fname: #print line all.append(line) return all Fname.close() else: try: all = [] ls = os.linesep rdsa=['rsa', 'dsa'] #print self.local_path os.chdir(self.local_path) try: for i in rdsa: FName = open('id_%s.pub'%(str(i))) for line in FName: #print line.rstrip('\n'), all.append(line) F_KEY = open(self.local_path+self.auth_key_file,'w') F_KEY.writelines(['%s%s'% (x,ls)for x in all]) print "%sFileName ===%10s\tCVM_Address === %s\t[ CVM 跳板机 NoKey OK ] %s" %(rx,self.auth_key_file,self.host,ry) return all FName.close() F_KEY.close() except IOError ,e: print "*** file open failed:",e except: print "Open %s file failed !" %(self.auth_key_file) raise def rsync_ssh(self,host): self.host = host try: scp = paramiko.Transport((self.host,self.port)) scp.connect(username=self.user,password=self.passwd) sftp = paramiko.SFTPClient.from_transport(scp) except Exception,e: print e scp.close() else: try: sftp.mkdir(self.remote_path) except IOError: #print('(assuming ~/.ssh already exists)') with sftp.open(self.remote_path+self.auth_key_file,'a+') as f: f.write("#Create SSH Address == %s\n"%(self.host)) with open(self.local_path+self.public_rsa_file,'r') as f: data = f.read() sftp.open(self.remote_path+self.auth_key_file,'a+').write(data) sftp.open(self.remote_path+self.auth_key_file,'a').write('\n') print '%sFileName ===%s\tCVM_Address === %s\t[ Finish NoKey OK ]%s' %(rx,self.auth_key_file,self.host,ry) else: with sftp.open(self.remote_path+self.auth_key_file, 'w') as f: f.write("#This Is NoKey Record .\n") with open(self.local_path+self.public_rsa_file,'r') as f: data = f.read() sftp.open(self.remote_path+self.auth_key_file,'w').write(data) print '%sFileName ===%s\tCVM_Address === %s\t[ Finish NoKey OK ]%s' %(rx,self.auth_key_file,self.host,ry) if __name__ == '__main__': #格式化打印 star="\033[1;33;40m=\033[0m" rx="\033[1;33;40m" ry="\033[0m" showStart = showServerStart() temp_data = idsTempDic() ids_sorted = sorted(temp_data.items(),key=lambda d:int(d[0])) for k,v in ids_sorted: print '\t'.join(v) showEnd = showServerEnd() while True: ids = [] errids = [] try: inputs=str(raw_input('%s请输入CVM_ID[多个请用空格隔开,全部(all),反选("-"开头),返回(b)]:%s'%(rx,ry)).strip().lower()) print inputs #inputs=raw_input('请输入CVM_ID[多个请用空格隔开,全部(all),反选("-"开头),返回(b)]:').strip().lower() if inputs == "b": print '%s您没有选择,返回主界面! %s'%(rx,ry) raw_input() continue elif inputs == "all": ids = temp_data.keys() # 此处返回字典key,类型为list类型 #print ids break elif inputs.startswith('-'): ex = inputs[1:].split() #print ''.join(ex) all = temp_data.keys() ids = [x for x in all if x not in ex] #反选后的CVM_ID exids = [x for x in ex if x not in all] #反选CVM_ID break else: ids = inputs.split() ids=sorted(set(ids),key=ids.index) for id in ids: running = temp_data.has_key(id) if running == False: ids.remove(id) errids.append(id) print ids elif running == True: print "%s输入正确的CVM_ID:%s %s"%(rx,id,ry) else: raise ValueError break except (NameError,KeyboardInterrupt): print "\n" sys.exit() except (ValueError,KeyError): print "%s 无此CVM_ID,输入错误,请重新输入! %s"%(rx,ry) raw_input() if not ids: print "%s你输入的CVM_ID有误%s"%(rx,ry) if inputs == "all": print "%s选择所有服务器 %s"%(rx,ry) elif inputs.startswith('-'): #print exids print "\033[1;33;40m去除的id为:%s\033[0m "%",".join(ex) print "\033[1;33;40m反选后id为:\033[0m", for gid in ids: name = temp_data[gid][0] if name: print name+' ', print '\t' else: print "\033[1;33;40m去除的id为:%s\033[0m "%",".join(ids) print "对应的游戏服为:", for gid in ids: name = temp_data[gid][0] if name: print name+' ', if errids: print "%s你输入的id部分有误,已经排除:%s"%(rx,ry) print '\t'.join(errids) try: if raw_input("\n确认?(y/n)") == "y": print "%s更新的CVM_ID为%s:"%(rx,ry)+' '.join(ids) sshdir ="~/.ssh" if os.path.exists(sshdir): pass else: for gid in ids: Host_Wan = temp_data[gid][3] Host_Lan = temp_data[gid][4] print Host_Wan,Host_Lan if Host_Lan == "10.1.1.1": SSH2 = SSH() #SSH2.key_rw(Host_Wan) #SSH2.rsync_ssh(Host_Lan) print "%s跳过运维跳板机%s" else: try: SSH2 = SSH() SSH2.rsync_ssh(Host_Lan) SSH2.known_hosts(Host_Lan) SSH2.known_hosts(Host_Wan) except Exception as e: print e print "%s所选CVM_ID %s已经更新完成! %s"%(rx,ids,ry) else: sys.exit() except (KeyboardInterrupt,EOFError): print "\n" sys.exit()创新互联公司专注于企业全网营销推广、网站重做改版、东丽网站定制设计、自适应品牌网站建设、H5场景定制、成都商城网站开发、集团公司官网建设、外贸营销网站建设、高端网站制作、响应式网页设计等建站业务,价格优惠性价比高,为东丽等各大城市提供网站开发制作服务。
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。