import sys, traceback

from . import model, db, custom
from .benchmark import benchmarking
from .flavors import FLAVORS

def cache_everything(*formats):
    custom.is_omniscient = lambda: True
    benchmarking.start()
    all = [e.ename for e in model.Element.getAll()]
    for i in xrange(len(all)):
        print >>sys.stderr, 'Caching %s/%s...' % (i+1, len(all))
        with db.begin_transaction():
            ename = all[i]
            e = model.Element.get(ename)
            for p in e:
                pv = e[p]
                for f in formats:
                    try:
                        pv.render(f)
                    except Exception, ex:
                        traceback.print_exc()