← Back to branch summary

~widelands-dev/widelands-website/django_staticfiles

« back to all changes in this revision

Viewing changes to djangoratings/models.py

  • Committer: franku
  • Date: 2018-11-18 17:03:56 UTC
  • mto: (508.1.1 widelands)
  • mto: This revision was merged to the branch mainline in revision 510.
  • Revision ID: somal@arcor.de-20181118170356-n9g0g4uv7o0sofa4
cleanup

Show diffs side-by-side

added added

removed removed

Lines of Context:
djangoratings/models.py
1
 
from datetime import datetime
2
 
 
3
 
from django.db import models
4
 
from django.contrib.contenttypes.models import ContentType
5
 
from django.contrib.contenttypes.fields import GenericForeignKey
6
 
from django.contrib.auth.models import User
7
 
 
8
 
try:
9
 
    from django.utils.timezone import now
10
 
except ImportError:
11
 
    now = datetime.now
12
 
 
13
 
from managers import VoteManager, SimilarUserManager
14
 
 
15
 
 
16
 
class Vote(models.Model):
17
 
    content_type = models.ForeignKey(ContentType, related_name='votes')
18
 
    object_id = models.PositiveIntegerField()
19
 
    key = models.CharField(max_length=32)
20
 
    score = models.IntegerField()
21
 
    user = models.ForeignKey(User, blank=True, null=True, related_name='votes')
22
 
    ip_address = models.GenericIPAddressField()
23
 
    cookie = models.CharField(max_length=32, blank=True, null=True)
24
 
    date_added = models.DateTimeField(default=now, editable=False)
25
 
    date_changed = models.DateTimeField(default=now, editable=False)
26
 
 
27
 
    objects = VoteManager()
28
 
 
29
 
    content_object = GenericForeignKey()
30
 
 
31
 
    class Meta:
32
 
        unique_together = (('content_type', 'object_id',
33
 
                            'key', 'user', 'ip_address', 'cookie'))
34
 
 
35
 
    def __unicode__(self):
36
 
        return u"%s voted %s on %s" % (self.user_display, self.score, self.content_object)
37
 
 
38
 
    def save(self, *args, **kwargs):
39
 
        self.date_changed = now()
40
 
        super(Vote, self).save(*args, **kwargs)
41
 
 
42
 
    def user_display(self):
43
 
        if self.user:
44
 
            return '%s (%s)' % (self.user.username, self.ip_address)
45
 
        return self.ip_address
46
 
    user_display = property(user_display)
47
 
 
48
 
    def partial_ip_address(self):
49
 
        ip = self.ip_address.split('.')
50
 
        ip[-1] = 'xxx'
51
 
        return '.'.join(ip)
52
 
    partial_ip_address = property(partial_ip_address)
53
 
 
54
 
 
55
 
class Score(models.Model):
56
 
    content_type = models.ForeignKey(ContentType)
57
 
    object_id = models.PositiveIntegerField()
58
 
    key = models.CharField(max_length=32)
59
 
    score = models.IntegerField()
60
 
    votes = models.PositiveIntegerField()
61
 
 
62
 
    content_object = GenericForeignKey()
63
 
 
64
 
    class Meta:
65
 
        unique_together = (('content_type', 'object_id', 'key'),)
66
 
 
67
 
    def __unicode__(self):
68
 
        return u"%s scored %s with %s votes" % (self.content_object, self.score, self.votes)
69
 
 
70
 
 
71
 
class SimilarUser(models.Model):
72
 
    from_user = models.ForeignKey(User, related_name='similar_users')
73
 
    to_user = models.ForeignKey(User, related_name='similar_users_from')
74
 
    agrees = models.PositiveIntegerField(default=0)
75
 
    disagrees = models.PositiveIntegerField(default=0)
76
 
    exclude = models.BooleanField(default=False)
77
 
 
78
 
    objects = SimilarUserManager()
79
 
 
80
 
    class Meta:
81
 
        unique_together = (('from_user', 'to_user'),)
82
 
 
83
 
    def __unicode__(self):
84
 
        print u"%s %s similar to %s" % (self.from_user, self.exclude and 'is not' or 'is', self.to_user)
85
 
 
86
 
 
87
 
class IgnoredObject(models.Model):
88
 
    user = models.ForeignKey(User)
89
 
    content_type = models.ForeignKey(ContentType)
90
 
    object_id = models.PositiveIntegerField()
91
 
 
92
 
    content_object = GenericForeignKey()
93
 
 
94
 
    class Meta:
95
 
        unique_together = (('content_type', 'object_id'),)
96
 
 
97
 
    def __unicode__(self):
98
 
        return self.content_object