summaryrefslogtreecommitdiffstats
path: root/Lib/lib-old/grep.py
blob: 292674676bb517a0b95f3f2d410f43a4ae7f7827 (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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# 'grep'

import regex
from regex_syntax import *

opt_show_where = 0
opt_show_filename = 0
opt_show_lineno = 1

def grep(pat, *files):
    return ggrep(RE_SYNTAX_GREP, pat, files)

def egrep(pat, *files):
    return ggrep(RE_SYNTAX_EGREP, pat, files)

def emgrep(pat, *files):
    return ggrep(RE_SYNTAX_EMACS, pat, files)

def ggrep(syntax, pat, files):
    if len(files) == 1 and type(files[0]) == type([]):
        files = files[0]
    global opt_show_filename
    opt_show_filename = (len(files) != 1)
    syntax = regex.set_syntax(syntax)
    try:
        prog = regex.compile(pat)
    finally:
        syntax = regex.set_syntax(syntax)
    for filename in files:
        fp = open(filename, 'r')
        lineno = 0
        while 1:
            line = fp.readline()
            if not line: break
            lineno = lineno + 1
            if prog.search(line) >= 0:
                showline(filename, lineno, line, prog)
        fp.close()

def pgrep(pat, *files):
    if len(files) == 1 and type(files[0]) == type([]):
        files = files[0]
    global opt_show_filename
    opt_show_filename = (len(files) != 1)
    import re
    prog = re.compile(pat)
    for filename in files:
        fp = open(filename, 'r')
        lineno = 0
        while 1:
            line = fp.readline()
            if not line: break
            lineno = lineno + 1
            if prog.search(line):
                showline(filename, lineno, line, prog)
        fp.close()

def showline(filename, lineno, line, prog):
    if line[-1:] == '\n': line = line[:-1]
    if opt_show_lineno:
        prefix = `lineno`.rjust(3) + ': '
    else:
        prefix = ''
    if opt_show_filename:
        prefix = filename + ': ' + prefix
    print prefix + line
    if opt_show_where:
        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