summaryrefslogtreecommitdiff
path: root/cms/models.py
blob: 05b761fa687824d8d206227539ae39f0805affd3 (plain)
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
import os

from django.core.exceptions import ValidationError
from django.core.files.storage import FileSystemStorage
from django.db import models


# Create your models here.

class Nav(models.Model):
    name = models.CharField(max_length=20)
    link = models.CharField(max_length=20)
    position = models.IntegerField()

    def __str__(self):
        return self.name


class SubNav(models.Model):
    nav = models.ForeignKey(Nav, on_delete=models.CASCADE)
    name = models.CharField(max_length=20)
    link = models.CharField(max_length=100)
    position = models.IntegerField()

    def __str__(self):
        return self.name


class Page(models.Model):
    permalink = models.CharField(max_length=100, unique=True)
    title = models.CharField(max_length=50)
    imports = models.TextField(help_text='External imports like css,js files, will be placed in <head> tag (already '
                                         'includes bootstrap4 and jQuery)', null=True, blank=True)
    content = models.TextField(help_text='Body of the page')
    pub_date = models.DateTimeField('date published', auto_now_add=True)

    def __str__(self):
        return self.title


def get_filename(instance, _):
    return 'static/cms/' + str(instance.filename)


def validate_filename(value):
    if os.path.exists('workshop_app/static/' + value):
        raise ValidationError('Static file with that name already exists! Please choose a unique name. You may use '
                              'foldername/filename to upload to a folder')


class StaticFile(models.Model):
    filename = models.CharField(max_length=70, unique=True, validators=[validate_filename])
    file = models.FileField(upload_to=get_filename, storage=FileSystemStorage(location='workshop_app', base_url='/'),
                            blank=False,
                            help_text='Please upload static file (image, css, js, etc). This file will be accessible '
                                      'at static/cms/filename')

    def __str__(self):
        return self.filename