mirror of
https://github.com/Hopiu/wagtail-modeltranslation.git
synced 2026-05-15 02:33:17 +00:00
64d78b2995
Patching of the search_fields to allow the indexation of translated fields on elasticsearch - closes #66
509 lines
14 KiB
Python
Executable file
509 lines
14 KiB
Python
Executable file
# -*- coding: utf-8 -*-
|
|
from django.core import validators
|
|
from django.db import models
|
|
from django.utils import six
|
|
from django.utils.translation import ugettext_lazy
|
|
from modelcluster.fields import ParentalKey
|
|
from wagtail.wagtailadmin.edit_handlers import FieldPanel, MultiFieldPanel, FieldRowPanel, InlinePanel, StreamFieldPanel
|
|
from wagtail.wagtailcore import blocks
|
|
from wagtail.wagtailcore.fields import StreamField
|
|
from wagtail.wagtailcore.models import Page as WagtailPage, Orderable
|
|
from wagtail.wagtailimages.edit_handlers import ImageChooserPanel
|
|
from wagtail.wagtailsearch import index
|
|
from wagtail.wagtailsnippets.models import register_snippet
|
|
|
|
|
|
class TestModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
text = models.TextField(blank=True, null=True)
|
|
url = models.URLField(blank=True, null=True)
|
|
email = models.EmailField(blank=True, null=True)
|
|
|
|
|
|
class UniqueNullableModel(models.Model):
|
|
title = models.CharField(null=True, unique=True, max_length=255)
|
|
|
|
|
|
# ######### Proxy model testing
|
|
|
|
class ProxyTestModel(TestModel):
|
|
class Meta:
|
|
proxy = True
|
|
|
|
def get_title(self):
|
|
return self.title
|
|
|
|
|
|
# ######### Fallback values testing
|
|
|
|
class FallbackModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
text = models.TextField(blank=True, null=True)
|
|
url = models.URLField(blank=True, null=True)
|
|
email = models.EmailField(blank=True, null=True)
|
|
description = models.CharField(max_length=255, null=True)
|
|
|
|
|
|
class FallbackModel2(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
text = models.TextField(blank=True, null=True)
|
|
url = models.URLField(blank=True, null=True)
|
|
email = models.EmailField(blank=True, null=True)
|
|
|
|
|
|
# ######### File fields testing
|
|
|
|
class FileFieldsModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
file = models.FileField(upload_to='modeltranslation_tests', null=True, blank=True)
|
|
file2 = models.FileField(upload_to='modeltranslation_tests')
|
|
image = models.ImageField(upload_to='modeltranslation_tests', null=True, blank=True)
|
|
|
|
|
|
# ######### Foreign Key / OneToOneField testing
|
|
|
|
class NonTranslated(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
|
|
|
|
class ForeignKeyModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
test = models.ForeignKey(TestModel, null=True, related_name="test_fks")
|
|
optional = models.ForeignKey(TestModel, blank=True, null=True)
|
|
hidden = models.ForeignKey(TestModel, blank=True, null=True, related_name="+")
|
|
non = models.ForeignKey(NonTranslated, blank=True, null=True, related_name="test_fks")
|
|
untrans = models.ForeignKey(TestModel, blank=True, null=True, related_name="test_fks_un")
|
|
|
|
|
|
class OneToOneFieldModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
test = models.OneToOneField(TestModel, null=True, related_name="test_o2o")
|
|
optional = models.OneToOneField(TestModel, blank=True, null=True)
|
|
# No hidden option for OneToOne
|
|
non = models.OneToOneField(NonTranslated, blank=True, null=True, related_name="test_o2o")
|
|
|
|
|
|
# ######### Custom fields testing
|
|
|
|
class OtherFieldsModel(models.Model):
|
|
"""
|
|
This class is supposed to include other newly added fields types, so that
|
|
adding new supported field doesn't end in adding new test model.
|
|
"""
|
|
# That's rich! PositiveIntegerField is only validated in forms, not in models.
|
|
int = models.PositiveIntegerField(default=42, validators=[validators.MinValueValidator(0)])
|
|
boolean = models.BooleanField(default=False)
|
|
nullboolean = models.NullBooleanField()
|
|
csi = models.CommaSeparatedIntegerField(max_length=255)
|
|
float = models.FloatField(blank=True, null=True)
|
|
decimal = models.DecimalField(max_digits=5, decimal_places=2, blank=True, null=True)
|
|
ip = models.IPAddressField(blank=True, null=True)
|
|
date = models.DateField(blank=True, null=True)
|
|
datetime = models.DateTimeField(blank=True, null=True)
|
|
time = models.TimeField(blank=True, null=True)
|
|
genericip = models.GenericIPAddressField(blank=True, null=True)
|
|
|
|
|
|
class FancyDescriptor(object):
|
|
"""
|
|
Stupid demo descriptor, that store int in database and return string of that length on get.
|
|
"""
|
|
|
|
def __init__(self, field):
|
|
self.field = field
|
|
|
|
def __get__(self, instance, owner):
|
|
length = instance.__dict__[self.field.name]
|
|
if length is None:
|
|
return ''
|
|
return 'a' * length
|
|
|
|
def __set__(self, obj, value):
|
|
if isinstance(value, six.integer_types):
|
|
obj.__dict__[self.field.name] = value
|
|
elif isinstance(value, six.string_types):
|
|
obj.__dict__[self.field.name] = len(value)
|
|
else:
|
|
obj.__dict__[self.field.name] = 0
|
|
|
|
|
|
class FancyField(models.PositiveIntegerField):
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs.setdefault('default', '')
|
|
super(FancyField, self).__init__(*args, **kwargs)
|
|
|
|
def contribute_to_class(self, cls, name):
|
|
super(FancyField, self).contribute_to_class(cls, name)
|
|
setattr(cls, self.name, FancyDescriptor(self))
|
|
|
|
def pre_save(self, model_instance, add):
|
|
value = super(FancyField, self).pre_save(model_instance, add)
|
|
# In this part value should be retrieved using descriptor and be a string
|
|
assert isinstance(value, six.string_types)
|
|
# We put an int to database
|
|
return len(value)
|
|
|
|
|
|
class DescriptorModel(models.Model):
|
|
normal = FancyField()
|
|
trans = FancyField()
|
|
|
|
|
|
# ######### Multitable inheritance testing
|
|
|
|
class MultitableModelA(models.Model):
|
|
titlea = models.CharField(ugettext_lazy('title a'), max_length=255)
|
|
|
|
|
|
class MultitableModelB(MultitableModelA):
|
|
titleb = models.CharField(ugettext_lazy('title b'), max_length=255)
|
|
|
|
|
|
class MultitableModelC(MultitableModelB):
|
|
titlec = models.CharField(ugettext_lazy('title c'), max_length=255)
|
|
|
|
|
|
class MultitableModelD(MultitableModelB):
|
|
titled = models.CharField(ugettext_lazy('title d'), max_length=255)
|
|
|
|
|
|
# ######### Abstract inheritance testing
|
|
|
|
class AbstractModelA(models.Model):
|
|
titlea = models.CharField(ugettext_lazy('title a'), max_length=255)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(AbstractModelA, self).__init__(*args, **kwargs)
|
|
self.titlea = 'title_a'
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class AbstractModelB(AbstractModelA):
|
|
titleb = models.CharField(ugettext_lazy('title b'), max_length=255)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(AbstractModelB, self).__init__(*args, **kwargs)
|
|
self.titleb = 'title_b'
|
|
|
|
|
|
# ######### Fields inheritance testing
|
|
|
|
class Slugged(models.Model):
|
|
slug = models.CharField(max_length=255)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class MetaData(models.Model):
|
|
keywords = models.CharField(max_length=255)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class Displayable(Slugged, MetaData):
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class BasePage(Displayable):
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class Page(BasePage):
|
|
title = models.CharField(max_length=255)
|
|
|
|
|
|
class RichText(models.Model):
|
|
content = models.CharField(max_length=255)
|
|
|
|
class Meta:
|
|
abstract = True
|
|
|
|
|
|
class RichTextPage(Page, RichText):
|
|
pass
|
|
|
|
|
|
# ######### Admin testing
|
|
|
|
class DataModel(models.Model):
|
|
data = models.TextField(blank=True, null=True)
|
|
|
|
|
|
class GroupFieldsetsModel(models.Model):
|
|
title = models.CharField(max_length=255)
|
|
text = models.TextField(blank=True, null=True)
|
|
email = models.EmailField(blank=True, null=True)
|
|
|
|
|
|
class NameModel(models.Model):
|
|
firstname = models.CharField(max_length=50)
|
|
lastname = models.CharField(max_length=50)
|
|
age = models.CharField(max_length=50)
|
|
slug = models.SlugField(max_length=100)
|
|
slug2 = models.SlugField(max_length=100)
|
|
|
|
|
|
# ######### Integration testing
|
|
|
|
class ThirdPartyModel(models.Model):
|
|
name = models.CharField(max_length=20)
|
|
|
|
|
|
class ThirdPartyRegisteredModel(models.Model):
|
|
name = models.CharField(max_length=20)
|
|
|
|
|
|
# ######### Manager testing
|
|
|
|
class ManagerTestModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
visits = models.IntegerField(ugettext_lazy('visits'), default=0)
|
|
description = models.CharField(max_length=255, null=True)
|
|
|
|
class Meta:
|
|
ordering = ('-visits',)
|
|
|
|
|
|
class CustomManager(models.Manager):
|
|
def get_queryset(self):
|
|
sup = super(CustomManager, self)
|
|
queryset = sup.get_queryset() if hasattr(sup, 'get_queryset') else sup.get_query_set()
|
|
return queryset.filter(title__contains='a').exclude(description__contains='x')
|
|
|
|
get_query_set = get_queryset
|
|
|
|
def custom_qs(self):
|
|
sup = super(CustomManager, self)
|
|
queryset = sup.get_queryset() if hasattr(sup, 'get_queryset') else sup.get_query_set()
|
|
return queryset
|
|
|
|
def foo(self):
|
|
return 'bar'
|
|
|
|
|
|
class CustomManagerTestModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
description = models.CharField(max_length=255, null=True, db_column='xyz')
|
|
objects = CustomManager()
|
|
|
|
another_mgr_name = CustomManager()
|
|
|
|
|
|
class CustomQuerySet(models.query.QuerySet):
|
|
pass
|
|
|
|
|
|
class CustomManager2(models.Manager):
|
|
def get_queryset(self):
|
|
return CustomQuerySet(self.model, using=self._db)
|
|
|
|
get_query_set = get_queryset
|
|
|
|
|
|
class CustomManager2TestModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
objects = CustomManager2()
|
|
|
|
|
|
# ######### Required fields testing
|
|
|
|
class RequiredModel(models.Model):
|
|
non_req = models.CharField(max_length=10, blank=True)
|
|
req = models.CharField(max_length=10)
|
|
req_reg = models.CharField(max_length=10)
|
|
req_en_reg = models.CharField(max_length=10)
|
|
|
|
|
|
# ######### Decorated registration testing
|
|
|
|
class DecoratedModel(models.Model):
|
|
title = models.CharField(ugettext_lazy('title'), max_length=255)
|
|
|
|
|
|
# ######### Wagtail Models
|
|
|
|
class TestRootPage(WagtailPage):
|
|
parent_page_types = []
|
|
subpage_types = ['tests.TestSlugPage1', 'tests.TestSlugPage2']
|
|
|
|
|
|
class TestSlugPage1(WagtailPage):
|
|
pass
|
|
|
|
|
|
class TestSlugPage2(WagtailPage):
|
|
pass
|
|
|
|
|
|
class PatchTestPage(WagtailPage):
|
|
description = models.CharField(max_length=50)
|
|
|
|
search_fields = (
|
|
index.SearchField('title'),
|
|
index.SearchField('description'),
|
|
)
|
|
|
|
|
|
@register_snippet
|
|
class PatchTestSnippet(models.Model):
|
|
name = models.CharField(max_length=10)
|
|
|
|
panels = [
|
|
FieldPanel('name')
|
|
]
|
|
|
|
|
|
# ######### Snippet Patching Models
|
|
|
|
@register_snippet
|
|
class FieldPanelSnippet(models.Model):
|
|
name = models.CharField(max_length=10)
|
|
|
|
panels = [
|
|
FieldPanel('name')
|
|
]
|
|
|
|
|
|
@register_snippet
|
|
class ImageChooserPanelSnippet(models.Model):
|
|
image = models.ForeignKey(
|
|
'wagtailimages.Image'
|
|
)
|
|
|
|
panels = [
|
|
ImageChooserPanel('image'),
|
|
]
|
|
|
|
|
|
@register_snippet
|
|
class FieldRowPanelSnippet(models.Model):
|
|
other_name = models.CharField(max_length=10)
|
|
|
|
panels = [
|
|
FieldRowPanel([
|
|
FieldPanel('other_name'),
|
|
]),
|
|
]
|
|
|
|
|
|
@register_snippet
|
|
class StreamFieldPanelSnippet(models.Model):
|
|
body = StreamField([
|
|
('text', blocks.CharBlock(max_length=10))
|
|
])
|
|
|
|
panels = [
|
|
StreamFieldPanel('body')
|
|
]
|
|
|
|
|
|
@register_snippet
|
|
class MultiFieldPanelSnippet(FieldPanelSnippet, ImageChooserPanelSnippet, FieldRowPanelSnippet):
|
|
panels = [
|
|
MultiFieldPanel(
|
|
FieldPanelSnippet.panels
|
|
),
|
|
MultiFieldPanel(
|
|
ImageChooserPanelSnippet.panels
|
|
),
|
|
MultiFieldPanel(
|
|
FieldRowPanelSnippet.panels
|
|
),
|
|
]
|
|
|
|
|
|
class BaseInlineModel(MultiFieldPanelSnippet):
|
|
field_name = models.CharField(max_length=10)
|
|
|
|
image_chooser = models.ForeignKey('wagtailimages.Image')
|
|
|
|
fieldrow_name = models.CharField(max_length=10)
|
|
|
|
panels = [
|
|
FieldPanel('field_name'),
|
|
ImageChooserPanel('image_chooser'),
|
|
FieldRowPanel([
|
|
FieldPanel('fieldrow_name'),
|
|
]),
|
|
] + MultiFieldPanelSnippet.panels
|
|
|
|
|
|
class SnippetInlineModel(BaseInlineModel):
|
|
page = ParentalKey('tests.InlinePanelSnippet', related_name='related_snippet_model')
|
|
|
|
|
|
@register_snippet
|
|
class InlinePanelSnippet(models.Model):
|
|
panels = [
|
|
InlinePanel('related_snippet_model')
|
|
]
|
|
|
|
|
|
# ######### Page Patching Models
|
|
|
|
class FieldPanelPage(WagtailPage):
|
|
name = models.CharField(max_length=50)
|
|
|
|
content_panels = [
|
|
FieldPanel('name'),
|
|
]
|
|
|
|
|
|
class ImageChooserPanelPage(WagtailPage):
|
|
image = models.ForeignKey(
|
|
'wagtailimages.Image'
|
|
)
|
|
|
|
content_panels = [
|
|
ImageChooserPanel('image'),
|
|
]
|
|
|
|
|
|
class FieldRowPanelPage(WagtailPage):
|
|
other_name = models.CharField(max_length=10)
|
|
|
|
content_panels = [
|
|
FieldRowPanel([
|
|
FieldPanel('other_name'),
|
|
]),
|
|
]
|
|
|
|
|
|
class StreamFieldPanelPage(WagtailPage):
|
|
body = StreamField([
|
|
('text', blocks.CharBlock(max_length=10))
|
|
])
|
|
|
|
content_panels = [
|
|
StreamFieldPanel('body')
|
|
]
|
|
|
|
|
|
class MultiFieldPanelPage(FieldPanelPage, ImageChooserPanelPage, FieldRowPanelPage):
|
|
content_panels = [
|
|
MultiFieldPanel(
|
|
FieldPanelPage.content_panels
|
|
),
|
|
MultiFieldPanel(
|
|
ImageChooserPanelPage.content_panels
|
|
),
|
|
MultiFieldPanel(
|
|
FieldRowPanelPage.content_panels
|
|
),
|
|
]
|
|
|
|
|
|
class PageInlineModel(BaseInlineModel):
|
|
page = ParentalKey('tests.InlinePanelPage', related_name='related_page_model')
|
|
|
|
|
|
class InlinePanelPage(WagtailPage):
|
|
content_panels = [
|
|
InlinePanel('related_page_model')
|
|
]
|