服务端测试环境hosts配置检查脚本

#!/usr/bin/python
# -*- coding:utf-8 -
B_server_ip_1 = "B env ip 1"
B_server_ip_2 = "B env ip 2"
tip_1 = "isXXX"
tip_2 = "isnotXXX"

with open(‘/etc/hosts‘) as f:
    for line in f.readlines():
        rule1 = "B env name" in line
        rule2 = not line.startswith("#")
        rule3 = B_server_ip_1 in line
        rule4 = B_server_ip_2 in line
        if rule1 and rule2:
            if rule3:
                print(tip_1)
            if rule4:
                print(tip_2)

# coding:utf-8
import base64
import paramiko

class SSH_Handler:
    client = paramiko.SSHClient()
    def __init__(self,ssh_key_path,server_ip,server_port,username):
        ssh_key = paramiko.RSAKey.from_private_key_file(ssh_key_path)
        self.client.load_system_host_keys()
        self.client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.client.connect(server_ip,port=server_port,username=username,pkey=ssh_key)
    def exec_command(self,command):
        stdin, stdout, stderr = self.client.exec_command(command)
        ret = []
        for line in stdout:
            ret.append(line.strip())
        return ret
    def __del__(self):
        self.client.close()

‘‘‘
    ssh_key_path：OpenSSH格式的用来连接服务器的私钥路径
    server_ip：要连接的服务器的ip地址
    server_port：要连接的服务器的端口
    username：登录服务器的用户名
    command：在服务器端要执行的命令
‘‘‘
if __name__ == ‘__main__‘:
    ssh_key_path = ‘E:\\sshKey‘
    server_ips = [‘A env ip 1‘,‘A env ip 2‘,‘A env ip 3‘]
    server_port = xxxx
    username = ‘username‘
    command = ‘python hosts_check.py‘
    results = []
    s = set()
    for server_ip in server_ips:
        result = SSH_Handler(ssh_key_path,server_ip,server_port,username).exec_command(command)[0]
        results.append(server_ip+":"+result)
        s.add(result)
    if len(s) ==1:
        print(s)
    else:
        print(results)