0
0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00

Split out get_context_data methods

This commit is contained in:
Matt Westcott 2021-12-07 22:32:23 +00:00 committed by LB (Ben Johnston)
parent e48e34d8cf
commit 180f78aae5

View File

@ -578,30 +578,29 @@ class TaskChooserView(BaseTaskChooserView):
return self.render_to_response() return self.render_to_response()
def render_to_response(self): def get_context_data(self):
search_form = TaskChooserSearchForm(task_type_choices=self.get_task_type_filter_choices()) search_form = TaskChooserSearchForm(task_type_choices=self.get_task_type_filter_choices())
tasks = search_form.task_model.objects.filter(active=True).order_by(Lower('name')) tasks = search_form.task_model.objects.filter(active=True).order_by(Lower('name'))
paginator = Paginator(tasks, per_page=10) paginator = Paginator(tasks, per_page=10)
tasks = paginator.get_page(self.request.GET.get('p')) tasks = paginator.get_page(self.request.GET.get('p'))
chooser_html_context = { return {
'tasks': tasks, 'tasks': tasks,
'search_form': search_form, 'search_form': search_form,
'create_form': self.create_form, 'create_form': self.create_form,
'can_create': self.can_create, 'can_create': self.can_create,
'add_url': reverse('wagtailadmin_workflows:task_chooser_create') + '?' + self.request.GET.urlencode() if self.create_model else None 'add_url': reverse('wagtailadmin_workflows:task_chooser_create') + '?' + self.request.GET.urlencode() if self.create_model else None,
'task_types': self.get_task_type_options(),
} }
if not self.create_form: def render_to_response(self):
chooser_html_context['task_types'] = self.get_task_type_options()
js_context = self.get_form_js_context() js_context = self.get_form_js_context()
js_context['step'] = 'chooser' js_context['step'] = 'chooser'
return render_modal_workflow( return render_modal_workflow(
self.request, 'wagtailadmin/workflows/task_chooser/chooser.html', None, self.request, 'wagtailadmin/workflows/task_chooser/chooser.html', None,
chooser_html_context, json_data=js_context self.get_context_data(), json_data=js_context
) )
@ -628,12 +627,18 @@ class TaskChooserCreateView(BaseTaskChooserView):
else: else:
return self.render_to_response() return self.render_to_response()
def render_to_response(self): def get_context_data(self):
tab_html = render_to_string("wagtailadmin/workflows/task_chooser/includes/create_tab.html", { return {
'create_form': self.create_form, 'create_form': self.create_form,
'add_url': reverse('wagtailadmin_workflows:task_chooser_create') + '?' + self.request.GET.urlencode() if self.create_model else None, 'add_url': reverse('wagtailadmin_workflows:task_chooser_create') + '?' + self.request.GET.urlencode() if self.create_model else None,
'task_types': self.get_task_type_options(), 'task_types': self.get_task_type_options(),
}, self.request) }
def render_to_response(self):
tab_html = render_to_string(
"wagtailadmin/workflows/task_chooser/includes/create_tab.html",
self.get_context_data(), self.request
)
js_context = self.get_form_js_context() js_context = self.get_form_js_context()
js_context['step'] = 'reshow_create_tab' js_context['step'] = 'reshow_create_tab'
@ -646,8 +651,8 @@ class TaskChooserCreateView(BaseTaskChooserView):
class TaskChooserResultsView(BaseTaskChooserView): class TaskChooserResultsView(BaseTaskChooserView):
def get(self, request): def get_context_data(self):
search_form = TaskChooserSearchForm(request.GET, task_type_choices=self.get_task_type_filter_choices()) search_form = TaskChooserSearchForm(self.request.GET, task_type_choices=self.get_task_type_filter_choices())
tasks = all_tasks = search_form.task_model.objects.filter(active=True).order_by(Lower('name')) tasks = all_tasks = search_form.task_model.objects.filter(active=True).order_by(Lower('name'))
q = '' q = ''
@ -662,15 +667,21 @@ class TaskChooserResultsView(BaseTaskChooserView):
# Pagination # Pagination
paginator = Paginator(tasks, per_page=10) paginator = Paginator(tasks, per_page=10)
tasks = paginator.get_page(request.GET.get('p')) tasks = paginator.get_page(self.request.GET.get('p'))
return TemplateResponse(request, "wagtailadmin/workflows/task_chooser/includes/results.html", { return {
'search_form': search_form, 'search_form': search_form,
'tasks': tasks, 'tasks': tasks,
'all_tasks': all_tasks, 'all_tasks': all_tasks,
'query_string': q, 'query_string': q,
'can_create': self.can_create, 'can_create': self.can_create,
}) }
def get(self, request):
return TemplateResponse(
request, "wagtailadmin/workflows/task_chooser/includes/results.html",
self.get_context_data()
)
def task_chosen(request, task_id): def task_chosen(request, task_id):