mirrors/wagtail
mirrors/wagtail
0
0
Fork 0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00
Code Issues Releases Wiki Activity

Add helper functions for message buttons

Browse Source
This commit is contained in:
Matt Westcott 2020-08-07 01:14:29 +01:00 committed by Matt Westcott
parent 6f8b2eaa99
commit cfdc7da4aa
1 changed files with 16 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
wagtail/admin/views/pages/create.py
View File

@ -111,6 +111,17 @@ class CreateView(TemplateResponseMixin, ContextMixin, View):
else:
return self.save_action()
def get_edit_message_button(self):
return messages.button(reverse('wagtailadmin_pages:edit', args=(self.page.id,)), _('Edit'))
def get_view_draft_message_button(self):
return messages.button(
reverse('wagtailadmin_pages:view_draft', args=(self.page.id,)), _('View draft'), new_window=True
)
def get_view_live_message_button(self):
return messages.button(self.page.url, _('View live'), new_window=True)
def save_action(self):
self.page = self.form.save(commit=False)
self.page.live = False
@ -156,15 +167,13 @@ class CreateView(TemplateResponseMixin, ContextMixin, View):
messages.success(
self.request,
_("Page '{0}' created and scheduled for publishing.").format(self.page.get_admin_display_title()),
buttons=[
messages.button(reverse('wagtailadmin_pages:edit', args=(self.page.id,)), _('Edit'))
]
buttons=[self.get_edit_message_button()]
)
else:
buttons = []
if self.page.url is not None:
buttons.append(messages.button(self.page.url, _('View live'), new_window=True))
buttons.append(messages.button(reverse('wagtailadmin_pages:edit', args=(self.page.id,)), _('Edit')))
buttons.append(self.get_view_live_message_button())
buttons.append(self.get_edit_message_button())
messages.success(
self.request,
_("Page '{0}' created and published.").format(self.page.get_admin_display_title()),
@ -194,20 +203,9 @@ class CreateView(TemplateResponseMixin, ContextMixin, View):
# Notification
buttons = []
if self.page.is_previewable():
buttons.append(
messages.button(
reverse('wagtailadmin_pages:view_draft', args=(self.page.id,)),
_('View draft'),
new_window=True
),
)
buttons.append(self.get_view_draft_message_button())
buttons.append(
messages.button(
reverse('wagtailadmin_pages:edit', args=(self.page.id,)),
_('Edit')
)
)
buttons.append(self.get_edit_message_button())
messages.success(
self.request,