Python——SSHClient.py

自己封裝好的遠端連線的元件，後期會不斷完善

1、為了更方便採集資訊系統以及資料庫的資訊，我做了該元件

2、為了讓語句執行更順暢，位置不發生錯亂，暫時採用time.sleep()的方式解決

3、對於第二點會想辦法將方法完善，以希望大家多多提出寶貴建議

#!/usr/bin/env python
# coding:utf-8
'''
@author: Ryan Bai(白瑞鈞)
@license:
@contact: brj880719@hotmail.com
@file: SSHClient.py
@create time: 2017/11/8 18:11
@desc:
'''
import paramiko
from paramiko.py3compat import u
import time
'''
@attention: : ssh客戶端使用
@author: 白瑞鈞
@param date:
'''
class SSHClient(object):
'''
@attention: 關閉 ssh 連結
@author: 白瑞鈞
@param ssh: ssh連結
'''
def close(self, ssh):
ssh.close()
'''
@attention: 建立 ssh 連結
@author: 白瑞鈞
@param v_username: 使用者名稱
@param v_password: 密碼
@param v_ip: IP
@param v_port: 埠號
'''
def sshConnection(self, v_username, v_password, v_ip, v_port=22):
# 建立SSH物件
ssh = paramiko.SSHClient()
# 把要連線的機器新增到known_hosts檔案中
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 連線伺服器
ssh.connect(hostname=v_ip, port=v_port, username=v_username, password=v_password)
return ssh
# endregion
'''
@attention: 執行單條命令
@author: 白瑞鈞
@param ssh: ssh連結
@param v_cmd: 需要執行的命令
'''
def sshExecByOne(self, ssh, v_cmd):
# 執行
stdin, stdout, stderr = ssh.exec_command(v_cmd)
result = stdout.read()
if not result:
result = stderr.read()
return result.decode()
'''
@attention: 執行命令集
@author: 白瑞鈞
@param ssh: ssh連結
@param l_cmd: 需要執行的命令集
@param exec_wait: 執行命令間隔時間
@param exit_wait: 退出等待時間
'''
def sshExecByMany(self, s, l_cmd, exec_wait, exit_wait):
ssh = s.invoke_shell()
# 執行
for v_cmd in l_cmd:
ssh.send(v_cmd)
ssh.send('\n')
time.sleep(exec_wait)
if v_cmd=='exit':
time.sleep(exit_wait)
result = u(ssh.recv(9999))
return result
if __name__ == '__main__':
getClient = SSHClient()
ssh = getClient.sshConnection('sys_admin', 'XSW@1qaz', '10.82.28.219')
l_cmd = ['sudo su - ',
'su - oracle',
'sqlplus / as sysdba',
u'select * from dual;',
'exit',
'df -h',
'exit']
result = getClient.sshExecByMany(ssh, l_cmd, 1, 1)
print(result)
getClient.close(ssh)
# getClient = SSHClient()
# ssh = getClient.sshConnection('sys_admin', 'XSW@1qaz', '10.82.28.219')
# result = getClient.sshExecByOne(ssh,'pwd')
# print(result)
# getClient.close(ssh)

來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/24851054/viewspace-2147009/，如需轉載，請註明出處，否則將追究法律責任。

Python——SSHClient.py

相關文章