Question

我正在用Python编写一个侦察工具，我在尝试在多行变量前面打印一个字符串时遇到了一些问题而没有编辑字符串本身

这是我的代码：

# ...
query1 = commands.getoutput("ls -1 modules/recon | grep '.*\.py$' | grep -v '__init__.py'")
print("module/%s/%s" % (module_type, query1.strip(".py"))

我想添加＆＃34;模块/＃module_type /＃module_name＆＃34;并且模块名称是唯一不断变化的东西。因此，使用shodan和bing模块（随机），输出看起来像这样：

modules/recon/shodan
modules/recon/bing

但我得到了

modules/recon/bing.py
shodan

谢谢！

Answer 1

你可以这样做：

from os import path

module_type = 'recon'
q = 'shoban.py\nbing.py'  # insert the your shell invocation here
modules = (path.splitext(m)[0] for m in q.split('\n'))
formatted = ('modules/%s/%s' % (module_type, m) for m in modules)
print('\n'.join(formatted))

输出：

modules/recon/shodan
modules/recon/bing

但是既然你已经从python调用了unix shell，你也可以使用sed进行字符串处理：

print(commands.getoutput("ls modules/recon/ | sed '/.py$/!d; /^__init__.py$/d; s/\.py$//; s/^/modules\/recon\//'"))

你也可以使用shell＆＃34; globbing＆＃34;如果您要查找模块的位置（例如modules / recon）与您需要输出的前缀匹配，则可以使命令更简单：

print(commands.getoutput("ls modules/recon/*.py | sed 's/.py$//; /\/__init__$/d'"))

另一种选择是使用python的标准库：

from os import path
import glob

module_type = 'recon'
module_paths = glob.iglob('modules/recon/*.py')
module_files = (m for m in map(path.basename, modules) if m != '__init___.py')
modules = (path.splitext(m)[0] for m in module_files)
formatted = ("modules/%s/%s" % (module_type, m) for m in modules)
print('\n'.join(formatted))

在前面的Python 2.7中逐行打印变量

1 个答案: