from django.db import models
from wagtail.fields import RichTextField
from wagtail.models import Page


class LegalIndexPage(Page):
    parent_page_types = ["blog.HomePage"]
    subpage_types = ["legal.LegalPage"]

    def serve(self, request):
        from django.shortcuts import redirect

        return redirect("/")

    def get_sitemap_urls(self, request=None):
        return []


class LegalPage(Page):
    body = RichTextField()
    last_updated = models.DateField()
    show_in_footer = models.BooleanField(default=True)

    parent_page_types = ["legal.LegalIndexPage"]
    subpage_types = []