363 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			363 lines
		
	
	
	
		
			12 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from cms.extensions import PageExtension
 | 
						|
from cms.extensions.extension_pool import extension_pool
 | 
						|
from cms.models.fields import PlaceholderField
 | 
						|
from cms.models.pluginmodel import CMSPlugin
 | 
						|
from django import forms
 | 
						|
from django.conf import settings
 | 
						|
from django.contrib.postgres.fields import ArrayField
 | 
						|
from django.contrib.sites.models import Site
 | 
						|
from django.db import models
 | 
						|
from django.utils.safestring import mark_safe
 | 
						|
from djangocms_text_ckeditor.fields import HTMLField
 | 
						|
from filer.fields.file import FilerFileField
 | 
						|
from filer.fields.image import FilerImageField
 | 
						|
 | 
						|
from datacenterlight.models import VMPricing, VMTemplate
 | 
						|
 | 
						|
 | 
						|
class CMSIntegration(models.Model):
 | 
						|
    name = models.CharField(
 | 
						|
        max_length=100, default='default',
 | 
						|
        help_text=(
 | 
						|
            'A unique name for the Integration. This name will be used to '
 | 
						|
            'fetch the Integration into pages'
 | 
						|
        )
 | 
						|
    )
 | 
						|
    footer_placeholder = PlaceholderField(
 | 
						|
        'datacenterlight_footer', related_name='dcl-footer-placeholder+'
 | 
						|
    )
 | 
						|
    navbar_placeholder = PlaceholderField(
 | 
						|
        'datacenterlight_navbar', related_name='dcl-navbar-placeholder+'
 | 
						|
    )
 | 
						|
    calculator_placeholder = PlaceholderField(
 | 
						|
        'datacenterlight_calculator',
 | 
						|
        related_name='dcl-calculator-placeholder+'
 | 
						|
    )
 | 
						|
    domain = models.ForeignKey(Site, null=True, blank=True)
 | 
						|
 | 
						|
    class Meta:
 | 
						|
        unique_together = ('name', 'domain')
 | 
						|
 | 
						|
    def __str__(self):
 | 
						|
        return self.name
 | 
						|
 | 
						|
 | 
						|
class CMSFaviconExtension(PageExtension):
 | 
						|
    favicon = FilerFileField(related_name="cms_favicon_image")
 | 
						|
 | 
						|
 | 
						|
extension_pool.register(CMSFaviconExtension)
 | 
						|
 | 
						|
 | 
						|
# Models for CMS Plugins
 | 
						|
 | 
						|
class DCLSectionPluginModel(CMSPlugin):
 | 
						|
    heading = models.CharField(
 | 
						|
        blank=True, null=True, max_length=100,
 | 
						|
        help_text='An optional heading for the Section',
 | 
						|
    )
 | 
						|
    content = HTMLField()
 | 
						|
    TEXT_DIRECTIONS = (
 | 
						|
        ('left', 'Left'),
 | 
						|
        ('right', 'Right')
 | 
						|
    )
 | 
						|
    text_direction = models.CharField(
 | 
						|
        choices=TEXT_DIRECTIONS, max_length=10, default=True,
 | 
						|
        help_text='The alignment of text in the section'
 | 
						|
    )
 | 
						|
    html_id = models.SlugField(
 | 
						|
        blank=True, null=True,
 | 
						|
        help_text=(
 | 
						|
            'An optional html id for the Section. Required to set as target '
 | 
						|
            'of a link on page'
 | 
						|
        )
 | 
						|
    )
 | 
						|
    plain_heading = models.BooleanField(
 | 
						|
        default=False,
 | 
						|
        help_text='Select to keep the heading style simpler.'
 | 
						|
    )
 | 
						|
    center_on_mobile = models.BooleanField(
 | 
						|
        default=False,
 | 
						|
        help_text='Select to center align content on small screens.'
 | 
						|
    )
 | 
						|
    background_gradient = models.BooleanField(
 | 
						|
        default=False,
 | 
						|
        help_text='Select to add a gradient background to the section.'
 | 
						|
    )
 | 
						|
 | 
						|
    def get_extra_classes(self):
 | 
						|
        extra_classes = self.text_direction
 | 
						|
        if self.center_on_mobile:
 | 
						|
            extra_classes += ' section-sm-center'
 | 
						|
        if self.background_gradient:
 | 
						|
            extra_classes += ' section-gradient'
 | 
						|
        if self.plain_heading:
 | 
						|
            extra_classes += ' split-section-plain'
 | 
						|
        return extra_classes
 | 
						|
 | 
						|
    def __str__(self):
 | 
						|
        return '#' + self.html_id if self.html_id else str(self.pk)
 | 
						|
 | 
						|
 | 
						|
