python使用ssh连接linux服务器切换到root的问题

小邪

1. #!/usr/env/bin python
   
2. # -*- coding:utf-8 -*-
   
3. # des:ssh登录系统，如果是非root帐号(普通帐号)登录采取行动，则su切换到root
   
4. 
   
5. # python 的SSHV2 protocol协议实现，用于python调用各种ssh功能;
   
6. import paramiko
   
7. # python内置时间模块
   
8. import time
   
9. 
   
10. #根据用户名密码从普通用户揽权到root用记执行给定命令
    
11. 
    
12. def ssh_su_root(ip,username,password,root_pwd,cmd):
    
13. #建立SSHClient对象
    
14.     myssh = paramiko.SSHClient()
    
15. #设置自动接收公钥
    
16.     myssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    
17.     try:
    
18. #使用给定用户名密码尝试连接服务器
    
19.          myssh.connect(ip, 22, username=username, password=password, timeout=4)
    
20.     except Exception as ex:
    
21.         print (str(ex))
    
22.         return
    
23.     # 如果默认root不允许远程登录，提供的是普通用户帐号+root帐号,
    
24.     #则调用invoke_shell,使用交互式shell方式切换同到root帐号，并执行命令。
    
25.     if username != 'root':
    
26.         print ("I am NOT root")
    
27.         ssh = myssh.invoke_shell()
    
28.         time.sleep(0.2)
    
29.         ssh.send('su -\n')
    
30.         buff = ''
    
31.     #循环检测密码输入提示，出现后发送root口令，
    
32.         while not (buff.endswith('Password:') or buff.endswith('密码: ')):
    
33.     #从接收缓存中读入最多9999bytes数据。
    
34.             resp = ssh.recv(9999)
    
35.             buff += resp
    
36.         ssh.send(root_pwd)
    
37.         ssh.send('\n')
    
38.         buff = ''
    
39.     #循环检测#提示符，出现后执行传入的命令。
    
40.         while not buff.endwith('# '):
    
41.             resp = ssh.recv(9999)
    
42.             buff += resp
    
43.         ssh.send[cmd]
    
44.         ssh.send('\n')
    
45.         
    
46.         buff =''
    
47.     #循环检测#提示符，意味着命令执行完成，出现后关闭ssh连接并输出结果。
    
48.         while not buff.endswith('# '):
    
49.             resp = ssh.recv(9999)
    
50.             buff += resp
    
51.         myssh.close()
    
52.         result = buff
    
53.         print ("su root result --->",result)
    
54. 
    
55.         myssh.close()
    
56.     else:
    
57.     #如果提供的是root帐号，则直接调用paramiko exec_command 方法，执行命令。
    
58.         
    
59.         print ("I am root")
    
60.         stdin, stdout, stderr = myssh.exec_command(cmd)
    
61.         stdin.write("Y")
    
62.         print ("stdout.readlines() --->", stdout.readlines())
    
63.         myssh.close()
    
64. 
    
65. if __name__ == "__main__":
    
66.     cmd = 'date'
    
67.     ip = '172.16.10.12'
    
68.     username = 'sysadmin'
    
69.     password = 'Inspur1!'
    
70.     ssh_su_root(ip, username, password, 'xxxxxxxxxxxx5', cmd)
    
71.     username = 'root'
    
72.     ssh_su_root(ip, username, password, 'xxxxxxxxxxxx', cmd)

复制代码

suchocolate

报错不全，尽量把报错信息贴全。

1）试试改成这样：buff += resp.decode()
3）另外第43行 ssh.send[cmd] 是不是写错了？

小邪

suchocolate 发表于 2021-11-12 14:34
报错不全，尽量把报错信息贴全。

1）试试改成这样：buff += resp.decode()

报错信息贴全了，就这些报错信息。改成resp.decode()我也试了，不会报错。但是会一直卡在不动的页面