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)