Prevent AppRegistryNotReady error when wagtail.contrib.sitemaps is in INSTALLED_APPS

Fixes #4729
pull/4579/merge
Matt Westcott 2018-08-13 15:27:44 +01:00 zatwierdzone przez Matt Westcott
rodzic 5536422c4a
commit 45536ada63
1 zmienionych plików z 1 dodań i 1 usunięć

Wyświetl plik

@ -2,7 +2,6 @@ import warnings
from django.contrib.sitemaps import Sitemap as DjangoSitemap
from wagtail.core.models import Site
from wagtail.core.utils import accepts_kwarg
from wagtail.utils.deprecation import RemovedInWagtail24Warning
@ -23,6 +22,7 @@ class Sitemap(DjangoSitemap):
def get_wagtail_site(self):
site = getattr(self.request, 'site', None)
if site is None:
from wagtail.core.models import Site
return Site.objects.select_related(
'root_page'
).get(is_default_site=True)