Start implementing basic CMS functionality for main page

core/models.py

@@ -67,3 +67,28 @@ class Session(models.Model):
     session = models.CharField(max_length=255, null=True, blank=True)
     subscription_id = models.CharField(max_length=255, null=True, blank=True)
     plan = models.ForeignKey(Plan, null=True, blank=True, on_delete=models.CASCADE)
+
+
+class ContentBlock(models.Model):
+    user = models.ForeignKey(User, on_delete=models.PROTECT)
+    position = models.IntegerField()
+    title = models.CharField(max_length=255, null=True, blank=True)
+    column1 = models.TextField(null=True, blank=True)
+    column2 = models.TextField(null=True, blank=True)
+    column3 = models.TextField(null=True, blank=True)
+    image1 = models.CharField(max_length=255, null=True, blank=True)
+    image2 = models.CharField(max_length=255, null=True, blank=True)
+    image3 = models.CharField(max_length=255, null=True, blank=True)
+
+    def save(self, *args, **kwargs):
+        """
+        Override the save function to  blank fields.
+        """
+        if self.column1 == "NONE":
+            self.column1 = None
+        if self.column2 == "NONE":
+            self.column2 = None
+        if self.column3 == "NONE":
+            self.column3 = None
+
+        super().save(*args, **kwargs)