1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
from django.utils.translation import ugettext as _
from base import *
class Vote(models.Model):
user = models.ForeignKey(User, related_name="votes")
node = models.ForeignKey(Node, related_name="votes")
value = models.SmallIntegerField()
action = models.OneToOneField(Action, related_name="vote")
voted_at = models.DateTimeField(default=datetime.datetime.now)
class Meta:
app_label = 'forum'
unique_together = ('user', 'node')
class Flag(models.Model):
user = models.ForeignKey(User, related_name="flags")
node = models.ForeignKey(Node, related_name="flags")
reason = models.CharField(max_length=300)
action = models.OneToOneField(Action, related_name="flag")
flagged_at = models.DateTimeField(default=datetime.datetime.now)
class Meta:
app_label = 'forum'
unique_together = ('user', 'node')
class Badge(BaseModel):
GOLD = 1
SILVER = 2
BRONZE = 3
type = models.SmallIntegerField()
cls = models.CharField(max_length=50, null=True)
awarded_count = models.PositiveIntegerField(default=0)
awarded_to = models.ManyToManyField(User, through='Award', related_name='badges')
def get_handler(self):
from forum.badges import BadgesMeta
return BadgesMeta.by_id.get(self.id, None)
@property
def name(self):
cls = self.get_handler()
return cls and cls.name or _("Unknown")
@property
def description(self):
cls = self.get_handler()
return cls and cls.description or _("No description available")
@models.permalink
def get_absolute_url(self):
return ('badge', [], {'id': self.id, 'slug': slugify(self.name)})
def save(self, *args, **kwargs):
if isinstance(self.awarded_count, models.expressions.ExpressionNode):
super(Badge, self).save(*args, **kwargs)
self.awarded_count = self.__class__.objects.filter(id=self.id).values_list('awarded_count', flat=True)[0]
else:
super(Badge, self).save(*args, **kwargs)
class Meta:
app_label = 'forum'
class Award(models.Model):
user = models.ForeignKey(User)
badge = models.ForeignKey('Badge', related_name="awards")
node = models.ForeignKey(Node, null=True)
awarded_at = models.DateTimeField(default=datetime.datetime.now)
trigger = models.ForeignKey(Action, related_name="awards", null=True)
action = models.OneToOneField(Action, related_name="award")
class Meta:
unique_together = ('user', 'badge', 'node')
app_label = 'forum'