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)