python - 用前面的字符串逐行打印变量 Python 2.7

Question

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

这是我的一小段代码：

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

我想添加“模块/#module_type/#module_name”，模块名称是唯一改变的东西。因此，使用 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\//'"))

如果您要查找模块的位置（例如 modules/recon）与您需要输出的前缀匹配，您还可以使用 shell 的“globbing”功能使命令更简单：

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))