Class LineIterator

cluster/code/extract_code.py:8–33

extractors = {}

class LineIterator(object):

    used_lines_per_file = {}

    def __init__(self, filename):
        self.enum_iter = enumerate(open(filename))
        self.linenum = -1
        self.used_lines = self.used_lines_per_file.setdefault(filename, set())

    def __iter__(self):
        return self

    def next(self):
        self.linenum, rv = self.enum_iter.next()
        return rv

    def use_line(self):
        self.used_lines.add(self.linenum)

    @classmethod
    def print_unused_lines(cls, filename):
        sys.stderr.write("*** unused lines in %r ***\n" % filename)
        used_lines = cls.used_lines_per_file[filename]
        for l, line in enumerate(open(filename)):
            if line.strip() and l not in used_lines:
                sys.stderr.write("%4d %s" % (l+1, line))

file_lines_used = {}