使用Python登录网关设备执行命令

古城微笑少年丶 2023-02-16 06:43 186阅读 0赞

# 一、概述 #

最近发现办公室有一个感测器网关，偶尔会掉线，重启之后，就正常了。

手动操作过程如下：

# telnet 10.212.82.90
    Trying 10.212.82.90...
    Connected to 10.212.82.90.
    Escape character is '^]'.
    (none) login: root
    Password: 
    Please make sure that what you're doing!  If you aren't administrator, please Don't do anything! :)
    # ps
      PID USER       VSZ STAT COMMAND
        1 root      1124 S    init       
        2 root         0 SW   [kthreadd]
        3 root         0 SW   [ksoftirqd/0]
        4 root      3888 S    /usr/app/bin/app

说明：先telnet进去，输入用户名和密码之后，执行ps命令。如果有/usr/app/bin/app进程，说明设备正常，否则就需要重启。

但是手动操作比较麻烦，需要使用python登录网关设备，检测是否正常。

# 二、Telnet对象 #

Telnet.read\_until(expected, timeout=None) ： \#读取连接服务器后显示的内容，直到遇到同expected相同的字节串。或者等待时间大于timeout时直接向下运行。

Telnet.read\_very\_eager() ： 读取从上次IO阻断到现在所有的内容，返回的是字节串，需要进行decode()编码。如果连接关闭或者没有可用数据时会抛出EOFError，如果没有其他可用的数据，返回的是b""，除非在IAC中间，否则不会阻碍。

Telnet.open(host, port=23\[, timeout\]) ： 连接到主机，端口号为第二个可选参数，默认为标准的Telnet端口(23)，可选的timeout参数指定连接的超时时间，如果未指定，将使用全局默认超时设置。不要尝试去重新打开一个已经连接的实例对象。

Telnet.close() ： 关闭连接。

Telnet.write(buffer) ： \# 将一个字节串(byte string)写进socket，如果连接被阻塞，这也会被阻塞，如果连接关闭，会抛出OSError。

Telnet.interact() ： telnet的交互功能，下面用了一个死循环保证用户能够一直输入命令进行某些操作，也可以使用Telnet.interact()这个方法来使所连接终端持久化，不过官网说 (emulates a very dumb Telnet client)直译是一个非常愚蠢的客户端。

官方文档，请参考：

[https://docs.python.org/zh-cn/3/library/telnetlib.html][https_docs.python.org_zh-cn_3_library_telnetlib.html]

# 三、检测脚本 #

check\_gateway.py

![aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9Db250cmFjdGVkQmxvY2suZ2lm][] ![aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9FeHBhbmRlZEJsb2NrU3RhcnQuZ2lm][]

#!/usr/bin/env python3
    # coding: utf-8
    import telnetlib
    import time
    
    class TelnetClient(object):
        def __init__(self, ip, user, pswd):
            self.tn = telnetlib.Telnet()
            self.host_ip = ip
            self.username = user
            self.password = pswd
            self.last_res = ''  # 记录上次命令执行结果
    
        # 此函数实现telnet登录主机
        def login_host(self):
            try:
                # self.tn = telnetlib.Telnet(host_ip,port=23)
                self.tn.open(self.host_ip)
            except:
                text = '{} 网络连接失败'.format(self.host_ip)
                print(text)
                return False
            # 等待login出现后输入用户名，最多等待0.3秒
            self.tn.read_until(b'Username: ', timeout=0.3)
    
            self.tn.write(self.username.encode('ascii') + b'\n')
            # 等待Password出现后输入用户名，最多等待0.3秒
            self.tn.read_until(b'Password: ', timeout=0.3)
    
            self.tn.write(self.password.encode('ascii') + b'\n')
            # 延时5秒再收取返回结果，给服务端足够响应时间
            # time.sleep(5)
    
            # 获取登录结果
            # read_very_eager()获取到的是的是上次获取之后本次获取之前的所有输出
            command_result = self.tn.read_very_eager().decode('utf-8')
    
            if 'Login invalid' in command_result:  # Cisco交换登录失败提示语
                text = '{} 登录失败，用户名或密码错误'.format(self.host_ip)
                print(text)
                return False
            else:
                text = '{} 登录成功'.format(self.host_ip)
                print(text)
                return True
    
        # 执行某一条命令
        def execute_command(self, command, show_res=False):
            self.tn.write(command.encode() + b'\n')
            time.sleep(3)
    
            # 获取命令结果
            command_result = self.tn.read_very_eager().decode('utf-8')
            # if show_res:
            #     print('命令执行结果：%s' % command_result)
            return command_result
    
        # 退出telnet
        def logout_host(self):
            self.tn.write(b"exit\n")
            # print('本次操作结束，连接断开\n')
    
    def check_gateway(ip):
        # 指定连接的交换机管理IP
        telnet_client = TelnetClient(ip=ip, user='root', pswd='123456')
    
        # 如果登录结果返加True，则执行命令，然后退出
        if not telnet_client.login_host():
            print("错误，ip: %s 登录失败"%ip)
            return False
    
        # 执行指定命令,查看进程
        res1 = telnet_client.execute_command('ps')
        # print(res1, type(res1))
        if not res1:
            print("错误，ip: %s 执行命令ps失败" % ip)
            return False
    
        if "/usr/app/bin/app" in res1:
            print("正常，ip: %s 设备正常" % ip)
            telnet_client.execute_command('exit')  # 退出配置模式
        else:
            print("错误，ip: %s 设备掉线" % ip)
            # 执行重启
            res2 = telnet_client.execute_command('reboot')
    
        # 退出登录
        telnet_client.logout_host()
        return True
    
    if __name__ == '__main__':
        ip = "10.212.82.90"
        check_gateway(ip)

**注意：请根据实际情况，修改ip地址，用户名和密码。**

执行脚本

# python3 check_gateway.py 
    10.212.82.90 登录成功
    正常，ip: 10.212.82.90 设备正常

本文参考链接：

[https://blog.csdn.net/XYliurui/article/details/103082512][https_blog.csdn.net_XYliurui_article_details_103082512]

[https_docs.python.org_zh-cn_3_library_telnetlib.html]: https://docs.python.org/zh-cn/3/library/telnetlib.html
[aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9Db250cmFjdGVkQmxvY2suZ2lm]: http://imgconvert.csdnimg.cn/aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9Db250cmFjdGVkQmxvY2suZ2lm
[aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9FeHBhbmRlZEJsb2NrU3RhcnQuZ2lm]: http://imgconvert.csdnimg.cn/aHR0cHM6Ly9pbWFnZXMuY25ibG9ncy5jb20vT3V0bGluaW5nSW5kaWNhdG9ycy9FeHBhbmRlZEJsb2NrU3RhcnQuZ2lm
[https_blog.csdn.net_XYliurui_article_details_103082512]: https://blog.csdn.net/XYliurui/article/details/103082512