怎么在python中利用pexpect库自动获取日志信息?很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。
pexpect 是 Python 语言实现的类 Expect 的一种实现,通过产生子应用程序并控制它们,通过期望模式对子应用程序的输出做出相应处理
项目地址:
https://github.com/pexpect/pexpect/tree/5eed1a31a2853a09b7367c59fbd1a4a0b53341df
它主要对人机对话进行模拟,来实现一些自动化的场景
比如:用来和 SSH、FTP、PASSWD、Telnet 等命令行程序进行自动化交互
在使用之前,我们先安装依赖包
# 安装依赖库
pip3 install pexpect
我们以从服务器 K8S 容器获取日志为例进行讲解
首先,初始化使用 pexpect 模拟 SSH 远程登录服务器
import pexpect
PROMPT = ['# ', '>>> ', '> ', '\$ ']
def login_with_pexpect():
"""
登录-pexpect
:return:
"""
# 端口号
port = **
# 用户名
user = **
# 密码
pwd = **
# ip地址
ip = **
ssh_cmd = "ssh -p {} {}@{}".format(port,user,ip)
# 指定登录命令,拿到程序操作句柄
child = pexpect.spawn(ssh_cmd, timeout=60, encoding='utf-8')
# 提示输入密码的字符出现,否则超时
ret = child.expect([pexpect.TIMEOUT, '[P|p]assword: '], timeout=10)
# 匹配成功后,输入密码,执行登陆操作
if ret == 1:
# 发送密码
child.sendline(pwd)
child.expect(PROMPT)
return child
else:
print('登录失败!')
其中
pexpect.spawn() :用于执行一个程序,并返回一个操作句柄
该方法常见的 3 个参数如下:
第一个参数是待执行的命令
第二个参数指定超时时间,程序执行后的输出检查,如果指定时间内没有匹配到结果,就会抛异常
第三个参数用于设置编码格式
child.expect() :对输出结果,利用正则表达式进行关键字匹配
代码中的含义是在 10s 内等待出现关键字 '[P|p]assword: '
最后,如果匹配的结果为 1,使用 sendline() 方法发送带回车符的字符串,模拟密码的输入,完成登录操作
接下来,我们将操作句柄发送命令进行一次封装,并获取返回结果
def send_command(child, cmd, expected_content=None, timeout=10):
"""
发送一条命令,并打印结果
:param expected_content:
:param child:
:param cmd:
:return:
"""
# 发送一条命令
if expected_content is None:
expected_content = ["#"]
child.sendline(cmd)
# 期望有命令行提示字符出现
child.expect(expected_content, timeout=timeout)
# 将之前的内容都输出
result = child.before
return result
需要指出的是,child.before 用于获取到匹配到关键字为止,缓存里面已有的全部数据
使用上面的方法,通过 kubectl 进入到对应的容器日志目录
# 登录pod容器
exec_enter_pod = 'kubectl -n %s exec -it %s bash' % (env, pod_name)
# 进入容器
send_command(child, exec_enter_pod)
# 进入到日志目录
send_command(child, "cd logs", expected_content='tomcat/logs#')
最后我们只需要组成获取日志的命令
比如:使用 grep/tail 命令组成一个提取日志文件的命令
然后使用句柄对象发送这条命令
最后,对返回内容进行清洗即可
看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注亿速云行业资讯频道,感谢您对亿速云的支持。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。