path: root/app/posts/models.py

import datetime
import os

from django.dispatch import receiver
from django.contrib.gis.db import models
from django.db.models.signals import post_save
from django.contrib.contenttypes.fields import GenericRelation, GenericForeignKey
from django.contrib.contenttypes.models import ContentType
from django.urls import reverse
from django.utils.functional import cached_property
from django.apps import apps
from django.conf import settings
from django.contrib.sitemaps import Sitemap
from django import forms

import urllib.request
import urllib.parse
import urllib.error
from django_gravatar.helpers import get_gravatar_url, has_gravatar, calculate_gravatar_hash
from django_comments.signals import comment_was_posted
from django_comments.models import Comment
from django_comments.moderation import CommentModerator, moderator

from taggit.managers import TaggableManager

from normalize.models import RelatedPost
from photos.models import LuxImage, LuxImageSize
from locations.models import Location
from books.models import Book
from fieldnotes.models import FieldNote
from taxonomy.models import TaggedItems, Category
from jrnl.models import Entry
from utils.util import render_images, render_products, parse_video, markdown_to_html, extract_main_image


class PostType(models.IntegerChoices):
    FIELD_TEST = 0, ('field test')
    REVIEW = 1, ('review')
    ESSAY = 2, ('essay')
    SRC = 3, ('src')
    JRNL = 4, ('jrnl')
    FIELD_NOTE = 5, ('field note')


class Post(models.Model):
    old_id = models.IntegerField(blank=True, null=True)
    title = models.CharField(max_length=200)
    short_title = models.CharField(max_length=200, blank=True, null=True)
    subtitle = models.CharField(max_length=200, blank=True)
    slug = models.SlugField(unique_for_date='pub_date')
    prologue_markdown = models.TextField(blank=True, null=True)
    prologue_html = models.TextField(blank=True, null=True)
    body_markdown = models.TextField()
    body_html = models.TextField(blank=True)
    epilogue_markdown = models.TextField(blank=True, null=True)
    epilogue_html = models.TextField(blank=True, null=True)
    dek = models.TextField(null=True, blank=True)
    meta_description = models.CharField(max_length=256, blank=True)
    pub_date = models.DateTimeField('Date published')
    last_updated = models.DateTimeField(auto_now=True)
    enable_comments = models.BooleanField(default=False)
    PUB_STATUS = (
        (0, 'Draft'),
        (1, 'Published'),
    )
    status = models.IntegerField(choices=PUB_STATUS, default=0)
    featured_image = models.ForeignKey(LuxImage, on_delete=models.CASCADE, null=True, blank=True)
    TEMPLATES = (
        (0, 'single'),
    )
    post_type = models.IntegerField(choices=PostType.choices, default=PostType.JRNL)
    template_name = models.IntegerField(choices=TEMPLATES, default=0)
    has_video = models.BooleanField(blank=True, default=False)
    has_code = models.BooleanField(blank=True, default=False)
    disclaimer = models.BooleanField(blank=True, default=True)
    books = models.ManyToManyField(Book, blank=True)
    field_notes = models.ManyToManyField(FieldNote, blank=True)
    related = models.ManyToManyField(RelatedPost, blank=True)
    point = models.PointField(null=True, blank=True)
    location = models.ForeignKey(Location, on_delete=models.CASCADE, null=True, blank=True)
    topics = models.ManyToManyField(Category, blank=True)
    originally_published_by = models.CharField(max_length=400, null=True, blank=True)
    originally_published_by_url = models.CharField(max_length=400, null=True, blank=True)
    field_notes = models.ManyToManyField(FieldNote, blank=True)
    books = models.ManyToManyField(Book, blank=True)
    jrnl = models.ManyToManyField(Entry, blank=True)

    class Meta:
        ordering = ('-pub_date',)
        get_latest_by = 'pub_date'

    def __str__(self):
        return self.title

    def get_absolute_url(self):
        if self.post_type == 0:
            return reverse('guides:reviews:review-detail', kwargs={"slug": self.slug})
        if self.post_type == 1:
            return reverse('guides:reviews:review-detail', kwargs={"slug": self.slug})
        if self.post_type == 2:
            return reverse('essays:detail', kwargs={"slug": self.slug})
        if self.post_type == 3:
            return reverse('src:detail', kwargs={"slug": self.slug})
        if self.post_type == 5:
            return reverse('fieldnote:detail', kwargs={"year": self.pub_date.year, "month": self.pub_date.strftime("%m"), "slug": self.slug})

    def comment_period_open(self):
        return self.enable_comments and datetime.datetime.today() - datetime.timedelta(30) <= self.pub_date

    def get_featured_image_thumb(self):
        return self.featured_image.get_image_by_size("tn")

    def get_content_type(self):
        return ContentType.objects.get(app_label="posts", model="post")

    @property
    def get_previous_published(self):
        return self.get_previous_by_pub_date(status__exact=1,post_type=self.post_type)

    @property
    def get_previous_admin_url(self):
        n = self.get_previous_by_pub_date()
        return reverse('admin:%s_%s_change' %(self._meta.app_label,  self._meta.model_name),  args=[n.id] )

    @property
    def get_next_published(self):
        return self.get_next_by_pub_date(status__exact=1,post_type=self.post_type)

    @property
    def get_next_admin_url(self):
        model = apps.get_model(app_label=self._meta.app_label, model_name=self._meta.model_name)
        try:
            return reverse('admin:%s_%s_change' %(self._meta.app_label,  self._meta.model_name),  args=[self.get_next_by_pub_date().pk] )
        except model.DoesNotExist:
            return ''

    @property
    def longitude(self):
        '''Get the site's longitude.'''
        return self.point.x

    @property
    def latitude(self):
        '''Get the site's latitude.'''
        return self.point.y

    @property
    def sitemap_priority(self):
        if self.post_type in [0,1,4,5]:
            return 1.0
        else:
            return 0.7

    def save(self, *args, **kwargs):
        created = self.pk is None
        if not created:
            md = render_images(self.body_markdown)
            prods = render_products(md)
            self.body_html = markdown_to_html(prods)
            if self.epilogue_html:
                self.epilogue_html = markdown_to_html(self.epilogue_markdown)
            if self.prologue_html:
                self.prologue_html = markdown_to_html(self.prologue_markdown)
            self.has_video = parse_video(self.body_html)
        if self.point:
            try:
                self.location = Location.objects.filter(geometry__contains=self.point).get()
            except Location.DoesNotExist:
                raise forms.ValidationError("There is no location associated with that point, add it:  %sadmin/locations/location/add/" % (settings.BASE_URL))
        if created and not self.featured_image:
            if self.post_type == PostType.FIELD_NOTE:
                self.featured_image = extract_main_image(self.body_markdown)
            else:
                self.featured_image = LuxImage.objects.latest()
        old = type(self).objects.get(pk=self.pk) if self.pk else None
        if old and old.featured_image != self.featured_image:  # Field has changed
            if self.featured_image:
                s = LuxImageSize.objects.get(name="featured_jrnl")
                ss = LuxImageSize.objects.get(name="picwide-med")
                self.featured_image.sizes.add(s)
                self.featured_image.sizes.add(ss)
                self.featured_image.save()
        if old and old.title != self.title or old and old.slug != self.slug:
            related, c = RelatedPost.objects.get_or_create(model_name=self.get_content_type(), entry_id = self.id, pub_date=self.pub_date)
            related.title = self.title
            related.slug = self.slug
            related.save()
        super(Post, self).save(*args, **kwargs)


