rgrep.py 1.44 KB
Newer Older
1
#! /usr/bin/env python3
2 3 4 5 6 7 8 9 10 11

"""Reverse grep.

Usage: rgrep [-i] pattern file
"""

import sys
import re
import getopt

12

13
def main():
14
    bufsize = 64 * 1024
15 16 17 18 19 20 21 22 23 24 25 26
    reflags = 0
    opts, args = getopt.getopt(sys.argv[1:], "i")
    for o, a in opts:
        if o == '-i':
            reflags = reflags | re.IGNORECASE
    if len(args) < 2:
        usage("not enough arguments")
    if len(args) > 2:
        usage("exactly one file argument required")
    pattern, filename = args
    try:
        prog = re.compile(pattern, reflags)
27
    except re.error as msg:
28
        usage("error in regular expression: %s" % msg)
29 30
    try:
        f = open(filename)
31
    except IOError as msg:
32
        usage("can't open %r: %s" % (filename, msg), 1)
33 34 35 36 37 38 39 40
    f.seek(0, 2)
    pos = f.tell()
    leftover = None
    while pos > 0:
        size = min(pos, bufsize)
        pos = pos - size
        f.seek(pos)
        buffer = f.read(size)
41
        lines = buffer.split("\n")
42 43 44 45 46 47 48 49 50 51 52
        del buffer
        if leftover is None:
            if not lines[-1]:
                del lines[-1]
        else:
            lines[-1] = lines[-1] + leftover
        if pos > 0:
            leftover = lines[0]
            del lines[0]
        else:
            leftover = None
53
        for line in reversed(lines):
54
            if prog.search(line):
55
                print(line)
56

57

58 59
def usage(msg, code=2):
    sys.stdout = sys.stderr
60 61
    print(msg)
    print(__doc__)
62 63
    sys.exit(code)

64

65 66
if __name__ == '__main__':
    main()