Jinja2 variable collector

get_vars.py

#!/usr/bin/python

import pprint
import re
import sys

RE = re.compile(r"{{ *(.*?) *}}")
VARS = {}


def find_vars(filename):
    with open(filename) as f:
        for line in f.readlines():
            for m in RE.finditer(line):
                for var in m.groups():
                    VARS[var] = VARS.get(var, 0) + 1


def main():
    for filename in sys.argv[1:]:
        find_vars(filename)
    for var in sorted(VARS.keys()):
        print(var, VARS[var])


if __name__ == '__main__':
    main()