使用Python登录网关设备执行命令
一、概述
最近发现办公室有一个感测器网关，偶尔会掉线，重启之后，就正常了。
手动操作过程如下：
# 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
三、检测脚本
check_gateway.py
#!/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 设备正常
本文参考链接：
