##// END OF EJS Templates
Changed tag popularity algorithm again a bit, now popularity is relative
neko259 -
r599:f8272ced default
parent child Browse files
Show More
@@ -1,91 +1,88 b''
1 from boards.models import Thread
1 from boards.models import Thread, Post
2 2 from django.db import models
3 3 from django.db.models import Count
4 4
5 5 __author__ = 'neko259'
6 6
7 7 MAX_TAG_FONT = 1
8 8 MIN_TAG_FONT = 0.2
9 9
10 10 OPENING_POST_POPULARITY = 0.1
11 11 REPLY_POPULARITY = 0.005
12 12 ARCHIVE_POPULARITY_MODIFIER = 0.1
13 13
14 14
15 15 class TagManager(models.Manager):
16 16
17 17 def get_not_empty_tags(self):
18 18 tags = self.annotate(Count('threads')) \
19 19 .filter(threads__count__gt=0).order_by('name')
20 20
21 21 return tags
22 22
23 23
24 24 class Tag(models.Model):
25 25 """
26 26 A tag is a text node assigned to the thread. The tag serves as a board
27 27 section. There can be multiple tags for each thread
28 28 """
29 29
30 30 objects = TagManager()
31 31
32 32 class Meta:
33 33 app_label = 'boards'
34 34
35 35 name = models.CharField(max_length=100)
36 36 threads = models.ManyToManyField(Thread, null=True,
37 37 blank=True, related_name='tag+')
38 38 linked = models.ForeignKey('Tag', null=True, blank=True)
39 39
40 40 def __unicode__(self):
41 41 return self.name
42 42
43 43 def is_empty(self):
44 44 return self.get_post_count() == 0
45 45
46 46 def get_post_count(self):
47 47 return self.threads.count()
48 48
49 49 def get_popularity(self):
50 popularity = 0.0
50 all_post_count = Post.objects.all().count()
51
52 tag_reply_count = 0.0
51 53
52 54 for thread in self.threads.all():
53 reply_count = thread.get_reply_count()
55 tag_reply_count += thread.get_reply_count()
54 56
55 thread_popularity = 0.0
56 thread_popularity += REPLY_POPULARITY * reply_count
57 thread_popularity += OPENING_POST_POPULARITY
58 if thread.archived:
59 thread_popularity *= ARCHIVE_POPULARITY_MODIFIER
60 popularity += thread_popularity
57 popularity = tag_reply_count / all_post_count
61 58
62 59 return popularity
63 60
64 61 def get_linked_tags(self):
65 62 tag_list = []
66 63 self.get_linked_tags_list(tag_list)
67 64
68 65 return tag_list
69 66
70 67 def get_linked_tags_list(self, tag_list=[]):
71 68 """
72 69 Returns the list of tags linked to current. The list can be got
73 70 through returned value or tag_list parameter
74 71 """
75 72
76 73 linked_tag = self.linked
77 74
78 75 if linked_tag and not (linked_tag in tag_list):
79 76 tag_list.append(linked_tag)
80 77
81 78 linked_tag.get_linked_tags_list(tag_list)
82 79
83 80 def get_font_value(self):
84 81 """Get tag font value to differ most popular tags in the list"""
85 82
86 83 popularity = self.get_popularity()
87 84
88 85 font_value = MIN_TAG_FONT + popularity
89 86 font_value = min(font_value, MAX_TAG_FONT)
90 87
91 88 return str(font_value)
General Comments 0
You need to be logged in to leave comments. Login now