diff --git a/feincms/module/page/sitemap.py b/feincms/module/page/sitemap.py index 89d22c760..fa1f640a7 100644 --- a/feincms/module/page/sitemap.py +++ b/feincms/module/page/sitemap.py @@ -3,10 +3,10 @@ # ------------------------------------------------------------------------ from django.db.models import Max +from django.db.models import get_model from django.contrib.sitemaps import Sitemap -from feincms.module.page.models import Page - +from feincms import settings # ------------------------------------------------------------------------ class PageSitemap(Sitemap): @@ -14,7 +14,7 @@ class PageSitemap(Sitemap): The PageSitemap can be used to automatically generate sitemap.xml files for submission to index engines. See http://www.sitemaps.org/ for details. """ - def __init__(self, navigation_only=False, max_depth=0, changefreq=None, queryset=None, filter=None, extended_navigation=False, *args, **kwargs): + def __init__(self, navigation_only=False, max_depth=0, changefreq=None, queryset=None, filter=None, extended_navigation=False, page_model=settings.FEINCMS_DEFAULT_PAGE_MODEL, *args, **kwargs): """ The PageSitemap accepts the following parameters for customisation of the resulting sitemap.xml output: @@ -42,6 +42,7 @@ def __init__(self, navigation_only=False, max_depth=0, changefreq=None, queryset if queryset is not None: self.queryset = queryset else: + Page = get_model(*page_model.split('.')) self.queryset = Page.objects.active() def items(self):