# -*- 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 #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('Новость с таким названием уже существует') class NewsDeleteForm(forms.ModelForm): url = forms.CharField(widget=forms.HiddenInput()) class Meta: model = News fields = ('url',)