forum/models/tag.py

   1 import datetime
   2 from base import *
   3
   4 from django.utils.translation import ugettext as _
   5
   6 from forum import modules
   7
   8 class ActiveTagManager(CachedManager):
   9     use_for_related_fields = True
  10
  11     def get_query_set(self):
  12         return super(ActiveTagManager, self).get_query_set().exclude(used_count__lt=1)
  13
  14 class Tag(BaseModel):
  15     name            = models.CharField(max_length=255, unique=True)
  16     created_by      = models.ForeignKey(User, related_name='created_tags')
  17     created_at      = models.DateTimeField(default=datetime.datetime.now, blank=True, null=True)
  18     marked_by       = models.ManyToManyField(User, related_name="marked_tags", through="MarkedTag")
  19     # Denormalised data
  20     used_count = models.PositiveIntegerField(default=0)
  21
  22     active = ActiveTagManager()
  23
  24     class Meta:
  25         ordering = ('-used_count', 'name')
  26         app_label = 'forum'
  27
  28     def __unicode__(self):
  29         return u'%s' % self.name
  30
  31     def add_to_usage_count(self, value):
  32         if self.used_count + value < 0:
  33             self.used_count = 0
  34         else:
  35             self.used_count = models.F('used_count') + value
  36
  37     def cache_key(self):
  38         return self._generate_cache_key(self.name)
  39
  40     @classmethod
  41     def infer_cache_key(cls, querydict):
  42         if 'name' in querydict:
  43             return cls._generate_cache_key(querydict['name'])
  44
  45         return BaseModel.infer_cache_key(querydict)
  46
  47     @classmethod
  48     def value_to_list_on_cache_query(cls):
  49         return 'name'
  50
  51     @models.permalink
  52     def get_absolute_url(self):
  53         return ('tag_questions', (), {'tag': self.name})
  54
  55 class MarkedTag(models.Model):
  56     TAG_MARK_REASONS = (('good', _('interesting')), ('bad', _('ignored')))
  57     tag = models.ForeignKey(Tag, related_name='user_selections')
  58     user = models.ForeignKey(User, related_name='tag_selections')
  59     reason = models.CharField(max_length=16, choices=TAG_MARK_REASONS)
  60
  61     class Meta:
  62         app_label = 'forum'
  63