cpython/Lib/grep.py

45 lines
1.0 KiB
Python

# 'grep'
import regex
from regex_syntax import *
import string
def grep(pat, filename):
return ggrep(RE_SYNTAX_GREP, pat, filename)
def egrep(pat, filename):
return ggrep(RE_SYNTAX_EGREP, pat, filename)
def emgrep(pat, filename):
return ggrep(RE_SYNTAX_EMACS, pat, filename)
def ggrep(syntax, pat, filename):
syntax = regex.set_syntax(syntax)
try:
prog = regex.compile(pat)
finally:
syntax = regex.set_syntax(syntax)
fp = open(filename, 'r')
lineno = 0
while 1:
line = fp.readline()
if not line: break
lineno = lineno + 1
if prog.search(line) >= 0:
if line[-1:] == '\n': line = line[:-1]
prefix = string.rjust(`lineno`, 3) + ': '
print prefix + line
if 0: # XXX
start, end = prog.regs()[0]
line = line[:start]
if '\t' not in line:
prefix = ' ' * (len(prefix) + start)
else:
prefix = ' ' * len(prefix)
for c in line:
if c <> '\t': c = ' '
prefix = prefix + c
if start == end: prefix = prefix + '\\'
else: prefix = prefix + '^'*(end-start)
print prefix