mirror of
https://github.com/SectorLabs/django-localized-fields.git
synced 2025-10-30 02:28:57 +03:00
added new LocalizedCharField, LocalizedTextField and LocalizedFileField fields
This commit is contained in:
@@ -1,12 +1,18 @@
|
||||
from .localized_field import LocalizedField
|
||||
from .localized_autoslug_field import LocalizedAutoSlugField
|
||||
from .localized_uniqueslug_field import LocalizedUniqueSlugField
|
||||
from .localized_char_field import LocalizedCharField
|
||||
from .localized_text_field import LocalizedTextField
|
||||
from .localized_file_field import LocalizedFileField
|
||||
|
||||
|
||||
__all__ = [
|
||||
'LocalizedField',
|
||||
'LocalizedAutoSlugField',
|
||||
'LocalizedUniqueSlugField',
|
||||
'LocalizedCharField',
|
||||
'LocalizedTextField',
|
||||
'LocalizedFileField'
|
||||
]
|
||||
|
||||
try:
|
||||
|
||||
16
localized_fields/fields/localized_char_field.py
Normal file
16
localized_fields/fields/localized_char_field.py
Normal file
@@ -0,0 +1,16 @@
|
||||
from ..forms import LocalizedCharFieldForm
|
||||
from .localized_field import LocalizedField
|
||||
from ..localized_value import LocalizedStingValue
|
||||
|
||||
|
||||
class LocalizedCharField(LocalizedField):
|
||||
attr_class = LocalizedStingValue
|
||||
|
||||
def formfield(self, **kwargs):
|
||||
"""Gets the form field associated with this field."""
|
||||
defaults = {
|
||||
'form_class': LocalizedCharFieldForm
|
||||
}
|
||||
|
||||
defaults.update(kwargs)
|
||||
return super().formfield(**defaults)
|
||||
@@ -187,7 +187,10 @@ class LocalizedField(HStoreField):
|
||||
# are any of the language fiels None/empty?
|
||||
is_all_null = True
|
||||
for lang_code, _ in settings.LANGUAGES:
|
||||
if value.get(lang_code):
|
||||
# NOTE(seroy): use check for None, instead of
|
||||
# `bool(value.get(lang_code))==True` condition, cause in this way
|
||||
# we can not save '' value
|
||||
if value.get(lang_code) is not None:
|
||||
is_all_null = False
|
||||
break
|
||||
|
||||
@@ -215,7 +218,9 @@ class LocalizedField(HStoreField):
|
||||
|
||||
primary_lang_val = getattr(value, settings.LANGUAGE_CODE)
|
||||
|
||||
if not primary_lang_val:
|
||||
# NOTE(seroy): use check for None, instead of `not primary_lang_val`
|
||||
# condition, cause in this way we can not save '' value
|
||||
if primary_lang_val is None:
|
||||
raise IntegrityError(
|
||||
'null value in column "%s.%s" violates not-null constraint' % (
|
||||
self.name,
|
||||
|
||||
151
localized_fields/fields/localized_file_field.py
Normal file
151
localized_fields/fields/localized_file_field.py
Normal file
@@ -0,0 +1,151 @@
|
||||
import datetime
|
||||
import posixpath
|
||||
|
||||
from django.core.files import File
|
||||
from django.db.models.fields.files import FieldFile
|
||||
from django.utils import six
|
||||
from django.core.files.storage import default_storage
|
||||
from django.utils.encoding import force_str, force_text
|
||||
|
||||
from localized_fields.fields import LocalizedField
|
||||
from localized_fields.fields.localized_field import LocalizedValueDescriptor
|
||||
from localized_fields.localized_value import LocalizedValue
|
||||
|
||||
from ..localized_value import LocalizedFileValue
|
||||
from ..forms import LocalizedFileFieldForm
|
||||
|
||||
|
||||
class LocalizedFieldFile(FieldFile):
|
||||
def __init__(self, instance, field, name):
|
||||
super(FieldFile, self).__init__(None, name)
|
||||
self.instance = instance
|
||||
self.field = field
|
||||
self.storage = field.storage
|
||||
self._committed = True
|
||||
|
||||
def save(self, name, content, lang, save=True):
|
||||
name = self.field.generate_filename(self.instance, name, lang)
|
||||
self.name = self.storage.save(name, content,
|
||||
max_length=self.field.max_length)
|
||||
self._committed = True
|
||||
|
||||
if save:
|
||||
self.instance.save()
|
||||
|
||||
save.alters_data = True
|
||||
|
||||
def delete(self, save=True):
|
||||
if not self:
|
||||
return
|
||||
|
||||
if hasattr(self, '_file'):
|
||||
self.close()
|
||||
del self.file
|
||||
|
||||
self.storage.delete(self.name)
|
||||
|
||||
self.name = None
|
||||
self._committed = False
|
||||
|
||||
if save:
|
||||
self.instance.save()
|
||||
|
||||
delete.alters_data = True
|
||||
|
||||
|
||||
class LocalizedFileValueDescriptor(LocalizedValueDescriptor):
|
||||
def __get__(self, instance, cls=None):
|
||||
value = super().__get__(instance, cls)
|
||||
for k, file in value.__dict__.items():
|
||||
if isinstance(file, six.string_types) or file is None:
|
||||
file = self.field.value_class(instance, self.field, file)
|
||||
value.set(k, file)
|
||||
|
||||
elif isinstance(file, File) and \
|
||||
not isinstance(file, LocalizedFieldFile):
|
||||
file_copy = self.field.value_class(instance, self.field,
|
||||
file.name)
|
||||
file_copy.file = file
|
||||
file_copy._committed = False
|
||||
value.set(k, file_copy)
|
||||
|
||||
elif isinstance(file, LocalizedFieldFile) and \
|
||||
not hasattr(file, 'field'):
|
||||
file.instance = instance
|
||||
file.field = self.field
|
||||
file.storage = self.field.storage
|
||||
|
||||
# Make sure that the instance is correct.
|
||||
elif isinstance(file, LocalizedFieldFile) \
|
||||
and instance is not file.instance:
|
||||
file.instance = instance
|
||||
return value
|
||||
|
||||
|
||||
class LocalizedFileField(LocalizedField):
|
||||
descriptor_class = LocalizedFileValueDescriptor
|
||||
attr_class = LocalizedFileValue
|
||||
value_class = LocalizedFieldFile
|
||||
|
||||
def __init__(self, verbose_name=None, name=None, upload_to='', storage=None,
|
||||
**kwargs):
|
||||
|
||||
self.storage = storage or default_storage
|
||||
self.upload_to = upload_to
|
||||
|
||||
super().__init__(verbose_name, name, **kwargs)
|
||||
|
||||
def deconstruct(self):
|
||||
name, path, args, kwargs = super(LocalizedFileField, self).deconstruct()
|
||||
kwargs['upload_to'] = self.upload_to
|
||||
if self.storage is not default_storage:
|
||||
kwargs['storage'] = self.storage
|
||||
return name, path, args, kwargs
|
||||
|
||||
def get_prep_value(self, value):
|
||||
"""Returns field's value prepared for saving into a database."""
|
||||
|
||||
if isinstance(value, LocalizedValue):
|
||||
prep_value = LocalizedValue()
|
||||
for k, v in value.__dict__.items():
|
||||
if v is None:
|
||||
prep_value.set(k, '')
|
||||
else:
|
||||
# Need to convert File objects provided via a form to
|
||||
# unicode for database insertion
|
||||
prep_value.set(k, six.text_type(v))
|
||||
return super().get_prep_value(prep_value)
|
||||
return super().get_prep_value(value)
|
||||
|
||||
def pre_save(self, model_instance, add):
|
||||
"""Returns field's value just before saving."""
|
||||
value = super().pre_save(model_instance, add)
|
||||
if isinstance(value, LocalizedValue):
|
||||
for lang, file in value.__dict__.items():
|
||||
if file and not file._committed:
|
||||
file.save(file.name, file, lang, save=False)
|
||||
return value
|
||||
|
||||
def generate_filename(self, instance, filename, lang):
|
||||
if callable(self.upload_to):
|
||||
filename = self.upload_to(instance, filename, lang)
|
||||
else:
|
||||
now = datetime.datetime.now()
|
||||
dirname = force_text(now.strftime(force_str(self.upload_to)))
|
||||
dirname = dirname.format(lang=lang)
|
||||
filename = posixpath.join(dirname, filename)
|
||||
return self.storage.generate_filename(filename)
|
||||
|
||||
def save_form_data(self, instance, data):
|
||||
if isinstance(data, LocalizedValue):
|
||||
for k, v in data.__dict__.items():
|
||||
if v is not None and not v:
|
||||
data.set(k, '')
|
||||
setattr(instance, self.attname, data)
|
||||
|
||||
def formfield(self, **kwargs):
|
||||
defaults = {'form_class': LocalizedFileFieldForm}
|
||||
if 'initial' in kwargs:
|
||||
defaults['required'] = False
|
||||
defaults.update(kwargs)
|
||||
return super().formfield(**defaults)
|
||||
14
localized_fields/fields/localized_text_field.py
Normal file
14
localized_fields/fields/localized_text_field.py
Normal file
@@ -0,0 +1,14 @@
|
||||
from ..forms import LocalizedTextFieldForm
|
||||
from .localized_char_field import LocalizedCharField
|
||||
|
||||
|
||||
class LocalizedTextField(LocalizedCharField):
|
||||
def formfield(self, **kwargs):
|
||||
"""Gets the form field associated with this field."""
|
||||
|
||||
defaults = {
|
||||
'form_class': LocalizedTextFieldForm
|
||||
}
|
||||
|
||||
defaults.update(kwargs)
|
||||
return super().formfield(**defaults)
|
||||
Reference in New Issue
Block a user