简介：运行python程序时，控制台会输出一些有用的信息。为了方便保存这些信息，有时需要对这些信息进行保存。
历史攻略：
起源：命令行执行ping ip，需求为将ping内容过程保存到文件
ping www.baidu.com输出结果：
正在 Ping www.baidu.com [163.177.151.110] 具有 32 字节的数据:来自 163.177.151.110 的回复: 字节=32 时间=22ms TTL=56来自 163.177.151.110 的回复: 字节=32 时间=14ms TTL=56来自 163.177.151.110 的回复: 字节=32 时间=13ms TTL=56来自 163.177.151.110 的回复: 字节=32 时间=12ms TTL=56163.177.151.110 的 Ping 统计信息:数据包: 已发送 = 4，已接收 = 4，丢失 = 0 (0% 丢失)，往返行程的估计时间(以毫秒为单位):最短 = 12ms，最长 = 22ms，平均 = 15ms
改造成Python脚本：
# -*- coding: utf-8 -*-# time: 2022/7/31 11:26# file: main.py# 公众号: 玩转测试开发import oscmd = "ping www.baidu.com"message = os.system(cmd)print("*" * 100)print(message)
执行脚本结果：
正在 Ping www.baidu.com [163.177.151.110] 具有 32 字节的数据:来自 163.177.151.110 的回复: 字节=32 时间=11ms TTL=56来自 163.177.151.110 的回复: 字节=32 时间=12ms TTL=56来自 163.177.151.110 的回复: 字节=32 时间=12ms TTL=56来自 163.177.151.110 的回复: 字节=32 时间=13ms TTL=56163.177.151.110 的 Ping 统计信息:数据包: 已发送 = 4，已接收 = 4，丢失 = 0 (0% 丢失)，往返行程的估计时间(以毫秒为单位):最短 = 11ms，最长 = 13ms，平均 = 12ms****************************************************************************************************0
产生的问题，即：需求是将控制台的数据当成变量进行保存，结果得到的结果为，星号下的结果：0，这不符合我们想要的结果。原因在于
解决方案1：通过subprocess.check_output()案例：执行命令，返回执行的结果，而不是打印
# -*- coding: utf-8 -*-# time: 2022/7/31 11:26# file: main.py# 公众号: 玩转测试开发import subprocesscmd = "ping www.baidu.com"message = subprocess.check_output(cmd)print("*" * 100)print(message.decode(encoding='gbk'))
执行结果：如遇到中文乱码，参考 - pycharm界面dos命令中文乱码问题处理
****************************************************************************************************正在 Ping www.baidu.com [163.177.151.109] 具有 32 字节的数据:来自 163.177.151.109 的回复: 字节=32 时间=12ms TTL=56来自 163.177.151.109 的回复: 字节=32 时间=69ms TTL=56来自 163.177.151.109 的回复: 字节=32 时间=12ms TTL=56来自 163.177.151.109 的回复: 字节=32 时间=87ms TTL=56163.177.151.109 的 Ping 统计信息:数据包: 已发送 = 4，已接收 = 4，丢失 = 0 (0% 丢失)，往返行程的估计时间(以毫秒为单位):最短 = 12ms，最长 = 87ms，平均 = 45ms
以上情况，可以覆盖绝大部分的场景，个别命令行执行，会出现输出为空的情况，如
起源案例2：
# -*- coding: utf-8 -*-# time: 2022/7/31 11:26# file: main.py# 公众号: 玩转测试开发import subprocesscmd = "locust -f pressure_test.py --headless --users 100 --spawn-rate 1 -H http://127.0.0.1:7890 --run-time 5s"message = subprocess.check_output(cmd)print("*" * 100)print(message.decode(encoding='gbk'))
执行结果：输出的message为空。
[2022-07-31 11:43:10,811] LAPTOP-3FI0HAN5/INFO/locust.main: Run time limit set to 5 seconds[2022-07-31 11:43:10,811] LAPTOP-3FI0HAN5/INFO/locust.main: Starting Locust 2.1.0[2022-07-31 11:43:10,811] LAPTOP-3FI0HAN5/INFO/root: Terminal was not a tty. Keyboard input disabledName # reqs # fails | Avg Min Max Median | req/s failures/s--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------Aggregated 0 0(0.00%) | 0 0 0 0 | 0.00 0.00[2022-07-31 11:43:10,812] LAPTOP-3FI0HAN5/INFO/locust.runners: Ramping to 100 users at a rate of 1.00 per secondName # reqs # fails | Avg Min Max Median | req/s failures/s----------------------------------------------------------------------------------------------------------------------------------------------------------------GET /hello 193 0(0.00%) | 11 7 26 10 | 0.00 0.00GET /world 193 0(0.00%) | 4 1 13 2 | 0.00 0.00----------------------------------------------------------------------------------------------------------------------------------------------------------------Aggregated 386 0(0.00%) | 7 1 26 9 | 0.00 0.00Name # reqs # fails | Avg Min Max Median | req/s failures/s----------------------------------------------------------------------------------------------------------------------------------------------------------------GET /hello 431 0(0.00%) | 14 7 29 15 | 52.50 0.00GET /world 429 0(0.00%) | 8 1 22 9 | 52.50 0.00----------------------------------------------------------------------------------------------------------------------------------------------------------------Aggregated 860 0(0.00%) | 11 1 29 10 | 105.00 0.00[2022-07-31 11:43:15,079] LAPTOP-3FI0HAN5/INFO/locust.main: Time limit reached. Stopping Locust.[2022-07-31 11:43:15,081] LAPTOP-3FI0HAN5/INFO/locust.main: Running teardowns...[2022-07-31 11:43:15,081] LAPTOP-3FI0HAN5/INFO/locust.main: Shutting down (exit code 0), bye.[2022-07-31 11:43:15,081] LAPTOP-3FI0HAN5/INFO/locust.main: Cleaning up runner...Name # reqs # fails | Avg Min Max Median | req/s failures/s----------------------------------------------------------------------------------------------------------------------------------------------------------------GET /hello 462 0(0.00%) | 15 7 34 15 | 108.48 0.00GET /world 461 0(0.00%) | 9 1 30 9 | 108.24 0.00----------------------------------------------------------------------------------------------------------------------------------------------------------------Aggregated 923 0(0.00%) | 12 1 34 11 | 216.72 0.00Response time percentiles (approximated)Type Name 50% 66% 75% 80% 90% 95% 98% 99% 99.9% 99.99% 100% # reqs--------|--------------------------------------------------------------------------------|---------|------|------|------|------|------|------|------|------|------|------|------|GET /hello 15 16 17 19 23 24 29 32 34 34 34 462GET /world 9 10 11 12 17 19 25 28 30 30 30 461--------|--------------------------------------------------------------------------------|---------|------|------|------|------|------|------|------|------|------|------|------|None Aggregated 11 15 16 17 21 24 27 30 34 34 34 923****************************************************************************************************
升级版解决方案1：subprocess.getstatusoutput()，即接受字符串形式的命令，返回 一个元组形式的结果，第一个元素是命令执行状态，第二个为执行结果。
# -*- coding: utf-8 -*-# time: 2022/7/31 11:26# file: main.py# 公众号: 玩转测试开发import subprocesscmd = "locust -f pressure_test.py --headless --users 100 --spawn-rate 1 -H http://127.0.0.1:7890 --run-time 5s"message, out = subprocess.getstatusoutput(cmd)print("*" * 100)print(message)print("*" * 100)file = "locust_test_report.txt"with open(file, "w")as f:f.write(out)print(f"进入文件{file}完毕")
执行结果：
升级版解决方案2：
subprocess.Popen(shell=True, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, universal_newlines=True, bufsize=1)
# -*- coding: utf-8 -*-# time: 2022/7/30 17:00# file: test1700.py# 公众号: 玩转测试开发# -*- coding: UTF8 -*-import subprocess# 常用编码GBK = 'gbk'UTF8 = 'utf-8'# 解码方式，一般 py 文件执行为utf-8 ，cmd 命令为 gbkcurrent_encoding = GBKpopen = subprocess.Popen('locust -f pressure_test.py --headless --users 100 --spawn-rate 1 -H http://127.0.0.1:7890 --run-time 20s',shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE,universal_newlines=True,bufsize=1)# 执行out, err = popen.communicate()print('std_err: ' + err)file = "test.txt"with open(file, "w", encoding="utf-8")as f_obj:f_obj.write(err)
执行结果：
文章评论