summaryrefslogtreecommitdiffstats
path: root/Lib/grep.py
blob: 21ef8d80171aa20f77cf0b3a0e0e149c69125ce8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# '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