1. 前言

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

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

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

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

2. pexpect 介绍

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

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

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

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

1. # 安装依赖库 
2. pip3 install pexpect 

3. 实战一下

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

3-1 SSH 登录

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

1. import pexpect 
2.  
3. PROMPT = ['# ', '>>> ', '> ', '\$ '] 
4.  
5. def login_with_pexpect(): 
6.     """ 
7.     登录-pexpect 
8.     :return: 
9.     """ 
10.     # 端口号 
11.     port = ** 
12.  
13.     # 用户名 
14.     user = ** 
15.  
16.     # 密码 
17.     pwd = ** 
18.  
19.     # ip地址 
20.     ip = ** 
21.  
22.     ssh_cmd = "ssh -p {} {}@{}".format(port,user,ip) 
23.  
24.     # 指定登录命令，拿到程序操作句柄 
25.     child = pexpect.spawn(ssh_cmd, timeout=60, encoding='utf-8') 
26.  
27.     # 提示输入密码的字符出现，否则超时 
28.     ret = child.expect([pexpect.TIMEOUT, '[P|p]assword: '], timeout=10) 
29.  
30.     # 匹配成功后，输入密码，执行登陆操作 
31.     if ret == 1: 
32.         # 发送密码 
33.         child.sendline(pwd) 
34.         child.expect(PROMPT) 
35.         return child 
36.     else: 
37.         print('登录失败！') 

其中

pexpect.spawn() ：用于执行一个程序，并返回一个操作句柄

该方法常见的 3 个参数如下：

• 第一个参数是待执行的命令
• 第二个参数指定超时时间，程序执行后的输出检查，如果指定时间内没有匹配到结果，就会抛异常
• 第三个参数用于设置编码格式

child.expect() ：对输出结果，利用正则表达式进行关键字匹配

代码中的含义是在 10s 内等待出现关键字 '[P|p]assword: '

最后，如果匹配的结果为 1，使用 sendline() 方法发送带回车符的字符串，模拟密码的输入，完成登录操作