class DCLBannerListPluginModel(CMSPlugin):
 | 
						|
    heading = models.CharField(
 | 
						|
        blank=True, null=True, max_length=100,
 | 
						|
        help_text='An optional heading for the Section',
 | 
						|
    )
 | 
						|
    html_id = models.SlugField(
 | 
						|
        blank=True, null=True,
 | 
						|
        help_text=(
 | 
						|
            'An optional html id for the Section. Required to set as target '
 | 
						|
            'of a link on page'
 | 
						|
        )
 | 
						|
    )
 | 
						|
 | 
						|
    def __str__(self):
 | 
						|
        return '#' + self.html_id if self.html_id else str(self.pk)
 | 
						|
 | 
						|
 | 
						|
class DCLBannerItemPluginModel(CMSPlugin):
 | 
						|
    content = HTMLField()
 | 
						|
    banner_text = HTMLField(
 | 
						|
        blank=True, null=True, max_length=100,
 | 
						|
        help_text='Optional text to be shown as banner in other half.',
 | 
						|
    )
 | 
						|
    banner_image = FilerImageField(
 | 
						|
        on_delete=models.CASCADE, null=True, blank=True,
 | 
						|
        help_text='Optional image to be used in the banner in other half.'
 | 
						|
    )
 | 
						|
    TEXT_DIRECTIONS = (
 | 
						|
        ('left', 'Left'),
 | 
						|
        ('right', 'Right')
 | 
						|
    )
 | 
						|
    text_direction = models.CharField(
 | 
						|
        choices=TEXT_DIRECTIONS, max_length=10, default=True,
 | 
						|
        help_text='The alignment of text in the section'
 | 
						|
    )
 | 
						|
 | 
						|
    def get_extra_classes(self):
 | 
						|
        extra_classes = ''
 | 
						|
        if self.text_direction == 'left':
 | 
						|
            extra_classes = 'flex-row-rev'
 | 
						|
        return extra_classes
 | 
						|
 | 
						|
 | 
						|
