# -*- coding: utf-8 -*- from django import forms from django.conf import settings from django.contrib.contenttypes.models import ContentType from ckeditor.widgets import CKEditorWidget #models from country.models import Country from models import FileModel, TmpFile, IMG_TYPES, PURPOSES, Photo #functions from functions.translate import populate, fill_trans_fields_all #python from PIL import Image import pytils, re class FileForm(forms.Form): file_path = forms.FileField(label='Выберите файл') model = forms.CharField(required=False, widget=forms.HiddenInput()) purposes = [('scheme teritory','Схема территории'),('preview','Превью'), ('preview2','Превью2')] purpose = forms.ChoiceField(label='Назначение', choices=purposes) def __init__(self, *args, **kwargs): """ creates dynamical translated fields """ super(FileForm, self).__init__(*args, **kwargs) # creates translated form fields, 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): # using enumerate for detect iteration number # first iteration is a default lang so it required fields required = True if lid == 0 else False self.fields['file_name_%s'%code] = forms.CharField(label='Имя файла',required=False, widget=forms.TextInput(attrs={'placeholder': 'Имя'})) self.fields['description_%s'%code] = forms.CharField(label='Описание', required=False, widget=forms.Textarea()) def save(self, request, obj=None): """ """ data = self.cleaned_data if not obj: return None else: file_obj = FileModel() file_obj.content_type = ContentType.objects.get_for_model(obj) file_obj.object_id = getattr(obj, 'id') #change all symbols than are not letter or number to '_'from file name # and translit name file_name = u'%s'%request['file_path'].name file_name = pytils.translit.translify(file_name) file_name = re.sub('[^\w\-_\.]', '_', file_name) file_name = re.sub('_+', '_', file_name) #file_name = re.sub('_+', '_', re.sub('[^\w\-_\. ]', '_', pytils.translit.translify(u'%s'%request['file_path'].name))) request['file_path'].name = file_name #------- file_obj.file_path = request['file_path'] file_obj.purpose = data['purpose'] #type of file type = str(data['file_path']).split('.')[-1] #if type is image save fields with image size if type.upper() in IMG_TYPES: f = Image.open(data['file_path']) file_obj.img_width, file_obj.img_height = f.size #saves file_type try: file_obj.file_type = type file_obj.save() except: file_obj.file_type = 'OTHER' file_obj.save() #fills and saves translated fields fill_trans_fields_all(FileModel, file_obj, data) return file_obj class FileModelForm(forms.Form): """ Create FileModel form __init__ uses for dynamic creates fields save function saves data in FileModel object. If it doesnt exist create TmpFile object """ file_path = forms.FileField(label='Выберите файл') #uses for comparing with TmpFile key key = forms.CharField(required=False, widget=forms.HiddenInput()) model = forms.CharField(required=False, widget=forms.HiddenInput()) purposes = [(item1, item2) for item1, item2 in PURPOSES] purpose = forms.ChoiceField(label='Назаначение', choices=purposes) def __init__(self, *args, **kwargs): """ creates dynamical translated fields """ super(FileModelForm, self).__init__(*args, **kwargs) # creates translated form fields, 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): # using enumerate for detect iteration number # first iteration is a default lang so it required fields required = True if lid == 0 else False self.fields['file_name_%s'%code] = forms.CharField(label='Имя файла',required=False, widget=forms.TextInput(attrs={'placeholder': 'Имя'})) self.fields['description_%s'%code] = forms.CharField(label='Описание', required=False, widget=CKEditorWidget()) def save(self, request, obj=None): """ if model exist (form.save(request.FILES, obj)) create FileModel object if model doesnt exist (form.save(request.FILES)) create TmpFile object """ data = self.cleaned_data #create TmpFile object or FileModel object if not obj: file_obj = TmpFile() file_obj.key = data['key'] else: file_obj = FileModel() file_obj.content_type = ContentType.objects.get_for_model(obj) file_obj.object_id = getattr(obj, 'id') #change all symbols than are not letter or number to '_'from file name # and translit name file_name = u'%s'%request['file_path'].name file_name = pytils.translit.translify(file_name) file_name = re.sub('[^\w\-_\.]', '_', file_name) file_name = re.sub('_+', '_', file_name) #file_name = re.sub('_+', '_', re.sub('[^\w\-_\. ]', '_', pytils.translit.translify(u'%s'%request['file_path'].name))) request['file_path'].name = file_name #------- file_obj.file_path = request['file_path'] file_obj.purpose = data['purpose'] #type of file type = str(data['file_path']).split('.')[-1] #if type is image save fields with image size if type.upper() in IMG_TYPES: f = Image.open(data['file_path']) file_obj.img_width, file_obj.img_height = f.size #saves file_type # save() uses because in the next loop data will be overwritten try: file_obj.file_type = type file_obj.save() except: file_obj.file_type = 'OTHER' file_obj.save() #fills and saves translated fields fill_trans_fields_all(FileModel, file_obj, data) #populates fields of FileModel object or TmpFile object for code, name in settings.LANGUAGES: if not obj: object = TmpFile._meta.translations_model.objects.get(language_code = code,master__id=getattr(file_obj,'id')) else: object = FileModel._meta.translations_model.objects.get(language_code = code,master__id=getattr(file_obj,'id')) populate(object, data, code) object.save() return file_obj class PhotoForm(forms.Form): file_path = forms.FileField(label='Выберите файл') # uses for comparing with TmpFile key key = forms.CharField(required=False, widget=forms.HiddenInput()) model = forms.CharField(required=False, widget=forms.HiddenInput()) def __init__(self, *args, **kwargs): """ creates dynamical translated fields """ super(PhotoForm, self).__init__(*args, **kwargs) # creates translated form fields, 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): # using 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=False, widget=forms.TextInput(attrs={'placeholder': 'Имя'})) self.fields['description_%s'%code] = forms.CharField(label='Описание', required=False, widget=CKEditorWidget()) def save(self,request, obj=None): """ if model exist (form.save(request.FILES, obj)) create FileModel object if model doesnt exist (form.save(request.FILES)) create TmpFile object """ data = self.cleaned_data #create TmpFile object or Photo object if not obj: file_obj = TmpFile() file_obj.key = data['key'] file_obj.purpose = 'photo' else: file_obj = Photo() file_obj.content_type = ContentType.objects.get_for_model(obj) file_obj.object_id = getattr(obj, 'id') # change all symbols than are not letter or number to '_'from file name # and translit name file_name = u'%s'%request['file_path'].name file_name = pytils.translit.translify(file_name) file_name = re.sub('[^\w\-_\.]', '_', file_name) file_name = re.sub('_+', '_', file_name) # file_name = re.sub('_+', '_', re.sub('[^\w\-_\. ]', '_', pytils.translit.translify(u'%s'%request['file_path'].name))) request['file_path'].name = file_name file_obj.file_path = request['file_path'] # type of file type = str(data['file_path']).split('.')[-1] # if type is image save fields with image size if type.upper() in IMG_TYPES: f = Image.open(data['file_path']) file_obj.img_width, file_obj.img_height = f.size #saves file_type # save() uses because in the next loop data will be overwritten try: file_obj.file_type = type file_obj.save() except: file_obj.file_type = 'OTHER' file_obj.save() #fills and saves translated fields fill_trans_fields_all(Photo, file_obj, data) #populates fields of FileModel object or TmpFile object for code, name in settings.LANGUAGES: if not obj: object = TmpFile._meta.translations_model.objects.get(language_code = code,master__id=getattr(file_obj,'id')) else: object = Photo._meta.translations_model.objects.get(language_code = code,master__id=getattr(file_obj,'id')) populate(object, data, code) object.save() return file_obj