[UPDATE] [20200312]
代码示例部分更新了下,主要更新了判断当前所处的模式的部分
现在我只判断>
, #
以及(config)#
,而不再去拼接hostname
,这也得益于同事的提示,我才发现这个问题
判断用户模式
if ">" in first_few_rows:
print("User Mode.")
判断特权模式
elif "#" in first_few_rows:
print("Enable Mode.")
判断配置模式
if "(config)#" in enter_conf_rows:
print("Entered CONFIG mode SUCCESS! ")
self.check_conf_mode = True
尽管(config)#
里包含了#
,但实际上,你不可能直接进入(config)#
模式,实际上配置模式总是在至少进入特权模式后,才能调用方法来进入,所以也就不存在逻辑上的矛盾
最初的写的时候绕进死胡同了,没想到这点……
因为代码发生了变化,原本的测试代码就无效了,我又没专门写新的测试,所以就干脆删掉了测试那一节,这真的不是因为我懒,真的(确信
除此之外,sleep
的时间也被做成了一个参数,它有默认值,也可以传其它的值进去
第一篇的传送门
https://zhuanlan.zhihu.com/p/108618633
这篇在上篇的代码基础上,以一个抽象过的模块+我个人写这个模块时的思路为引,希望对你使用SSH做自动化这么难受的事能有一些启发(笑……
先说下会遇到的问题
首先一个问题就是,不同的厂家,甚至同厂家的不同设备在登录的时候,初始回显就是不同的,如果你看过上一篇,那么你也就会知道,在上一篇最后的代码中,echolist
这个变量存的是一个逐行读取的列表,但是,由于不同设备登录时初始回显不一样,这就导致我们很难去判断,你真正工作的那些命令的索引……也就是说,你不知道从第几个元素开始,才是你真正想要关心的回显数据。
所以,摘掉这些“无用”的初始回显是第一个问题。