Limnoria/tools/find-strings.py

#!/usr/bin/env python

import os
import sys
import symbol
import parser

import supybot.utils as utils

def strings(node):
    if node[0] == 3:
        yield (node[2], node[1])
    for x in node:
        if isinstance(x, list):
            for t in strings(x):
                yield t

def getText(filename):
    try:
        fd = file(filename)
        return fd.read()
    finally:
        fd.close()
                
def main():
    out = file('strings.txt', 'w')
    for filename in sys.argv[1:]:
        s = getText(filename)
        node = parser.ast2list(parser.suite(s), True)
        for (lineno, string) in strings(node):
            if string.startswith("r'") or string.startswith('r"'):
                continue
            string = eval(string)
            if len(string) <= 3:
                continue
            out.write('%s: %s: %s\n' % (filename, lineno, string))
    out.close()

if __name__ == '__main__':
    main()