82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
from django.core.urlresolvers import reverse
|
|
from django.contrib.contenttypes import generic
|
|
from django.contrib.comments.models import Comment
|
|
from django.db.models.signals import post_save
|
|
|
|
ALLOW_ALL, REQUIRE_LOGIN, DISALLOW_ALL = range(3)
|
|
SUBSITE_POST_STATUS = (
|
|
(ALLOW_ALL, 'Allow All Posts'),
|
|
(REQUIRE_LOGIN, 'Require Login'),
|
|
(DISALLOW_ALL, 'Allow No Posts'),
|
|
)
|
|
|
|
class Subsite(models.Model):
|
|
slug = models.SlugField(max_length=50, primary_key=True)
|
|
name = models.CharField(max_length=50)
|
|
description = models.TextField()
|
|
|
|
theme = models.CharField(help_text='name of base theme template', max_length=100)
|
|
|
|
ideas_per_page = models.IntegerField(default=10)
|
|
post_status = models.IntegerField(default=ALLOW_ALL, choices=SUBSITE_POST_STATUS)
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('ideas_popular', args=[self.slug])
|
|
|
|
def user_can_post(self, user):
|
|
if self.post_status == DISALLOW_ALL:
|
|
return False
|
|
elif self.post_status == ALLOW_ALL:
|
|
return True
|
|
elif self.post_status == REQUIRE_LOGIN:
|
|
return not user.is_anonymous()
|
|
|
|
class IdeaManager(models.Manager):
|
|
|
|
def with_user_vote(self, user):
|
|
return self.extra(select={'user_vote':'SELECT value FROM brainstorm_vote WHERE idea_id=brainstorm_idea.id AND user_id=%s'}, select_params=[user.id])
|
|
|
|
class Idea(models.Model):
|
|
|
|
title = models.CharField(max_length=100)
|
|
description = models.TextField()
|
|
score = models.IntegerField(default=0)
|
|
|
|
submit_date = models.DateTimeField(auto_now_add=True)
|
|
|
|
user = models.ForeignKey(User, null=True, related_name='ideas')
|
|
subsite = models.ForeignKey(Subsite, related_name='ideas')
|
|
|
|
comments = generic.GenericRelation(Comment, object_id_field='object_pk')
|
|
|
|
objects = IdeaManager()
|
|
|
|
def __unicode__(self):
|
|
return self.title
|
|
|
|
def get_absolute_url(self):
|
|
return reverse('idea_detail', args=[self.subsite_id, self.id])
|
|
|
|
class Vote(models.Model):
|
|
user = models.ForeignKey(User, related_name='idea_votes')
|
|
idea = models.ForeignKey(Idea, related_name='votes')
|
|
value = models.IntegerField()
|
|
timestamp = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __unicode__(self):
|
|
return '%s %s on %s' % (self.user, self.value, self.idea)
|
|
|
|
class Meta:
|
|
unique_together = (('user', 'idea'),)
|
|
|
|
def update_idea_votes(sender, instance, created, **kwargs):
|
|
score = instance.idea.votes.aggregate(score=models.Sum('value'))['score']
|
|
instance.idea.score = score
|
|
instance.idea.save()
|
|
post_save.connect(update_idea_votes, sender=Vote)
|