Find and highlight strings in python code, helpful for finding things for internationalization.

findstrings.py

"""
Usage: python findstrings.py <file.py> [<file.py> [...]]
"""

import tokenize
import token

def findstrings(readline):
    for tok in tokenize.tokenize(readline):
        if token.tok_name[tok.type] == "STRING":
            yield tok

if __name__ == "__main__":
    from termcolor import colored
    import sys
    for filename in sys.argv[1:]:
        with open(filename, "rb") as f:
            for tok in findstrings(f.readline):
                line = tok.line.rstrip()
                head = line[0:tok.start[1]]
                middle = colored(tok.string, "red")
                tail = line.splitlines()[-1][tok.end[1]:]
                print("{}:{}:{}{}{}".format(filename, tok.start[0], head, middle, tail))