python pexpect.before与after_python基于pexpect库自动获取日志信息

不念不忘少年蓝@ 2022-10-27 13:40 17阅读 0赞

1. 前言

对大部分的人来说，解决 Bug 都是依靠关键字去日志去定位问题！

在调试情况下，我们可以实时在控制台查看日志；但对于部署到服务器上的应用，日志都存放在服务器某个目录下，没法通过本地查看到

这种情况下，就需要我们先登录服务器，然后进入到日志目录文件夹，最后通过日志文件去定位问题；如果涉及到 K8s 容器，可能还需要使用 kubectl 命令进入到服务对应的容器中，进入到日志目录，才能开始定位问题，这一切显得非常繁琐且低效

本篇文章介绍一款 Python 依赖库：pexpect，作为一款命令行自动化工具，它可以帮助我们一键获取日志，快速定位问题！

2. pexpect 介绍

pexpect 是 Python 语言实现的类 Expect 的一种实现，通过产生子应用程序并控制它们，通过期望模式对子应用程序的输出做出相应处理

项目地址：

它主要对人机对话进行模拟，来实现一些自动化的场景

比如：用来和 SSH、FTP、PASSWD、Telnet 等命令行程序进行自动化交互

在使用之前，我们先安装依赖包

\# 安装依赖库

pip3 install pexpect

3. 实战一下

我们以从服务器 K8S 容器获取日志为例进行讲解

3-1 SSH 登录

首先，初始化使用 pexpect 模拟 SSH 远程登录服务器

import pexpect

PROMPT = \['\# ', '>>> ', '> ', '\\$ '\]

def login\_with\_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() 方法发送带回车符的字符串，模拟密码的输入，完成登录操作

3-2  封装发送命令

接下来，我们将操作句柄发送命令进行一次封装，并获取返回结果

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 用于获取到匹配到关键字为止，缓存里面已有的全部数据

3-3  进入容器 pod 日志目录

使用上面的方法，通过 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\#')

3-4  获取日志内容

最后我们只需要组成获取日志的命令

比如：使用 grep/tail 命令组成一个提取日志文件的命令

然后使用句柄对象发送这条命令

最后，对返回内容进行清洗即可

4. 最后

文中使用 pexpect 实现了一个非常简单的场景

事实上，大部分人机对话自动化交互都可以由它来实现，比如，下载日志、实时配置、远程打包下载等

以上就是python基于pexpect库自动获取日志信息的详细内容，更多关于python pexpect库自动获取日志的资料请关注脚本之家其它相关文章！

python pexpect.before与after_python基于pexpect库自动获取日志信息

发表评论取消回复

还没有评论，来说两句吧...

相关阅读

相关 Python使用platform库获取系统信息：操作系统信息、硬件信息、python环境信息

相关 Springboot定期自动清除日志信息

相关 python pexpect模块_python pexpect模块

相关 python pexpect.before与after_python基于pexpect库自动获取日志信息

相关 Python批量管理主机(paramiko、fabric与pexpect)

相关 Python批量管理主机(paramiko、fabric与pexpect)

相关 python与pexpect实现ssh操作

相关 Python的Pexpect模块详解

相关 Pexpect库的简单学习

相关 Python获取屏幕坐标，自动发送信息