使用prarmiko模块切换root执行命令VIP专享VIP免费

Paramiko模块是非常优秀的ssh连接库。通常使用paramiko.SSHClient类型的exec_command来执行命令，返回一个包含了代表标准输入、标准输出、标准错误的三个元素的元祖。通过操作这三个ChannelFile对象可以获取到命令的标准输出以及命令的退出状态，并且如果命令没有执行完成，获取标注输出或退出状态时，进程将被阻塞，直到命令执行完成，还可以使用标准输入完成简单的交互操作，具体的使用方式，参见Paramiko使用exec_command很方便，但也有不足之处，很多linux环境都是禁止root直接登录的，而使用exec_command无法完成su–root这种交互方式的切换命令，执行时进程将被永久阻塞。invoke_shell方法，用于创建一个子shell进程，这样所有的操作都可以在该子shell中进程，su切换用户操作不受影响，但该方法没有exec_command那种方便的ChannelFile对象，所有的标准输出和标准错误内容通过invoke_shell返回对象的recv方法来获取，每一次调用recv只会从上一次返回的地方开始返回，也没有直接获取命令退出状态的方法，不过这些缺点可以通过代码来实现。importparamikoimporttimedef_shell_exec(shell,cmd,sleep,is_exec=True):ifis_exec:#执行shell命令shell.send(cmd+'\n')time.sleep(sleep)#recv方法返回收集到的shell命令的标准输出和标准错误。#参数为返回多少个自己节数的标准输出和标准错误，值应该大于实际的返回，这样才能保证输出的完整stdout=shell.recv(1024*100)#按行分割得到的信息out_list=stdout.decode().split('\n')returnout_listdefmain():ssh=paramiko.SSHClient()ssh.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())ssh.connect('192.168.1.28',22,'test','huawei')#invoke_shell方法，创建一个子shell，所有的命令都在该shell中进行。shell=ssh.invoke_shell()print('Begintosuroot...')out_list=_shell_exec(shell,'su-root',1)last_str=out_list[len(out_list)-1].strip()#获取的linux系统的标准输出和标准错误按行分割后#第一行内容为传递的linux命令，最后一行在无交互时，内容应该为PS1变量的值#否则应该是交互的输出信息，切换root为交互操作，最后一行的内容应该为"Password:"ifnotlast_str.endswith('Password:'):raiseValueError("Failedtoexecsuroot")out_list=_shell_exec(shell,'huawei',5)result_str=out_list[len(out_list)-2].strip()#输入密码后，交互完成，此时最后一行应该为PS1变量的值#单数第二行如何出现failure内容，则表示root密码输入错误，切换失败ifresult_str.endswith('failure'):raiseValueError("Authenticationfailure,pleasecheckroot'spassword")print('Successfully')cmd_list=list()#第一条命令是更改PS1变量，即标识符的样式#每条命令执行完成后，都会返回到标识符，通过判断命令输出的最后是否等于PS1即可判断命令是否执行完成cmd_list.append('exportPS1="[\\u]#";echo$?')cmd_list.append('whoami;echo$?')cmd_list.append('pwd;echo$?')cmd_list.append('cd/opt;bashtest.sh;echo$?')cmd_list.append('cd/opt;bashtest2.sh;echo$?')cmd_list.append('pwd;echo$?')#逐条执行数组中的命令print('Begintoexeccommand...')forcmdincmd_list:print('"{0}"...'.format(cmd))out_list=list()#声明一个列表变量，然后将shell返回的标准输出和标准错误分割后的列表元素添加到列表#这样做是为了得到所有的标准输出和标准错误，#因为每一次执行recv，返回的内容将从上次一次获取后的地方开始，得到的标准输出和标准错误将不完整out_list.extend(_shell_exec(shell,cmd,1))last_str=out_list[len(out_list)-1].strip()#通过循环，判断返回的列表最后一个元素是否等于为PS1变量，来判断命令是否执行完成whilenotlast_str.endswith('[root]#'):print('"{0}"isnotfinsh,pleasewriting...'.format(cmd))out_list.extend(_shell_exec(shell,cmd,1,False))last_str=out_list[len(out_list)-1].strip()result_str=out_list[len(out_list)-2].strip()ifresult_str!='0':print('Failedtoexec"{0}",errormessageisasfollows'.format(cmd))#输出列表中，除去第一个和最后个即为命令的标准输入和标准错误foriinout_list[1:len(out_list)-1]:ifi!='':print('{0}'.format(i))else:print('"{0}"...done'.format(cmd))ssh.close()if__name__=='__main__':main()