**包含兩個主要功能**
1. SshTest類，用於探測ssh是否可以連線以及執行ssh命令。
2. PushDing，用於推送釘釘訊息

**本文包含以下三個部分**
1. python指令碼正文
2. 只探測ssh是否可連線樣例模板。
3. 探測ssh並執行作業系統命令以及推送釘釘訊息樣例模板。

# 1. python指令碼正文
> 下面指令碼為通用模板，可直接使用

```python
#!/bin/python3
# 本模組兩個包含兩個通用功能：
"""
1.SshTest類，用於探測ssh是否可以連線以及執行ssh命令。
2.PushDing，用於推送釘釘訊息
"""

# python內建模組
import re
import datetime
import time
import os
import hmac
import hashlib
import urllib
import base64
import requests
import json
# paramiko需要下載模組
import paramiko

"""
一、SshTest類：
1.探測ssh是否可達
2.check_ssh_status方法，用於探測ssh是否可以連線，無法連線將報錯，返回三個標誌值：
    1.send_msg，顯示主機、探測時間、以及ssh無法連線資訊
    2.warn_type，告警型別，顯示ssh無法連線
    3.sshstat，ssh是否可連線標誌位，為boolean值，true為可連線，false為不可連線
3.exec_ssh_cmd用於ssh探測成功之後，執行ssh命令。
"""
class SshTest:
    def __init__(self, host, port, username, pwd):
        self.host = host
        self.port = port
        self.username = username
        self.pwd = pwd
        self.ssh = paramiko.SSHClient()
        self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.timenow = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') + ' '
        try:
            self.ssh.connect(self.host, self.port, self.username, self.pwd, timeout=5)
            self.sshinvalid = False
            # print(self.sshinvalid)
        except Exception as errmsg:
            self.sshinvalid = True
            self.errormsg = errmsg
            # print(self.sshinvalid)

def check_ssh_status(self):
        if self.sshinvalid:
            send_msg = ('**告警時間：** ' + self.timenow  + ' \\\n **告警主機：** ' + str(self.host) + ' \\\n **告警內容：** ' + str(self.errormsg) + ' ')
            warn_type='主機SSH無法連線'
            sshstat = False
            # 如遇到無法ssh連線，報錯詳細資訊
            # print(self.errormsg)
            print('ssh連線報錯為: ' + str(self.errormsg))
        else:
            send_msg = ('**告警時間：** ' + self.timenow  + ' \\\n **告警主機：** ' + str(self.host) + ' \\\n **告警內容：** ')
            warn_type='主機SSH連線正常'
            sshstat = True
            print(warn_type)
        return send_msg,warn_type,sshstat

def exec_ssh_cmd(self,cmd):
        if self.sshinvalid:
            i = 'SSH無法連線，請檢查主機'
            # print(i)
            cmd_out = []
            cmd_out.append(i)
        else:
            stdin, stdout, stderr = self.ssh.exec_command(cmd)
            # 定義一個空列表，後續用於儲存命令輸出
            cmd_out = []
            for i in stderr.readlines():
                print('執行命令錯誤為： ' + i)
                cmd_out.append(i)
            # print(stdout.readlines)
            for i in stdout.readlines():
                print('命令執行成功，輸出為： ' + i)
                cmd_out.append(i)
        # 如果有多行，則下面i返回的是最後一行資料，一定注意
        return cmd_out

"""
推送釘釘類：
1.初始化例項傳入釘釘機器人webhook以及secret即可。
2.傳送訊息，傳入需要傳送的訊息以及告警型別字串即可，傳入訊息為markdown格式。
"""
class PushDing:
    def __init__(self,webhook,secret):
        self.webhook = webhook
        self.secret = secret

# 釘釘傳送函式，輸入markdown格式訊息文字即可傳送
    def send_message(self,send_msg,warn_type):
        ### 釘釘配置 begin ###                                                                                                                              
        # 如果是內網，需要有網際網路代理配置                                                                                                                  
        # proxies = {'http':'http://IP:PORT','https':'http://IP:PORT'}                                                                                      
        # os.environ["https_proxy"] = "http://IP:PORT"                                                                                            
        webhook = self.webhook                        
        timestamp = str(round(time.time() * 1000))                                                                                                          
        secret = self.secret                                                                     
        secret_enc = secret.encode('utf-8')                                                                                                                 
        string_to_sign = '{}\n{}'.format(timestamp, secret)                                                                                                 
        string_to_sign_enc = string_to_sign.encode('utf-8')                                                                                                 
        hmac_code = hmac.new(secret_enc, string_to_sign_enc, digestmod=hashlib.sha256).digest()                                                             
        sign = urllib.parse.quote_plus(base64.b64encode(hmac_code))                                                                                         
        webhook = webhook + '×tamp=' + timestamp + '&sign=' + sign
        # timestamp 與系統當前時間戳如果相差1小時以上，則認為是非法的請求，所以需要注意傳送訊息伺服器時間不能與標準時間相差超過一小時                                                                         
        headers = {'Content-Type': 'application/json'}                                                                                                      
        user = "all"                                                                                                                                        
        ### 釘釘配置 end ### 
        data={                                                                                                                                          
            "msgtype": "markdown",                                                                                                                      
            "markdown": {                                                                                                                               
                "title": f"{warn_type}告警",                                                                                                                  
                "text": f"# <font face="\"華雲彩繪\"" color="\"#FF0000\"" size="\"10\"">{warn_type}</font> \n " + send_msg                                      
            },                                                                                                                                          
            "at": {                                                                                                                                     
                "atMobiles": [                                                                                                                          
                    user                                                                                                                                
                ],                                                                                                                                      
                "isAtAll": False                                                                                                                        
            }                                                                                                                                           
        }                                                                                                                                               
        #print(data)                                                                                                                                    
        x=requests.post(url=webhook,data=json.dumps(data),headers=headers)                                                                              
        # 列印傳送釘釘結果                                                                                                                              
        print('釘釘推送結果為： ' + str(x.json()) + '\n')
```

