import datetime
from django.db import models

import markdown

PUB_STATUS = (
    (0, 'Draft'),
    (1, 'Published'),
)


def markdown_processor(md):
    return markdown.markdown(md, ['footnotes'], safe_mode=False)


class Experiment(models.Model):
    title = models.CharField(max_length=254)
    slug = models.SlugField()
    date_created = models.DateField(auto_now_add=True)
    date_start = models.DateField()
    date_end_actual = models.DateField(blank=True, null=True)
    date_end_projected = models.DateField(blank=True)
    duration = models.PositiveSmallIntegerField(default=30)
    status = models.IntegerField(choices=PUB_STATUS, default=0)
    body_html = models.TextField(blank=True)
    body_markdown = models.TextField()

    class Meta:
        app_label = 'projects'
        ordering = ('-date_start',)

    # Returns the string representation of the model.
    def __str__(self):
        return self.title

    def get_absolute_url(self):
        return "/projects/experiments/%s/%s/" % (self.pub_date.strftime("%Y").lower(), self.slug)

    def days_remaining(self):
        return self.date_end_projected - datetime.date.today()

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

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

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

    def save(self):
        self.body_html = markdown_processor(self.body_markdown)
        self.date_end_projected = self.date_start + datetime.timedelta(self.duration)
        super(Experiment, self).save()