Add DCLCalculatorPluginModel

This commit is contained in:
PCoder 2018-04-09 21:18:18 +02:00
parent d07cc41d0a
commit c7afbb32c0

View file

@ -6,6 +6,8 @@ from django.utils.safestring import mark_safe
from djangocms_text_ckeditor.fields import HTMLField from djangocms_text_ckeditor.fields import HTMLField
from filer.fields.image import FilerImageField from filer.fields.image import FilerImageField
from datacenterlight.models import VMPricing
class CMSIntegration(models.Model): class CMSIntegration(models.Model):
name = models.CharField( name = models.CharField(
@ -275,3 +277,12 @@ class DCLSectionPromoPluginModel(CMSPlugin):
if self.background_image: if self.background_image:
extra_classes += ' promo-with-bg' extra_classes += ' promo-with-bg'
return extra_classes return extra_classes
class DCLCalculatorPluginModel(DCLSectionPluginModel):
pricing = models.ForeignKey(
VMPricing,
default=VMPricing.get_default_pricing(),
help_text='Choose a pricing that will be associated with this '
'Calculator'
)