expomap/apps/file/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
#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
from django.utils.translation import ugettext as _


class FileForm(forms.Form):
    file_path = forms.FileField(label=_(u'Выберите файл'))
    model = forms.CharField(required=False, widget=forms.HiddenInput())
    purpose = forms.ChoiceField(
        label=_(u'Назначение'),
        choices=PURPOSES,
        required=False
    )

    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
                self.fields['file_name_%s' % code] = forms.CharField(label=_(u'Имя файла'), required=False, widget=forms.TextInput(attrs={'placeholder': 'Имя'}))
                self.fields['description_%s' % code] = forms.CharField(label=_(u'Описание'), 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']

        if data.get('purpose'):
            file_obj.purpose = data.get('purpose')
        else:
            file_obj.purpose = 'preview'
        #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 FileUpdateForm(forms.Form):
    """
    Форма обновления файла в админ панели
    """
    file_path = forms.FileField(label=_(u'Выберите файл'))
    purpose = forms.ChoiceField(label=_(u'Назаначение'), choices=PURPOSES)

    def __init__(self, *args, **kwargs):
        """
        Создаём динамические поля переводов
        """
        super(FileUpdateForm, self).__init__(*args, **kwargs)
        for lid, (code, name) in enumerate(settings.LANGUAGES):
            self.fields['file_name_%s' % code] = forms.CharField(
                label=_(u'Имя файла'),
                required=False,
                widget=forms.TextInput(attrs={'placeholder': 'Имя'})
            )
            self.fields['description_%s' % code] = forms.CharField(
                label=_(u'Описание'),
                required=False,
                widget=CKEditorWidget()
            )

    def save(self, request, id=None):
        data = self.cleaned_data

        if id is None:
            f = FileModel()
        else:
            f = FileModel.objects.get(id=id)

        f.file_path = data['file_path']
        f.purpose = data['purpose']
        f.save()

        # fill translated fields and save object
        fill_trans_fields_all(FileModel, f, data, id=id)

        return f


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=_(u'Выберите файл'))
    #uses for comparing with TmpFile key
    key = forms.CharField(required=False, widget=forms.HiddenInput())
    model = forms.CharField(required=False, widget=forms.HiddenInput())

    purpose = forms.ChoiceField(label=_(u'Назаначение'), 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=_(u'Имя файла'),required=False,  widget=forms.TextInput(attrs={'placeholder': 'Имя'}))
                self.fields['description_%s'%code] = forms.CharField(label=_(u'Описание'), 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=_(u'Выберите файл'))
    # 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=_(u'Имя файла'),required=False,  widget=forms.TextInput(attrs={'placeholder': 'Имя'}))
                self.fields['description_%s'%code] = forms.CharField(label=_(u'Описание'), 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