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.
54 lines
1.6 KiB
54 lines
1.6 KiB
from django.db import models |
|
from django.utils.translation import ugettext_lazy as _ |
|
|
|
|
|
class StyleMixin(models.Model): |
|
class StyleField(models.CharField): |
|
""" |
|
Allows overriding of STYLE_CHOICES in subclasses. |
|
""" |
|
|
|
def contribute_to_class(self, cls, name, **kwargs): |
|
if hasattr(cls, 'STYLE_CHOICES'): |
|
self.choices = cls.STYLE_CHOICES |
|
super().contribute_to_class(cls, name, **kwargs) |
|
|
|
STYLE_CHOICES = ( |
|
('default', _("default")), |
|
) |
|
|
|
style = StyleField(_("style"), max_length=50, null=True, blank=True) |
|
|
|
class Meta: |
|
abstract = True |
|
|
|
def get_style_slug(self): |
|
return getattr(self, 'style', None) or 'default' |
|
|
|
# Compatibiliy with super classes not having a prefixed_path method |
|
def prefixed_path(self, path): |
|
if hasattr(super(), 'prefixed_path'): |
|
return super().prefixed_path(path) |
|
else: |
|
return path |
|
|
|
def get_template_names(self): |
|
if hasattr(super(), 'get_template_names'): |
|
template_names = super().get_template_names() |
|
else: |
|
template_names = [] |
|
|
|
return template_names + [ |
|
self.prefixed_path( |
|
"style/_{style}.html".format( |
|
style=self.get_style_slug()), |
|
) |
|
] |
|
|
|
def get_plugin_context(self, context=None, **kwargs): |
|
if hasattr(super(), 'get_plugin_context'): |
|
context = super().get_plugin_context(**kwargs) |
|
else: |
|
context = {} |
|
context['style'] = self.get_style_slug() |
|
return context
|
|
|