from wagtail.wagtailcore.models import Page
from django.db import models
from wagtail.wagtailadmin.edit_handlers import FieldPanel, MultiFieldPanel
from wagtail.wagtailimages.edit_handlers import ImageChooserPanel
from wagtailmetadata.models import MetadataPageMixin
from bs4 import BeautifulSoup


class Entity(MetadataPageMixin, Page):
    is_home = False

    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)
    post_date = models.DateTimeField(null=True, blank=True)

    promote_panels = [
        MultiFieldPanel([
            FieldPanel('slug'),
            FieldPanel('seo_title'),
            FieldPanel('post_date'),
            FieldPanel('search_description'),
            ImageChooserPanel('search_image'),
        ], 'Common page configuration'),
    ]

    @property
    def image(self):
        return self.search_image

    @property
    def short_body(self):
        body_words = BeautifulSoup(str(self.body)).get_text().split(' ')
        return ' '.join(body_words[:30]) + '...'  # limit to 30 words (ish)

    def get_meta_description(self):
        return self.search_description or self.short_body

    class Meta:
        abstract = True