72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
import warnings
|
|
|
|
from cms.utils.permissions import get_current_user
|
|
from cms.wizards.wizard_base import Wizard
|
|
from cms.wizards.wizard_pool import AlreadyRegisteredException, wizard_pool
|
|
from django import forms
|
|
from django.conf import settings
|
|
from django.utils.text import slugify
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from parler.forms import TranslatableModelForm
|
|
|
|
from .cms_appconfig import BlogConfig
|
|
from .models import Post
|
|
|
|
|
|
class PostWizardForm(TranslatableModelForm):
|
|
default_appconfig = None
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
if 'initial' not in kwargs or not kwargs.get('initial', False):
|
|
kwargs['initial'] = {}
|
|
kwargs['initial']['app_config'] = self.default_appconfig
|
|
if 'data' in kwargs and kwargs['data'] is not None:
|
|
kwargs['data']['1-app_config'] = self.default_appconfig
|
|
super(PostWizardForm, self).__init__(*args, **kwargs)
|
|
self.fields['app_config'].widget = forms.Select(
|
|
attrs=self.fields['app_config'].widget.attrs,
|
|
choices=self.fields['app_config'].widget.choices,
|
|
)
|
|
self.fields['app_config'].widget.attrs['disabled'] = True
|
|
|
|
class Meta:
|
|
model = Post
|
|
fields = ['app_config', 'title', 'abstract', 'categories']
|
|
|
|
class Media:
|
|
js = ('admin/js/jquery.js', 'admin/js/jquery.init.js',)
|
|
|
|
def save(self, commit=True):
|
|
self.instance._set_default_author(get_current_user())
|
|
return super(PostWizardForm, self).save(commit)
|
|
|
|
|
|
class PostWizard(Wizard):
|
|
pass
|
|
|
|
|
|
for config in BlogConfig.objects.all().order_by('namespace'):
|
|
seed = slugify('{0}.{1}'.format(config.app_title, config.namespace))
|
|
new_wizard = type(str(seed), (PostWizard,), {})
|
|
new_form = type(str('{0}Form').format(seed), (PostWizardForm,), {
|
|
'default_appconfig': config.pk
|
|
})
|
|
post_wizard = new_wizard(
|
|
title=_('New {0}').format(config.object_name),
|
|
weight=200,
|
|
form=new_form,
|
|
model=Post,
|
|
description=_('Create a new {0} in {1}').format(config.object_name, config.app_title),
|
|
)
|
|
try:
|
|
wizard_pool.register(post_wizard)
|
|
except AlreadyRegisteredException: # pragma: no cover
|
|
if settings.DEBUG:
|
|
raise
|
|
else:
|
|
warnings.warn('Wizard {0} cannot be registered. Please make sure that '
|
|
'BlogConfig.namespace {1} and BlogConfig.app_title {2} are'
|
|
'unique together'.format(seed, config.namespace, config.app_title))
|