DNSlookup中的命令行参数

Question

我正在尝试检查IP地址的反向查找，然后将结果写入TXT文件。但是我不知道如何将IP地址作为命令行参数（Linux环境）而不是在脚本中写入IP。

。

我的脚本：

import sys, os, re, shlex, urllib, subprocess 
cmd='dig' -x 8.8.8.8 @192.1.1.1
proc = subprocess.Popen(shlex.split(cmd), stdout=subprocess.PIPE)
out, err = proc.communicate()
sys.stdout = open("/tmp/test.txt", "w")
print(out)
sys.stdout.close()

Answer 1

您可以使用sys.argv获得CLI参数：

import sys, subprocess 
cmd = 'dig -x %s @192.1.1.1' % sys.argv[1]
proc = subprocess.Popen(cmd.split(), stdout=subprocess.PIPE)
out, err = proc.communicate()
with open("/tmp/test.txt", "w+") as f:
    f.write(out)