# -*- coding: utf-8 -*- from django import forms from django.conf import settings from ckeditor.widgets import CKEditorWidget from django.core.exceptions import ValidationError from django.forms.util import ErrorList #models from models import Company from country.models import Country from theme.models import Theme from city.models import City #functions from functions.translate import fill_with_signal from functions.form_check import is_positive_integer, translit_with_separator from functions.files import check_tmp_files from functions.custom_fields import LocationWidget class CompanyForm(forms.Form): """ Create Company form In __init__ function creates dynamical fields save function saves data in Company object. If it doesnt exist create new object """ url = forms.CharField(label='URL', widget=forms.TextInput(attrs={'placeholder': 'Введите URL'})) country = forms.ModelChoiceField(label='Страна', queryset=Country.objects.all(), empty_label=None) theme = forms.ModelMultipleChoiceField(label='Тематики', queryset=Theme.objects.all()) #creates select input with empty choices cause it will be filled with ajax city = forms.ChoiceField(label='Город', choices=[('','')]) tag = forms.MultipleChoiceField(label='Теги', required=False) staff_number = forms.CharField(label='Количество сотрудников', required=False, widget=forms.TextInput(attrs={'placeholder': 'Количество сотрудников'})) #uses locationwidget address = forms.CharField(label='Адрес', required=False, widget=LocationWidget) phone = forms.CharField(label='Телефон', required=False, widget=forms.TextInput(attrs={'placeholder': 'Введите телефон'})) fax = forms.CharField(label='Факс', required=False, widget=forms.TextInput(attrs={'placeholder': 'Введите факс'})) web_page = forms.CharField(label='Веб-сайт', required=False, widget=forms.TextInput(attrs={'placeholder': 'Введите адрес сайта'})) email = forms.CharField(label='Email', required=False, widget=forms.TextInput(attrs={'placeholder': 'Введите email'})) social = forms.CharField(label='Социальные страници', required=False) foundation = forms.CharField(label='Год основания', required=False, widget=forms.TextInput(attrs={'placeholder': 'Год основания'})) #field for comparing tmp files key = forms.CharField(required=False, widget=forms.HiddenInput()) # company_id = forms.CharField(required=False, widget=forms.HiddenInput()) def __init__(self, *args, **kwargs): """ create dynamical translated fields fields """ super(CompanyForm, 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['name_%s' % code] = forms.CharField(label='Название', required=required) self.fields['description_%s' % code] = forms.CharField(label='Описание', required=False, widget=CKEditorWidget) self.fields['specialization_%s' % code] = forms.CharField(label='Специализация', required=False) self.fields['address_inf_%s' % code] = forms.CharField(label='Доп инф по адресу', required=False, 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 company object with id = id N/A add new Company object usage: form.save(obj) - if change company form.save() - if add company """ data = self.cleaned_data #create new Company object or get exists if not id: company = Company() else: company = Company.objects.get(id=id) company.theme.clear() company.tag.clear() #simple fields company.url = translit_with_separator(data['url'].strip()).lower() company.staff_number = data['staff_number'] company.address = data['address'] company.phone = data['phone'] company.fax = data['fax'] company.web_page = data['web_page'] company.email = data['email'] company.foundation = data['foundation'] company.social = data['social'] if data.get('country'): company.country = Country.objects.get(id=data['country'].id) if data.get('city'): company.city = City.objects.get(id=data['city']) # fill translated fields and save object fill_with_signal(Company, company, data) #fill manytomany fields for item in data['theme']: company.theme.add(item.id)#.id cause select uses queryset for item in data['tag']: company.tag.add(item) # uses because in the next loop data will be overwritten company.save() #save files check_tmp_files(company, data['key']) def clean(self): id = self.cleaned_data.get('company_id') url = self.cleaned_data.get('url') company = Company.objects.filter(url=translit_with_separator(url)) if company and str(company[0].id) != id: msg = 'Такой урл уже занят' self._errors['url'] = ErrorList([msg]) del self.cleaned_data['url'] return self.cleaned_data def clean_foundation(self): """ checking foundation """ cleaned_data = super(CompanyForm, self).clean() foundation = cleaned_data.get('foundation').strip() return is_positive_integer(foundation) def clean_web_page(self): cleaned_data = super(CompanyForm, self).clean() web_page = cleaned_data.get('web_page') if not web_page: return web_page import socket try: socket.getaddrinfo(web_page, 80) return web_page except: return forms.ValidationError('Введите правильный адрес страници') def clean_phone(self): """ phone checking """ cleaned_data = super(CompanyForm, self).clean() phone = cleaned_data.get('phone') if not phone: return deduct = ('-','(',')','.',' ') for elem in deduct: phone = phone.replace(elem, '') if phone.isdigit(): return phone else: raise ValidationError('Введите правильный телефон') def clean_fax(self): """ fax checking """ cleaned_data = super(CompanyForm, self).clean() fax = cleaned_data.get('fax') if not fax: return deduct = ('-','(',')','.',' ') for elem in deduct: fax = fax.replace(elem, '') if fax.isdigit(): return fax else: raise ValidationError('Введите правильный факс') class CompanyDeleteForm(forms.ModelForm): id = forms.CharField(widget=forms.HiddenInput()) class Meta: model = Company fields = ('id',)