|
|
from django.template.loader import render_to_string
|
|
|
from boards.models import Thread, Post
|
|
|
from django.db import models
|
|
|
from django.db.models import Count, Sum
|
|
|
from django.core.urlresolvers import reverse
|
|
|
from boards.models.base import Viewable
|
|
|
|
|
|
__author__ = 'neko259'
|
|
|
|
|
|
MAX_TAG_FONT = 1
|
|
|
MIN_TAG_FONT = 0.2
|
|
|
|
|
|
TAG_POPULARITY_MULTIPLIER = 20
|
|
|
|
|
|
ARCHIVE_POPULARITY_MODIFIER = 0.5
|
|
|
|
|
|
|
|
|
class TagManager(models.Manager):
|
|
|
|
|
|
def get_not_empty_tags(self):
|
|
|
"""
|
|
|
Gets tags that have non-archived threads.
|
|
|
"""
|
|
|
|
|
|
tags = self.annotate(Count('threads')) \
|
|
|
.filter(threads__count__gt=0).order_by('name')
|
|
|
|
|
|
return tags
|
|
|
|
|
|
|
|
|
class Tag(models.Model, Viewable):
|
|
|
"""
|
|
|
A tag is a text node assigned to the thread. The tag serves as a board
|
|
|
section. There can be multiple tags for each thread
|
|
|
"""
|
|
|
|
|
|
objects = TagManager()
|
|
|
|
|
|
class Meta:
|
|
|
app_label = 'boards'
|
|
|
ordering = ('name',)
|
|
|
|
|
|
name = models.CharField(max_length=100, db_index=True)
|
|
|
threads = models.ManyToManyField(Thread, null=True,
|
|
|
blank=True, related_name='tag+')
|
|
|
linked = models.ForeignKey('Tag', null=True, blank=True)
|
|
|
|
|
|
def __unicode__(self):
|
|
|
return self.name
|
|
|
|
|
|
def is_empty(self):
|
|
|
"""
|
|
|
Checks if the tag has some threads.
|
|
|
"""
|
|
|
|
|
|
return self.get_thread_count() == 0
|
|
|
|
|
|
def get_thread_count(self):
|
|
|
return self.threads.count()
|
|
|
|
|
|
def get_popularity(self):
|
|
|
"""
|
|
|
Gets tag's popularity value as a percentage of overall board post
|
|
|
count.
|
|
|
"""
|
|
|
|
|
|
all_post_count = Post.objects.count()
|
|
|
|
|
|
tag_reply_count = 0.0
|
|
|
|
|
|
tag_reply_count += self.get_post_count()
|
|
|
tag_reply_count +=\
|
|
|
self.get_post_count(archived=True) * ARCHIVE_POPULARITY_MODIFIER
|
|
|
|
|
|
popularity = tag_reply_count / all_post_count
|
|
|
|
|
|
return popularity
|
|
|
|
|
|
def get_linked_tags(self):
|
|
|
"""
|
|
|
Gets tags linked to the current one.
|
|
|
"""
|
|
|
|
|
|
tag_list = []
|
|
|
self.get_linked_tags_list(tag_list)
|
|
|
|
|
|
return tag_list
|
|
|
|
|
|
def get_linked_tags_list(self, tag_list=[]):
|
|
|
"""
|
|
|
Returns the list of tags linked to current. The list can be got
|
|
|
through returned value or tag_list parameter
|
|
|
"""
|
|
|
|
|
|
linked_tag = self.linked
|
|
|
|
|
|
if linked_tag and not (linked_tag in tag_list):
|
|
|
tag_list.append(linked_tag)
|
|
|
|
|
|
linked_tag.get_linked_tags_list(tag_list)
|
|
|
|
|
|
def get_font_value(self):
|
|
|
"""
|
|
|
Gets tag font value to differ most popular tags in the list
|
|
|
"""
|
|
|
|
|
|
popularity = self.get_popularity()
|
|
|
|
|
|
font_value = popularity * Tag.objects.get_not_empty_tags().count()
|
|
|
font_value = max(font_value, MIN_TAG_FONT)
|
|
|
font_value = min(font_value, MAX_TAG_FONT)
|
|
|
|
|
|
return str(font_value)
|
|
|
|
|
|
def get_post_count(self, archived=False):
|
|
|
"""
|
|
|
Gets posts count for the tag's threads.
|
|
|
"""
|
|
|
|
|
|
posts_count = 0
|
|
|
|
|
|
threads = self.threads.filter(archived=archived)
|
|
|
if threads.exists():
|
|
|
posts_count = threads.annotate(posts_count=Count('replies')).aggregate(
|
|
|
posts_sum=Sum('posts_count'))['posts_sum']
|
|
|
|
|
|
if not posts_count:
|
|
|
posts_count = 0
|
|
|
|
|
|
return posts_count
|
|
|
|
|
|
def get_url(self):
|
|
|
return reverse('tag', kwargs={'tag_name': self.name})
|
|
|
|
|
|
def get_view(self, *args, **kwargs):
|
|
|
return render_to_string('boards/tag.html', {
|
|
|
'tag': self,
|
|
|
})
|
|
|
|