djangocms-css-background/models.py at master · django-cms-plugins/djangocms-css-background · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
from django.db import models
from cms.models import CMSPlugin
from filer.fields.image import FilerImageField


class CssBackground(CMSPlugin):

    BG_CHOICES = (
        ('image', 'image'),
        ('rgb', 'rgb'),
        ('rgba', 'rgba'),
    )

    bg_type = models.CharField(
        max_length=5, choices=BG_CHOICES, default='image')
    image = FilerImageField(
        null=True, blank=True,
        default=None, verbose_name='image')
    r = models.IntegerField(default=0)
    g = models.IntegerField(default=0)
    b = models.IntegerField(default=0)
    a = models.IntegerField(default=0)

    class Meta:
        verbose_name = "css background"

    def clean(self):
        from django.core.exceptions import ValidationError
        if self.bg_type == 'image' and not self.image:
            raise ValidationError('You must select an image')
        elif self.bg_type == 'rgb' and (
                not self.r or not self.g or not self.b):
            raise ValidationError('You must provide "r", "g" and "b" values')
        elif self.bg_type == 'rgba' and (
                not self.r or not self.g or not self.b):
            raise ValidationError(
                'You must provide "r", "g", "b" and "a" values')

    @property
    def css_background(self):

        if self.bg_type == 'image':

            return ''.join([
                'background-image: url(',
                self.image.url,
                ');'])

        elif self.bg_type == 'rgb':

            return ''.join([
                'background-color: rgb(',
                str(self.r), ',',
                str(self.g), ',',
                str(self.b), ');'])

        elif self.bg_type == 'rgba':

            return ''.join([
                'background-color: rgb(',
                str(self.r), ',',
                str(self.g), ',',
                str(self.b), ',',
                str(self.a), ');'])

        return ''

    def __unicode__(self):
        if self.bg_type == 'image':
            return self.image.label
        elif self.bg_type == 'rgb':
            return u'rgb: %s-%s-%s' % (self.r, self.g, self.b)
        elif self.bg_type == 'rgba':
            return u'rgba: %s-%s-%s-%s' % (self.r, self.g, self.b, self.a)