def test_register_unregister_isregistered(self):
wizard_pool._clear()
self.assertEqual(len(wizard_pool._entries), 0)
wizard_pool.register(self.page_wizard)
# Now, try to register the same thing
with self.assertRaises(AlreadyRegisteredException):
wizard_pool.register(self.page_wizard)
self.assertEqual(len(wizard_pool._entries), 1)
self.assertTrue(wizard_pool.is_registered(self.page_wizard))
self.assertTrue(wizard_pool.unregister(self.page_wizard))
self.assertEqual(len(wizard_pool._entries), 0)
# Now, try to unregister something that is not registered
self.assertFalse(wizard_pool.unregister(self.user_settings_wizard))
def test_register_unregister_isregistered(self):
wizard_pool._clear()
self.assertEqual(len(wizard_pool._entries), 0)
wizard_pool.register(self.page_wizard)
# Now, try to register the same thing
with self.assertRaises(AlreadyRegisteredException):
wizard_pool.register(self.page_wizard)
self.assertEqual(len(wizard_pool._entries), 1)
self.assertTrue(wizard_pool.is_registered(self.page_wizard))
self.assertTrue(wizard_pool.unregister(self.page_wizard))
self.assertEqual(len(wizard_pool._entries), 0)
# Now, try to unregister something that is not registered
self.assertFalse(wizard_pool.unregister(self.user_settings_wizard))
def populate(self):
#admin_menu.disabled=True
#print(self.toolbar.find_items(Menu, name="Local")[0].item.name)
'''
language_menu = self.toolbar.get_menu(LANGUAGE_MENU_IDENTIFIER )
language_menu.disabled=True
print(language_menu)
page = self.toolbar.get_menu(PAGE_MENU_IDENTIFIER )
page.disabled=True
print(page)
'''
#eliminando el wizzard
for wizard in wizard_pool.get_entries():
if wizard.title == "New Article":
wizard_pool.unregister(wizard)
if (page.reverse_id in defaults.ROOT_REVERSE_IDS
or page.get_ancestor_pages().filter(
reverse_id__in=defaults.ROOT_REVERSE_IDS).exists()):
return False
return super().user_has_add_permission(user, page)
class RichieCMSPageWizard(ExcludeRootReverseIDMixin, CMSPageWizard):
"""A page wizard that can not be created below our Richie extended pages."""
class RichieCMSSubPageWizard(ExcludeRootReverseIDMixin, CMSSubPageWizard):
"""A subpage wizard that can not be created below our Richie extended pages."""
wizard_pool.unregister(cms_page_wizard)
wizard_pool.register(
RichieCMSPageWizard(
title=_("New page"),
weight=100,
form=CreateCMSPageForm,
model=Page,
description=_("Create a new page next to the current page."),
))
wizard_pool.unregister(cms_subpage_wizard)
wizard_pool.register(
RichieCMSSubPageWizard(
title=_("New sub page"),
weight=110,
form=CreateCMSSubPageForm,
)
self.fields["app_config"] = forms.ChoiceField(
label=_("Section"), required=True, choices=app_config_choices
)
def clean(self):
cd = self.cleaned_data
app_config = NewsBlogConfig.objects.get(pk=self.cleaned_data["app_config"])
cd["app_config"] = app_config
return cd
def save(self, commit=True):
article = super(CreateNewsBlogArticleForm, self).save(commit=False)
article.owner = self.user
article.is_published = True
article.save()
return article
wizard_pool.unregister(newsblog_article_wizard)
newsblog_article_wizard = NewsBlogArticleWizard(
title=_(u"New news/blog article"),
weight=200,
form=CreateNewsBlogArticleForm,
description=_(u"Create a new news/blog article."),
)
wizard_pool.register(newsblog_article_wizard)