我在GitHub上找到了一个我想要编辑的脚本供我自己使用。
原始脚本为https://github.com/PoorBillionaire/sitereview
我想在我的fork中实现的想法是在文本文件中查找网站列表,而不是每次手动输入python脚本来检查网站,然后将结果输出到文件。
所以,我已经开始研究这个以及我需要在脚本中添加一些选项的事情
p = ArgumentParser()
p.add_argument("-d", "--domains_file", dest = "domains", default = "",
type = "string", help = "A file containing a list of domains to query against BlueCoat.", action="store_true")
p.add_argument("-o", "--output", dest = "output", default = False, help = "Output results to file", action="store_true")
所以这将允许使用
script.py -d ~/home/domainlist.txt -o ~/home/results.txt
目前在哪里
script.py URL
但我不明白的是如何告诉脚本转到文件并检查第一行并获取结果并将其转储到文件中,然后返回并检查输入的下一行文件并转发并转储等,直到没有大多数行要处理..
有人能指出我在哪里工作或提供我需要的一些样本,我可以尝试将它们编辑到我练习的分支中吗?
答案 0 :(得分:0)
试试这个:
args = p.parse()
input_file, dest_file = args.d, args.o
with f_input, f_dest as open(input_file,'r'), open(dest_file,'w'):
for line in f_input.readline():
f_dest.write(line)
f_dest.flush()
如果您遇到模块问题,请阅读argparse
的文档