Esempio n. 1
0
    def test_dedicated_controller_wrapper_after_milestone_reached(self):
        conf = AppConfig(minimal=True)

        def f():
            pass

        milestones.environment_loaded.reach()
        conf.register_controller_wrapper(None, controller=f)
Esempio n. 2
0
 def test_global_controller_wrapper_after_milestone_reached(self):
     milestones.environment_loaded.reach()
     conf = AppConfig(minimal=True)
     conf.register_controller_wrapper(None)
# fix imports for appengine environments
import fix_imports
(fix_imports)

from tg import AppConfig
from tg import redirect
from google.appengine.api import users
from main import MainController


def controller_wrapper(next_caller):
    def call(*args, **kw):
        user = users.get_current_user()
        if not user:
            login_url = users.create_login_url('/')
            redirect(login_url)

        return next_caller(*args, **kw)

    return call


config = AppConfig(minimal=True, root_controller=MainController())
config.register_controller_wrapper(controller_wrapper)
app = config.make_wsgi_app()