class PostModerator(CommentModerator):
    '''
    Moderate everything except people with multiple approvals
    '''
    email_notification = True

    def moderate(self, comment, content_object, request):
        previous_approvals = Comment.objects.filter(user_email=comment.email, is_public=True)
        for approval in previous_approvals:
            if approval.submit_date <= datetime.datetime.today() - datetime.timedelta(21):
                approve = True
        if previous_approvals.count() > 2 and approve:
            return False
            # do entry build right here so it goes to live site
        return True
moderator.register(Post, PostModerator)


@receiver(comment_was_posted, sender=Comment)
def cache_gravatar(sender, comment, **kwargs):
    gravatar_exists = has_gravatar(comment.email)
    grav_dir = settings.IMAGES_ROOT + '/gravcache/'
    if gravatar_exists:
        url = get_gravatar_url(comment.email, size=60)
        if not os.path.isdir(grav_dir):
            os.makedirs(grav_dir)
        local_grav = '%s/%s.jpg' % (grav_dir, calculate_gravatar_hash(comment.email))
        urllib.request.urlretrieve(url, local_grav)


@receiver(post_save, sender=Post)
def post_save_events(sender, update_fields, created, instance, **kwargs):
    related, created = RelatedPost.objects.get_or_create(model_name=instance.get_content_type(), entry_id = instance.id, pub_date=instance.pub_date, title=instance.title, slug=instance.slug)
    post_save.disconnect(post_save_events, sender=Post)
    instance.save()
    post_save.connect(post_save_events, sender=Post)


class PostSitemap(Sitemap):
    changefreq = "never"
    protocol = "https"

    def items(self):
        return Post.objects.filter(status=1)

    def lastmod(self, obj):
        return obj.pub_date

    def priority(self, obj):
        return obj.sitemap_priority


"""
for p in src: 
    s, created = Post.objects.get_or_create( 
    old_id=p.id,
    title=p.title, 
    slug=p.slug, 
    body_markdown=p.body_markdown, 
    pub_date=p.pub_date,
    enable_comments=p.enable_comments, 
    has_code=p.has_code,
    status=p.status, 
    meta_description=p.meta_description, 
    post_type=3, 
    ) 
    print(p) 
    for t in p.topics.all(): 
        c,created = Category.objects.get_or_create(
             slug=t.slug,
             name=t.name,
             pluralized_name=t.pluralized_name
        ) 
        s.topics.add(c)
"""