expomap/news/forms.py

# -*- coding: utf-8 -*-
from django import forms
from django.conf import settings
from django.contrib.contenttypes.models import ContentType
from ckeditor.widgets import CKEditorWidget
from django.core.exceptions import ValidationError
from django.forms.util import ErrorList
#models and forms
from models import News, TYPES
from theme.models import Theme
from accounts.models import User
from django.db.models.loading import get_model
#functions
from functions.translate import fill_with_signal
from functions.files import check_tmp_files
from functions.form_check import translit_with_separator



class NewsForm(forms.Form):
    """
    Create News form

    In __init__ function creates dynamical fields

    save function saves data in News object. If it doesnt exist create new object
    """
    date = forms.DateField(label='Дата')
    type = forms.ChoiceField(label='Тип новости', choices=[(item, item) for item in TYPES])
    paid = forms.BooleanField(label='Платная', required=False)
    #relations
    theme = forms.ModelMultipleChoiceField(label='Тематики', queryset=Theme.objects.all())
    #creates select input with empty choices cause it will be filled with ajax
    tag = forms.MultipleChoiceField(label='Теги', required=False)
    user = forms.ModelChoiceField(label='Организатор', queryset=User.objects.exclude(organiser__isnull=True),
                                                        empty_label=None)
    #
    event = forms.ChoiceField(label='Тип события', choices=[(None, ''), ('conference.Conference', 'Конференция'),
                                                        ('exposition.Exposition', 'Выставка')], required=False)
    event_id = forms.ChoiceField(label='Событие', choices=[(None,'')], required=False)
    #field for comparing tmp files
    key = forms.CharField(required=False, widget=forms.HiddenInput())

    def __init__(self, *args, **kwargs):
        """
        create dynamical translated fields fields
        """
        super(NewsForm, self).__init__(*args, **kwargs)
        #creates translated forms example: name_ru, name_en
        # len(10) is a hack for detect if settings.LANGUAGES is not configured it return all langs
        if len(settings.LANGUAGES) in range(10):
            for lid, (code, name) in enumerate(settings.LANGUAGES):
                # uses enumerate for detect iteration number
                # first iteration is a default lang so it required fields
                required = True if lid == 0 else False
                self.fields['main_title_%s' % code] = forms.CharField(label='Заголовок', required=required,
                                                                 widget=forms.TextInput(attrs={'style':'width: 550px'}))
                self.fields['preview_%s' % code] = forms.CharField(label='Превью', required=required, widget=CKEditorWidget)
                self.fields['description_%s' % code] = forms.CharField(label='Описание', required=required, widget=CKEditorWidget)
                #meta data
                self.fields['title_%s' % code] = forms.CharField(label='Тайтл', required=False, max_length=255,
                                                                 widget=forms.TextInput(attrs={'style':'width: 550px'}))
                self.fields['keywords_%s' % code] = forms.CharField(label='Дескрипшен', required=False, max_length=255,
                                                                 widget=forms.TextInput(attrs={'style':'width: 550px'}))
                self.fields['descriptions_%s' % code] = forms.CharField(label='Кейвордс', required=False, max_length=255,
                                                                 widget=forms.TextInput(attrs={'style':'width: 550px'}))

    def save(self, id=None):
        """
        change news object with id = id
        N/A add new News object
        usage: form.save(id) - if change company
               form.save() - if add company
        """
        data = self.cleaned_data
        #create new Company object or get exists
        if not id:
            news = News()
        else:
            news = News.objects.get(id=id)
            news.theme.clear()
            news.tag.clear()

        #save relation if its filled
        if not data['event_id']:
            obj = get_model(data['event'].split('.')[0],data['event'].split('.')[1]).objects.get(id=data['event_id'])
            news.content_type = ContentType.objects.get_for_model(obj)#
            news.object_id = data['event_id']
        #simple fields
        news.url = translit_with_separator(data['main_title_ru'].strip())
        news.date = data['date']
        news.type = data['type']
        news.paid = data['paid']

        if data.get('user'):
            news.user = data['user']

        # fill translated fields and save object
        fill_with_signal(News, news, data)

        #fill manytomany fields
        for item in data['theme']:
            news.theme.add(item.id)#.id cause select uses queryset

        for item in data['tag']:
            news.tag.add(item)

        # uses because in the next loop data will be overwritten
        news.save()

        #save files
        check_tmp_files(news, data['key'])

    def clean_main_title_ru(self):
        """
        check name which must be unique because it generate slug field
        """
        cleaned_data = super(NewsForm, self).clean()
        main_title_ru = cleaned_data.get('main_title_ru')
        try:
            news = News.objects.get(url=translit_with_separator(main_title_ru))
            if (news.url == translit_with_separator(main_title_ru)):
                return main_title_ru
        except:
            return main_title_ru

        raise ValidationError('Новость с таким названием уже существует')

    def clean(self):
        event_id = self.cleaned_data.get('event_id')
        event = self.cleaned_data.get('event')
        if event and event_id=='':
            msg = 'Выберите событие'
            self._errors['event'] = ErrorList([msg])
            del self.cleaned_data['event']

        return self.cleaned_data

class NewsDeleteForm(forms.ModelForm):
    url = forms.CharField(widget=forms.HiddenInput())

    class Meta:
        model = News
        fields = ('url',)