背景
用python脚本文件模拟一个shell命令执行,并且把输出打印出来。
适用场景比如在服务器端是Python脚本,需要执行shell命令,并且把shell命令的执行输出结果获取出来。本文记录一个使用文件作为输出的传输方式,提供一个临时实验的代码模板
代码模板
原理:将cmd用system命令执行,并把输出放在本地目录,然后用cat打印出收集的信息。
import sys
from urllib.parse import quote
import os
import pwd
import subprocess
#command = [""]
#command = command + sys.argv
cmd = sys.argv[1].split()
command = list(cmd)
if "man" == command[0]:
#print("yes")
grepcmd=" | head -n 150"
else:
grepcmd=""
try:
os.system(f"{sys.argv[1]} {grepcmd} > ./pyshelloutput.log")
command=("cat", "./pyshelloutput.log")
process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE) #, shell=True)
#out,err = process.communicate()
#print(process.stdout)
#out = subprocess.check_output(grepcmd, stdin=process.stdout)
out,err = process.communicate()
out = out.decode()
except Exception as e:
out=e
out = "# \"" + str(command) + "\"\n\n" + str(out).replace("\\n", "\n\n") # need 2 \n ding markdown can see
print(str(out))
坑:
- subprocess.Popen直接执行命令很多命令没有权限
用法
python3 cmd.py "ps -u"
效果: