Dont crash when we have no topics in a forum

2012-02-25 12:46:07 -05:00
parent e0b19fb4bc
commit bc90ed04f9
1 changed files with 8 additions and 2 deletions
--- a/forums/models.py
+++ b/forums/models.py
@@ -43,7 +43,10 @@ class Forum(MPTTModel):
        return self.topic_set.count()

    def latestTopic(self):
-        return self.topic_set.extra(order_by = ['created'])[0]
+        try:
+            return self.topic_set.extra(order_by = ['created'])[0]
+        except IndexError, e:
+            return None

    @models.permalink
    def get_absolute_url(self):
@@ -69,7 +72,10 @@ class Topic(models.Model):
        super(Topic, self).save(*args, **kwargs)

    def lastPost(self):
-        return self.rootPost.get_descendants(True).extra(order_by = ['updated'])[0]
+        try:
+            return self.rootPost.get_descendants(True).extra(order_by = ['updated'])[0]
+        except IndexError, e:
+            return None

    @models.permalink
    def get_absolute_url(self):