class DCLLinkPluginModel(CMSPlugin):
 | 
						|
    target = models.CharField(
 | 
						|
        max_length=100,
 | 
						|
        help_text='Url or #id to navigate to'
 | 
						|
    )
 | 
						|
    text = models.CharField(
 | 
						|
        max_length=50,
 | 
						|
        help_text='Text for the menu item'
 | 
						|
    )
 | 
						|
    title = models.CharField(
 | 
						|
        blank=True, null=True, max_length=100,
 | 
						|
        help_text=(
 | 
						|
            'Optional title text, that will be shown when a user '
 | 
						|
            'hovers over the link'
 | 
						|
        )
 | 
						|
    )
 | 
						|
    separator = models.BooleanField(
 | 
						|
        default=False,
 | 
						|
        help_text='Select to include a separator after the previous link'
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class DCLNavbarPluginModel(CMSPlugin):
 | 
						|
    logo_light = FilerImageField(
 | 
						|
        on_delete=models.CASCADE, null=True, blank=True,
 | 
						|
        help_text='Logo to be used on transparent navbar',
 | 
						|
        related_name="dcl_navbar_logo_light",
 | 
						|
    )
 | 
						|
    logo_dark = FilerImageField(
 | 
						|
        on_delete=models.CASCADE, null=True, blank=True,
 | 
						|
        help_text='Logo to be used on white navbar',
 | 
						|
        related_name="dcl_navbar_logo_dark",
 | 
						|
    )
 | 
						|
    logo_url = models.URLField(max_length=300, null=True, blank=True)
 | 
						|
    language_dropdown = models.BooleanField(
 | 
						|
        default=True,
 | 
						|
        help_text='Select to include the language selection dropdown.'
 | 
						|
    )
 | 
						|
    show_login_option = models.BooleanField(
 | 
						|
        default=True,
 | 
						|
        help_text='Uncheck this if you do not want to show login/dashboard.'
 | 
						|
    )
 | 
						|
 | 
						|
    def get_logo_dark(self):
 | 
						|
        # used only if atleast one logo exists
 | 
						|
        return self.logo_dark.url if self.logo_dark else self.logo_light.url
 | 
						|
 | 
						|
    def get_logo_light(self):
 | 
						|
        # used only if atleast one logo exists
 | 
						|
        return self.logo_light.url if self.logo_light else self.logo_dark.url
 | 
						|
 | 
						|
 | 
						|
class DCLNavbarDropdownPluginModel(CMSPlugin):
 | 
						|
    target = models.CharField(
 | 
						|
        max_length=100, null=True, blank=True,
 | 
						|
        help_text='Optional Url or #id to navigate on click'
 | 
						|
    )
 | 
						|
    text = models.CharField(
 | 
						|
        max_length=50,
 | 
						|
        help_text='Text for the dropdown toggle'
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class DCLContactPluginModel(CMSPlugin):
 | 
						|
    heading = models.CharField(max_length=100, default="Contact", blank=True)
 | 
						|
    organization_name = models.CharField(
 | 
						|
        max_length=100, default="ungleich glarus ag", blank=True
 | 
						|
    )
 | 
						|
    email = models.EmailField(max_length=200, default="info@ungleich.ch")
 | 
						|
    address = models.CharField(
 | 
						|
        max_length=100, default="In der Au 7, Schwanden 8762", blank=True
 | 
						|
    )
 | 
						|
    country = models.CharField(
 | 
						|
        max_length=100, default="Switzerland", blank=True
 | 
						|
    )
 | 
						|
    form_header = models.CharField(
 | 
						|
        max_length=100, default="Send us a message.", blank=True
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class DCLFooterPluginModel(CMSPlugin):
 | 
						|
    copyright_label = models.CharField(
 | 
						|
        max_length=100, default='ungleich glarus ag', blank=True,
 | 
						|
        help_text='Name of the company alongside the copyright year'
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class DCLSectionIconPluginModel(CMSPlugin):
 | 
						|
    fontawesome_icon_name = models.CharField(
 | 
						|
        max_length=30,
 | 
						|
        help_text=mark_safe(
 | 
						|
            'Name of the fontawesome icon to use. '
 | 
						|
            '<a href="https://fontawesome.com/v4.7.0/icons/" target="_blank">'
 | 
						|
            'Refer docs.</a>'
 | 
						|
        )
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class DCLSectionImagePluginModel(CMSPlugin):
 | 
						|
    image = FilerImageField(
 | 
						|
        on_delete=models.CASCADE,
 | 
						|
        help_text=(
 | 
						|
            'Image file to be used in section. Add multiple plugins '
 | 
						|
            'to add more than one image'
 | 
						|
        )
 | 
						|
    )
 | 
						|
    caption = models.CharField(
 | 
						|
        max_length=100, null=True, blank=True,
 | 
						|
        help_text='Optional caption for the image.'
 | 
						|
    )
 | 
						|
 | 
						|
 | 
						|
class DCLSectionPromoPluginModel(CMSPlugin):
 | 
						|
    background_image = FilerImageField(
 | 
						|
        on_delete=models.CASCADE, null=True, blank=True,
 | 
						|
        help_text=('Optional background image for the Promo Section'),
 | 
						|
        related_name="dcl_section_promo_promo",
 | 
						|
    )
 | 
						|
    heading = models.CharField(
 | 
						|
        blank=True, null=True, max_length=100,
 | 
						|
        help_text='An optional heading for the Promo Section',
 | 
						|
    )
 | 
						|
    subheading = models.CharField(
 | 
						|
        blank=True, null=True, max_length=200,
 | 
						|
        help_text='An optional subheading for the Promo Section',
 | 
						|
    )
 | 
						|
    content = HTMLField()
 | 
						|
    html_id = models.SlugField(
 | 
						|
        blank=True, null=True,
 | 
						|
        help_text=(
 | 
						|
            'An optional html id for the Section. Required to set as target '
 | 
						|
            'of a link on page'
 | 
						|
        )
 | 
						|
    )
 | 
						|
    plain_heading = models.BooleanField(
 | 
						|
        default=False,
 | 
						|
        help_text='Select to keep the heading style simpler.'
 | 
						|
    )
 | 
						|
    text_center = models.BooleanField(
 | 
						|
        default=False,
 | 
						|
        help_text='Select to center align content on small screens.'
 | 
						|
    )
 | 
						|
 | 
						|
    def __str__(self):
 | 
						|
        return '#' + self.html_id if self.html_id else str(self.pk)
 | 
						|
 | 
						|
    def get_extra_classes(self):
 | 
						|
        extra_classes = ''
 | 
						|
        if self.text_center:
 | 
						|
            extra_classes += ' text-center'
 | 
						|
        if self.plain_heading:
 | 
						|
            extra_classes += ' promo-section-plain'
 | 
						|
        if self.background_image:
 | 
						|
            extra_classes += ' promo-with-bg'
 | 
						|
        return extra_classes
 | 
						|
 | 
						|
 | 
						|
class MultipleChoiceArrayField(ArrayField):
 | 
						|
    """
 | 
						|
    A field that allows us to store an array of choices.
 | 
						|
    Uses Django's Postgres ArrayField
 | 
						|
    and a MultipleChoiceField for its formfield.
 | 
						|
    """
 | 
						|
    VMTemplateChoices = []
 | 
						|
    if settings.OPENNEBULA_DOMAIN != 'test_domain':
 | 
						|
        VMTemplateChoices = list(
 | 
						|
            (
 | 
						|
                str(obj.opennebula_vm_template_id),
 | 
						|
                (obj.name + ' - ' + VMTemplate.IPV6.title()
 | 
						|
                    if obj.vm_type == VMTemplate.IPV6 else obj.name
 | 
						|
                 )
 | 
						|
             )
 | 
						|
            for obj in VMTemplate.objects.all()
 | 
						|
        )
 | 
						|
 | 
						|
    def formfield(self, **kwargs):
 | 
						|
        defaults = {
 | 
						|
            'form_class': forms.MultipleChoiceField,
 | 
						|
            'choices': self.VMTemplateChoices,
 | 
						|
        }
 | 
						|
        defaults.update(kwargs)
 | 
						|
        # Skip our parent's formfield implementation completely as we don't
 | 
						|
        # care for it.
 | 
						|
        # pylint:disable=bad-super-call
 | 
						|
        return super(ArrayField, self).formfield(**defaults)
 | 
						|
 | 
						|
 | 
						|
class DCLCalculatorPluginModel(CMSPlugin):
 | 
						|
    pricing = models.ForeignKey(
 | 
						|
        VMPricing,
 | 
						|
        related_name="dcl_custom_pricing_vm_pricing",
 | 
						|
        help_text='Choose a pricing that will be associated with this '
 | 
						|
                  'Calculator'
 | 
						|
    )
 | 
						|
    vm_type = models.CharField(
 | 
						|
        max_length=50, choices=VMTemplate.VM_TYPE_CHOICES,
 | 
						|
        default=VMTemplate.PUBLIC
 | 
						|
    )
 | 
						|
    vm_templates_to_show = MultipleChoiceArrayField(
 | 
						|
        base_field=models.CharField(
 | 
						|
            blank=True,
 | 
						|
            max_length=256,
 | 
						|
        ),
 | 
						|
        default=list,
 | 
						|
        blank=True,
 | 
						|
        help_text="Recommended: If you wish to show all templates of the "
 | 
						|
                  "corresponding VM Type (public/ipv6only), please do not "
 | 
						|
                  "select any of the items in the above field. "
 | 
						|
                  "This will allow any new template(s) added "
 | 
						|
                  "in the backend to be automatically listed in this "
 | 
						|
                  "calculator instance."
 | 
						|
    )
 | 
						|
    default_selected_template = models.CharField(
 | 
						|
        default="Devuan Ascii",
 | 
						|
        null=True,
 | 
						|
        max_length=128,
 | 
						|
        help_text="Write the name of the template that you need selected as"
 | 
						|
                  " default when the calculator loads"
 | 
						|
    )
 |