context_processor.py
0b9483521bdcmaster

context_processor.py
View Options

			# -- coding: utf-8 --
			## This file is part of Invenio.
			## Copyright (C) 2012, 2013 CERN.
			##
			## Invenio is free software; you can redistribute it and/or
			## modify it under the terms of the GNU General Public License as
			## published by the Free Software Foundation; either version 2 of the
			## License, or (at your option) any later version.
			##
			## Invenio is distributed in the hope that it will be useful, but
			## WITHOUT ANY WARRANTY; without even the implied warranty of
			## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
			## General Public License for more details.
			##
			## You should have received a copy of the GNU General Public License
			## along with Invenio; if not, write to the Free Software Foundation, Inc.,
			## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
			"""
			invenio.ext.template.context_processor
			--------------------------------------

			This module provides additional decorator for extending template context
			with new objects.
			"""

			from flask import g


			def register_template_context_processor(f):
			g._template_context_processor.append(f)


			def setup_app(app):
			"""Initializes template context processor extension."""

			@app.before_request
			def reset_template_context():
			"""Resets custom template context buffer."""
			g._template_context_processor = []

			@app.context_processor
			def inject_template_context():
			"""Updates `Jinja2` context by dynamic context processors."""
			context = {}
			for func in getattr(g, '_template_context_processor', []):
			context.update(func())
			reset_template_context()
			return context

			return app