使用 python 递归 grep

Question

我是Python新手，正在努力学习。我正在尝试使用 python 实现一个简单的递归 grep 进行处理，这就是我到目前为止所得到的。

p = subprocess.Popen('find . -name [ch]', shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
  for line in p.stdout.readlines():
    q = subprocess.Popen('grep searchstring %s', line, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    print q.stdout.readlines()

有人可以告诉我如何解决这个问题以完成它应该做的事情吗？

Answer 1

您应该使用 os.walk 函数来浏览文件。使用字符串方法或正则表达式来过滤结果。检查 http://docs.python.org/library/os.html 了解有关如何使用 os.walk。

import os
import re

def findfiles(path, regex):
    regObj = re.compile(regex)
    res = []
    for root, dirs, fnames in os.walk(path):
        for fname in fnames:
            if regObj.match(fname):
                res.append(os.path.join(root, fname))
    return res

print findfiles('.', r'my?(reg|ex)')

现在对于 grep 部分，您可以使用 open 函数循环遍历文件。

def grep(filepath, regex):
    regObj = re.compile(regex)
    res = []
    with open(filepath) as f:
        for line in f:
            if regObj.match(line):
                res.append(line)
    return res

如果您想获取行号，您可能需要查看 enumerate 函数。

编辑添加 grep 函数

Answer 2

您可以使用 python-textops3 ：

示例，从当前目录中的所有 .py 文件中 grep 所有“导入”：

from textops import *

print('\n'.join(('.' | find('*.py') | cat() | grep('import')))) 

它是纯 python，无需分叉进程。

Answer 3

p = subprocess.Popen('find . -name [ch]', shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
  for line in p.stdout.readlines():
    q = subprocess.Popen('grep searchstring %s', line, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    print q.stdout.readlines()

1. 第2行的缩进会异常，for需要与
2. 'grep searchstring %s'上面的p对齐，line不行字符串替换，您需要将 , 替换为 %

通过这些更改和实际搜索值，它可以在我的 OS X 机器上运行。最终脚本：

import subprocess
p = subprocess.Popen('find . -name *.py', shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
for line in p.stdout.readlines():
    print line
    q = subprocess.Popen('grep import %s' % line, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    print q.stdout.readlines()

Answer 4

也许一个例子可以帮助您，命令 find 。 -打印| grep "python" 相当于：

import subprocess

pc1 = subprocess.Popen('find . -print', stdout=subprocess.PIPE, shell=True)
pc2 = subprocess.Popen('grep "python"', stdin=pc1.stdout, shell=True,
                       stdout=subprocess.PIPE)

print pc2.communicate()[0]