kopia lustrzana https://github.com/wagtail/wagtail
Pass SnippetViewSet.get_queryset function as the queryset instead of accessing the viewset directly
rodzic
87cbf01894
commit
fbd3cca30f
|
@ -164,10 +164,11 @@ class IndexView(generic.IndexViewOptionalFeaturesMixin, generic.IndexView):
|
||||||
table_class = InlineActionsTable
|
table_class = InlineActionsTable
|
||||||
|
|
||||||
def get_base_queryset(self):
|
def get_base_queryset(self):
|
||||||
base_queryset = self.viewset.get_queryset(self.request)
|
# Allow the queryset to be a callable that takes a request
|
||||||
if base_queryset is None:
|
# so that it can be evaluated in the context of the request
|
||||||
return super().get_base_queryset()
|
if callable(self.queryset):
|
||||||
return base_queryset
|
self.queryset = self.queryset(self.request)
|
||||||
|
return super().get_base_queryset()
|
||||||
|
|
||||||
def _get_title_column(self, field_name, column_class=SnippetTitleColumn, **kwargs):
|
def _get_title_column(self, field_name, column_class=SnippetTitleColumn, **kwargs):
|
||||||
# Use SnippetTitleColumn class to use custom template
|
# Use SnippetTitleColumn class to use custom template
|
||||||
|
@ -794,6 +795,7 @@ class SnippetViewSet(ViewSet):
|
||||||
def index_view(self):
|
def index_view(self):
|
||||||
return self.index_view_class.as_view(
|
return self.index_view_class.as_view(
|
||||||
model=self.model,
|
model=self.model,
|
||||||
|
queryset=self.get_queryset,
|
||||||
template_name=self.get_index_template(),
|
template_name=self.get_index_template(),
|
||||||
header_icon=self.icon,
|
header_icon=self.icon,
|
||||||
filterset_class=self.filterset_class,
|
filterset_class=self.filterset_class,
|
||||||
|
@ -812,6 +814,7 @@ class SnippetViewSet(ViewSet):
|
||||||
def index_results_view(self):
|
def index_results_view(self):
|
||||||
return self.index_view_class.as_view(
|
return self.index_view_class.as_view(
|
||||||
model=self.model,
|
model=self.model,
|
||||||
|
queryset=self.get_queryset,
|
||||||
template_name=self.get_index_results_template(),
|
template_name=self.get_index_results_template(),
|
||||||
header_icon=self.icon,
|
header_icon=self.icon,
|
||||||
filterset_class=self.filterset_class,
|
filterset_class=self.filterset_class,
|
||||||
|
|
Ładowanie…
Reference in New Issue