You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
4.2 KiB
101 lines
4.2 KiB
# -*- 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
|
|
#functions
|
|
from functions.translate import populate, fill_trans_fields_all
|
|
#python
|
|
from PIL import Image
|
|
import pytils, re
|
|
|
|
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=required, 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 |