pycode2worldcloud.py

import fileinput
import tokenize

import matplotlib.pyplot as plt
from wordcloud import WordCloud

if __name__ == '__main__':
    # If some files don't have the same encoding, this is likely to not work correctly.
    tokens = tokenize.tokenize(fileinput.input(mode='rb').readline)
    tokens_no_string = (t for t in tokens if t.type not in [tokenize.ENCODING, tokenize.STRING])
    tokens_lowercase_str = (t.string.lower() for t in tokens_no_string)
    wordcloud = WordCloud().generate("\n".join(tokens_lowercase_str))
    plt.imsave("/tmp/foo.png", wordcloud)