python自动化之admin自动切换root

[toc]

python自动化之admin自动切换root

ssh登录系统，如果是非root帐号(普通帐号)登录采取行动，则su切换到root

#!/usr/env/bin python
# -*- coding:utf-8 -*-
# des:ssh登录系统，如果是非root帐号(普通帐号)登录采取行动，则su切换到root

# python 的SSHV2 protocol协议实现，用于python调用各种ssh功能;
import paramiko 
# python内置时间模块
import time 

#根据用户名密码从普通用户揽权到root用记执行给定命令

def ssh_su_root(ip,username,password,root_pwd,cmd):
#建立SSHClient对象
    myssh = paramiko.SSHClient()
#设置自动接收公钥
    myssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    try:
#使用给定用户名密码尝试连接服务器
         myssh.connect(ip, 22, username=username, password=password, timeout=4)
    except Exception, ex:
        print str(ex)
        return
    # 如果默认root不允许远程登录，提供的是普通用户帐号+root帐号,
    #则调用invoke_shell,使用交互式shell方式切换同到root帐号，并执行命令。
    if username != 'root':
        print "I am NOT root"
        ssh = myssh.invoke_shell()
        time.sleep(0.2)
        ssh.send('su -\n')
        buff = ''
    #循环检测密码输入提示，出现后发送root口令，
        while not (buff.endswith('Password:') or buff.endswith('密码: ')):
    #从接收缓存中读入最多9999bytes数据。
            resp = ssh.recv(9999)
            buff += resp 
        ssh.send(root_pwd)
        ssh.send('\n')
        buff = ''
    #循环检测#提示符，出现后执行传入的命令。
        while not buff.endwith('# '):
            resp = ssh.recv(9999)
            buff += resp 
        ssh.send[cmd]
        ssh.send('\n')

        buff =''
    #循环检测#提示符，意味着命令执行完成，出现后关闭ssh连接并输出结果。
        while not buff.endswith('# '):
            resp = ssh.recv(9999)
            buff += resp
        myssh.close()
        result = buff
        print "su root result --->",result 

        myssh.close()
    else:
    #如果提供的是root帐号，则直接调用paramiko exec_command 方法，执行命令。

        print "I am root"
        stdin, stdout, stderr = myssh.exec_command(cmd)
        stdin,write("Y")
        print "stdout.readlines() --->", stdout.readlines()
        myssh.close()

if __name__ == "__main__":
    cmd = 'date'
    ip = 'xx.xxx.xx.xxx'
    username = 'admin'
    password = 'xxxxxxxxxx'
    ssh_su_root(ip, username, password, 'xxxxxxxxxxxx', cmd)
    username = 'root'
    ssh_su_root(ip, username, password, 'xxxxxxxxxxxx', cmd)