from django.db import models

import tagging

from tagging.fields import TagField

from agdj.blog.managers import PublicManager

from agdj.utils import cached_property, dictproperty

class Entry(models.Model):

    "A blog entry"

    id = models.AutoField(primary_key=True)

    title = models.CharField(max_length=64)

    source = models.TextField()

    description = models.TextField()

    pub_date = models.DateTimeField(auto_now_add=True)

    slug = models.CharField(max_length=64, unique=True)

    public = models.BooleanField()

    SLUG_ONLY = "slug-only"

    DATE_BASED = "date-based"

    URL_SCHEME_CHOICES = (

        (SLUG_ONLY, "Slug Only"),

        (DATE_BASED, "Date-based"),

        )

    feed_url_scheme = models.CharField(

        max_length=64, choices=URL_SCHEME_CHOICES,

        help_text="Does this post use the new date-based URL or the old "

        "slug-based one in feed URLs and GUIDs?",

        default=DATE_BASED)

    # This is a hack that even I don't understand

    text_tags = TagField(db_column="tags")

    objects = models.Manager()

    public_objects = PublicManager()

    class Meta:

        db_table = u'agus_blog_entry'

        ordering = ("-pub_date",)

        verbose_name_plural = u'Entries'

    def __unicode__(self):

        return self.title

    def save(self, *args, **kwargs):

        from django.core.cache import cache

        # If this is an update, then clear the cache keys.

        if self.pk:

            cache_key = "%s.%s.%s(%s)" % (self.__class__.__module__,

                                          self.__class__.__name__,

                                          "rendered",

                                          self.pk)

            cache.delete(cache_key)

        return super(Entry, self).save(*args, **kwargs)

    @dictproperty

    def urls(self, name):

        from django.core.urlresolvers import reverse

        if name == "feed_url":

            if self.feed_url_scheme == self.DATE_BASED:

                return self.urls.view

            else:

                return reverse("redirect-to-date-url", args=[self.slug])

        elif name == "view":

                return reverse("view-blog-post",

                               args=[self.pub_date.year,

                                     self.pub_date.strftime("%b").lower(),

                                     self.pub_date.day,

                                     self.slug])

    @cached_property

    def rendered(self):

        from agdj.markup import markdown

        return markdown(self.source)

    @property

    def public_comments(self):

        return self.comments.filter(public=True)

    @property

    def spam_count(self):

        return Comment.with_unpublished.filter(

            public=False, entry=self).count()

tagging.register(Entry)

class Comment(models.Model):

    "A comment on a blog entry"

    id = models.AutoField(primary_key=True)

    ip = models.TextField(blank=True)

    name = models.CharField(max_length=64, blank=True)

    url = models.CharField(max_length=128, blank=True)

    public = models.BooleanField(default=False)

    entry = models.ForeignKey(Entry, related_name="comments", db_column="agus_blog_entry_entry_id")

    pub_date = models.DateTimeField(auto_now_add=True)

    text = models.TextField(blank=True)

    # This should be the _default_manager

    objects = PublicManager()

    with_unpublished = models.Manager()

    class Meta:

        db_table = u'agus_blog_comment'

        ordering = ("-pub_date",)

    def __unicode__(self):

        return u"Comment %d on '%s'" % (self.id, self.entry)

    @property

    def rendered(self):

        from agdj.markup import markdown_untrusted

        return markdown_untrusted(self.text)

    def save(self, *args, **kwargs):

        # I'm not sure why - maybe because the database was originally

        # created by SQLAlchemy and not syncdb - but the AutoField

        # doesn't really work.

        if self.id is None:

            from django.db import connection

            cursor = connection.cursor()

            cursor.execute("select nextval('agus_blog_comment_id_seq');")

            self.id = cursor.fetchone()[0]

        super(self.__class__, self).save(*args, **kwargs)