# 2. 樣例1，只ssh探測主機
```python
#!/bin/python3
# 本指令碼用於批次探測主機ssh是否可以連線，無法連線反饋結果以及推送釘釘機器人訊息。

# 匯入內建模組
import datetime
# 匯入自定義模組
import sshtest

"""
變數賦值區域
1. hostlist填寫自己的主機IP登入使用者名稱密碼即可。
"""
hostlist=[
{'host': '192.168.xxx.xxx', 'port': '22', 'user': 'oracle', 'pwd': 'xxxxxx'},
{'host': '192.168.xxx.xxx', 'port': '22', 'user': 'oracle', 'pwd': 'xxxxxx'}
]

"""
執行命令並推送釘釘機器人
"""
for con_info in hostlist:
    host = con_info['host']
    port = con_info['port']
    username = con_info['user']
    pwd = con_info['pwd']
    # 列印當前時間以及探測IP
    print(datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') + ' Host IP : ' + str(host))
    # 初始化例項
    ssh = sshtest.SshTest(host, port, username, pwd)
    """
    # 檢查ssh狀態,返回的資料為tuple，
    第一個元素為ssh探測後推送訊息正文
    第二個為ssh探測狀態
    第三個為ssh探測標誌位，true為ssh正常，false為無法連線
    check_ssh_status()方法會自動列印ssh連線報錯資訊，比如密碼不正確，22埠無法連線等
    """
    ssh_return = ssh.check_ssh_status()
    print(str(host) + '主機SSH狀態為： ' + str(ssh_return[2]) + '\n')
    ssh.ssh.close()
```

# 3 樣例2，探測主機以及執行命令並推送釘釘機器人訊息
```python
#!/bin/python3
# 本指令碼用於批次探測主機ssh是否可以連線，無法連線反饋結果以及推送釘釘機器人訊息。

# 匯入內建模組
import datetime
# 匯入自定義模組
import sshtest

"""
變數賦值區域
1. hostlist填寫自己的主機IP登入使用者名稱密碼即可。
2. cmd填寫需要ssh執行的命令
3. webhook，釘釘機器人地址。
4. secret，釘釘機器人金鑰
"""
hostlist=[
{'host': '192.168.xxx.xxx', 'port': '22', 'user': 'oracle', 'pwd': 'xxxxxx'},
{'host': '192.168.xxx.xxx', 'port': '22', 'user': 'oracle', 'pwd': 'xxxxxx'}
]
cmd = '你需要執行的SSH命令'
webhook = '你的釘釘機器人webhook地址'
secret = '你的釘釘機器人金鑰'

"""
執行命令並推送釘釘機器人
"""
for con_info in hostlist:
    host = con_info['host']
    port = con_info['port']
    username = con_info['user']
    pwd = con_info['pwd']
    print(str(datetime.datetime.now()) + ' Host IP : ' + str(host))
    print('執行ssh探測以及ssh命令')
    # 初始化例項
    ssh = sshtest.SshTest(host, port, username, pwd)
    """
    # 檢查ssh狀態,返回的資料為tuple，
    第一個元素為ssh探測後推送訊息正文
    第二個為ssh探測狀態
    第三個為ssh探測標誌位，true為ssh正常，false為無法連線
    """
    ssh_return = ssh.check_ssh_status()
    # print(ssh_return[2])
    # 當ssh可連線，則執行命令，否則不執行
    if ssh_return[2]:
        cmd_result = ssh.exec_ssh_cmd(cmd)
        print(cmd_result)
        s = "\\\n" .join(cmd_result)
        # 執行完畢，關閉ssh連線
    else:
        s = '主機無法連線，命令未執行.'
    ssh.ssh.close()
    # print(cmd_result)
    pushding = sshtest.PushDing(webhook,secret)
    pushding.send_message(ssh_return[0] + s ,ssh_return[1])

```

Python執行SSH命令通用模板SSH連線狀態探測以及釘釘推送

相關文章