def test_middleware_set_theme_for_site(client, site, settings): settings.theme = 'personal' settings.save() set_theme('wagtail') assert get_theme() == 'wagtail' client.get('/') assert get_theme() == 'personal'
def process_request(self, request): try: site = request.site except AttributeError: raise ImproperlyConfigured( "ThemeMiddleware must be added after SiteMiddleware") if site: theme_settings = ThemeSettings.for_site(site) theme = theme_settings.theme if theme: set_theme(theme)
def process_request(self, request): try: site = Site.find_for_request(request) except: # noqa site = None if not site: raise ImproperlyConfigured("Site not found!") theme_settings = ThemeSettings.for_site(site) theme = theme_settings.theme if theme: set_theme(theme)
def __call__(self, request): site = Site.find_for_request(request) if site is None: raise ImproperlyConfigured("Site not found!") theme_settings = ThemeSettings.for_site(site) theme = theme_settings.theme if theme is not None: set_theme(theme) response = self.get_response(request) return response
def process_request(self, request): try: site = request.site except: # noqa site = None if not site: raise ImproperlyConfigured( "ThemeMiddleware must be added after SiteMiddleware") theme_settings = ThemeSettings.for_site(site) theme = theme_settings.theme if theme: set_theme(theme)
def test_get_theme(): # Check when no theme is set set_theme(None) assert not get_theme() # Check when theme is set set_theme('brand') assert get_theme() == 'brand' # Check when theme is overridden set_theme('personal') assert get_theme() == 'personal'
def test_get_dirs_no_theme_set(): set_theme(None) template = get_template('base.html') assert template.render() == 'Base'
def test_get_dirs_with_extend(): set_theme('brand') template = get_template('page.html') assert template.render() == 'Page - Themes Brand Base'
def test_get_dirs(): set_theme('brand') template = get_template('base.html') assert template.render() == 'Themes Brand Base'