Begin implementing the CMS functions

This commit is contained in:
2022-07-21 13:50:36 +01:00
parent 3ff8cd6166
commit c5e811878b
4 changed files with 35 additions and 16 deletions

View File

@@ -72,6 +72,7 @@ class Session(models.Model):
class ContentBlock(models.Model):
user = models.ForeignKey(User, on_delete=models.PROTECT)
position = models.IntegerField()
page = models.CharField(max_length=255, null=True, blank=True)
title = models.CharField(max_length=255, null=True, blank=True)
column1 = models.TextField(null=True, blank=True)
column2 = models.TextField(null=True, blank=True)
@@ -80,15 +81,24 @@ class ContentBlock(models.Model):
image2 = models.CharField(max_length=255, null=True, blank=True)
image3 = models.CharField(max_length=255, null=True, blank=True)
def __str__(self):
return f"[{self.position}] {self.page} {self.title}"
def save(self, *args, **kwargs):
"""
Override the save function to blank fields.
"""
if self.column1 == "NONE":
if self.column1 == "":
self.column1 = None
if self.column2 == "NONE":
if self.column2 == "":
self.column2 = None
if self.column3 == "NONE":
if self.column3 == "":
self.column3 = None
if self.image1 == "":
self.image1 = None
if self.image2 == "":
self.image2 = None
if self.image3 == "":
self.image3 = None
super().save(*args, **kwargs)