Python封裝(01)：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
@attention: ssh客戶端使用
@desc:
'''
import paramiko
from paramiko.py3compat import u
import time
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 s: 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)