目的:学习用python脚本登录交换机并执行相关操作


环境:windows 7  Python 3.5.3


情景:最初原因是因为工作,需要在交换机上查询IP地址的来源端口,不停的跳转查看很麻烦,于是决定写脚本,这里只写关于用Python登录和之后的可用方法。


telnet是内置模块,模块名telnetlib。


# 定义登录的用户名、密码和登录的设备地址

username = 'abc'

password = '123'

switch_ip = '1.2.3.4'


# 连接

tel_con = telnetlib.Telnet(switch_ip[, port=23, timeout=10])

# 读取识别并认证

tel_con.read_until('Username:')

tel_con.write(username + rn)

tel_con.read_until('Password:')

tel_con.write(password + rn)

到这里就已经算登录交换机了,下面就可以执行需要的操作:

一般是识别'>'这个符号,当然要根据实际情况,视图模式就可能是其他的了,我在查询的时候一般用dis arp和dis cu获得结果,然后进行过滤处理,这里read_until可能有时候在执行命令时返回的结果不匹配,则语句一直处于等待,可以使用read_all()返回所有的然后用正则表达式(私链),或用延时time.sleep加read_very_eager()的方法。


最后记得要退出,交换机里退出用quit,连接退出用

tel_con.close()


这里还有几个小技巧:

1)设置返回除错信息可以排除问题tel_con.set_debuglevel(5),这里有0-5的等级,数字越大,信息越多。

2)有些情况传递给write方法的字符串需要注意编码

3)read_until()有一个超时设置,超过时间会返回,否则脚本会一直等待

4)read_all()可能得不到EOF而报错。

5)有需要可以学习第三方模块pexpect,它可以实现ssh, telnet, ftp等交互式自动登录


tip:

给懒惰的人和健忘的自己提供一个函数,避免自己重复总结:

def telnet_connect(_ip, _user, _password)
    try:
        _tel_con = telnetlip.Telnet(_ip)
    except:
        return 'out'
    _tel_con.read_until(b'sername:')
    _tel_con.write(encode(_user) + rn)
    _tel_con.read_until(b'assword:')
    _tel_con.write(encode(_password) + rn)
    time.sleep(2)
    _result = decode(_tel_con.read_very_eager())
    if 正则表达式判断是否为密码错误:
        return 'no'
    else:
        return _tel_con

        
# 在应用于网络设备如交换机等,可以写几个函数,如编码、解码和排版命令加rn后延迟


这里有一些参考资料:

用pexpect和paramiko模块实现ssh登录

Paramiko模块登陆SSH并执行上传等操作

paramiko下载及说明(官链)

pexpect下载及说明(官链)