68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
from django.contrib.auth.models import Group
|
|
from django.forms import CheckboxSelectMultiple
|
|
from django.db import models
|
|
from wagtail.admin.panels import FieldPanel
|
|
from wagtail.fields import RichTextField
|
|
from wagtail.models import Page
|
|
from wagtail.models.copying import ParentalManyToManyField
|
|
|
|
|
|
class HomePage(Page):
|
|
body = RichTextField(blank=True)
|
|
|
|
content_panels = Page.content_panels + ["body"]
|
|
|
|
|
|
class CoursePage(Page):
|
|
body = RichTextField(blank=True)
|
|
course_image = models.ForeignKey(
|
|
"wagtailimages.Image",
|
|
null=True,
|
|
blank=True,
|
|
on_delete=models.SET_NULL,
|
|
related_name="+",
|
|
)
|
|
allowed_groups = ParentalManyToManyField(
|
|
Group,
|
|
related_name="course_pages",
|
|
help_text="Select a group to restrict access to this course. Non-members will be prompted to purchase the course to view modules.",
|
|
)
|
|
|
|
def _user_has_access(self, user):
|
|
if not user.is_authenticated:
|
|
return False
|
|
user_group_ids = user.groups.values_list("id", flat=True)
|
|
return self.allowed_groups.filter(id__in=user_group_ids).exists()
|
|
|
|
def get_context(self, request):
|
|
context = super().get_context(request)
|
|
context["user_has_access"] = self._user_has_access(request.user)
|
|
return context
|
|
|
|
content_panels = Page.content_panels + [
|
|
FieldPanel("course_image"),
|
|
FieldPanel("body"),
|
|
FieldPanel("allowed_groups", widget=CheckboxSelectMultiple),
|
|
]
|
|
|
|
|
|
class CourseModulePage(Page):
|
|
body = RichTextField(blank=True)
|
|
|
|
@property
|
|
def course(self):
|
|
if hasattr(self, "get_parent"):
|
|
parent = self.get_parent()
|
|
if parent and hasattr(parent, "specific"):
|
|
return parent.specific
|
|
return None
|
|
|
|
@property
|
|
def full_title(self):
|
|
course = self.course
|
|
if course:
|
|
return f"{course.title} - {self.title}"
|
|
return self.title
|
|
|
|
content_panels = Page.content_panels + ["body"]
|