背景
用python脚本文件模拟一个shell命令执行,并且把输出打印出来。
适用场景比如在服务器端是Python脚本,需要执行shell命令,并且把shell命令的执行输出结果获取出来。本文记录一个使用文件作为输出的传输方式,提供一个临时实验的代码模板
代码模板
原理:将cmd用system命令执行,并把输出放在本地目录,然后用cat打印出收集的信息。
import sys
from urllib.parse import quote
import os
import pwdimport 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=eout = "# \"" + 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"
效果: