def wrap(func_self, *args, **kwargs): orig_site = func_self.site func_self.site = Site.find_by_host(self.domain) try: return original_func(func_self, *args, **kwargs) finally: func_self.site = orig_site
def _remember_site(request): if 'site_id' in request.session: request.ctx.site = Site.load(request.session['site_id']) else: #if '__sid' in request.params: # request.ctx.site = Site.find_by_host(request.params['__sid']) #else: request.ctx.site = Site.find_by_host(request.host) if request.ctx.site: request.session['site_id'] = request.ctx.site.site_id if request.ctx.site: return True
def setUp(self): self.settings = paste.deploy.appconfig('config:unittest.ini', relative_to='.') app = self.init_app(self.settings) self.app = TestApp(app) self.site = Site.find_by_host(self.get_host())