forum/models/base.py

   1 import datetime
   2 from urllib import quote_plus, urlencode
   3 from django.db import models, IntegrityError, connection, transaction
   4 from django.utils.http import urlquote  as django_urlquote
   5 from django.utils.html import strip_tags
   6 from django.core.urlresolvers import reverse
   7 from django.contrib.contenttypes import generic
   8 from django.contrib.contenttypes.models import ContentType
   9 from django.core.cache import cache
  10 from django.template.defaultfilters import slugify
  11 from django.db.models.signals import post_delete, post_save, pre_save, pre_delete
  12 from django.utils.translation import ugettext as _
  13 from django.utils.safestring import mark_safe
  14 from django.contrib.sitemaps import ping_google
  15 import django.dispatch
  16 from django.conf import settings
  17 from forum import const
  18 import logging
  19
  20 from forum.const import *
  21
  22 class CachedManager(models.Manager):
  23     use_for_related_fields = True
  24
  25     def get(self, *args, **kwargs):
  26         try:
  27             pk = [v for (k,v) in kwargs.items() if k in ('pk', 'pk__exact', 'id', 'id__exact') or k.endswith('_ptr__pk')][0]
  28         except:
  29             pk = None
  30
  31         if pk is not None:
  32             key = self.model.cache_key(pk)
  33             obj = cache.get(key)
  34
  35             if obj is None:
  36                 obj = super(CachedManager, self).get(*args, **kwargs)
  37                 cache.set(key, obj, 60 * 60)
  38
  39             return obj
  40
  41         return super(CachedManager, self).get(*args, **kwargs)
  42
  43     def get_or_create(self, *args, **kwargs):
  44         try:
  45             return self.get(*args, **kwargs)
  46         except:
  47             return super(CachedManager, self).get_or_create(*args, **kwargs)
  48
  49
  50 class BaseModel(models.Model):
  51     objects = CachedManager()
  52
  53     class Meta:
  54         abstract = True
  55         app_label = 'forum'
  56
  57     def __init__(self, *args, **kwargs):
  58         super(BaseModel, self).__init__(*args, **kwargs)
  59         self._original_state = dict([(k, v) for k,v in self.__dict__.items() if not k in kwargs])
  60
  61     @classmethod
  62     def cache_key(cls, pk):
  63         return '%s.%s:%s' % (settings.APP_URL, cls.__name__, pk)
  64
  65     def get_dirty_fields(self):
  66         missing = object()
  67         return dict([(k, self._original_state.get(k, None)) for k,v in self.__dict__.items()
  68                  if self._original_state.get(k, missing) == missing or self._original_state[k] != v])
  69
  70     def save(self, *args, **kwargs):
  71         super(BaseModel, self).save(*args, **kwargs)
  72         self._original_state = dict(self.__dict__)
  73         cache.set(self.cache_key(self.pk), self, 86400)
  74
  75     def delete(self):
  76         cache.delete(self.cache_key(self.pk))
  77         super(BaseModel, self).delete()
  78
  79
  80 class ActiveObjectManager(models.Manager):
  81     use_for_related_fields = True
  82     def get_query_set(self):
  83         return super(ActiveObjectManager, self).get_query_set().filter(canceled=False)
  84
  85 class UndeletedObjectManager(models.Manager):
  86     def get_query_set(self):
  87         return super(UndeletedObjectManager, self).get_query_set().filter(deleted=False)
  88
  89 class GenericContent(BaseModel):
  90     """
  91         Base class for Vote, Comment and FlaggedItem
  92     """
  93     content_type   = models.ForeignKey(ContentType)
  94     object_id      = models.PositiveIntegerField()
  95     content_object = generic.GenericForeignKey('content_type', 'object_id')
  96
  97     class Meta:
  98         abstract = True
  99         app_label = 'forum'
 100
 101 class MetaContent(BaseModel):
 102     node = models.ForeignKey('Node', null=True, related_name='%(class)ss')
 103
 104     def __init__(self, *args, **kwargs):
 105         if 'content_object' in kwargs:
 106             kwargs['node'] = kwargs['content_object']
 107             del kwargs['content_object']
 108
 109         super (MetaContent, self).__init__(*args, **kwargs)
 110
 111     @property
 112     def content_object(self):
 113         return self.node.leaf
 114
 115     class Meta:
 116         abstract = True
 117         app_label = 'forum'
 118
 119 from user import User
 120
 121 class UserContent(models.Model):
 122     user = models.ForeignKey(User, related_name='%(class)ss')
 123
 124     class Meta:
 125         abstract = True
 126         app_label = 'forum'
 127
 128
 129 marked_deleted = django.dispatch.Signal(providing_args=["instance", "deleted_by"])
 130
 131 class DeletableContent(models.Model):
 132     deleted     = models.BooleanField(default=False)
 133     deleted_at  = models.DateTimeField(null=True, blank=True)
 134     deleted_by  = models.ForeignKey(User, null=True, blank=True, related_name='deleted_%(class)ss')
 135
 136     active = UndeletedObjectManager()
 137
 138     class Meta:
 139         abstract = True
 140         app_label = 'forum'
 141
 142     def mark_deleted(self, user):
 143         if not self.deleted:
 144             self.deleted = True
 145             self.deleted_at = datetime.datetime.now()
 146             self.deleted_by = user
 147             self.save()
 148             marked_deleted.send(sender=self.__class__, instance=self, deleted_by=user)
 149             return True
 150         else:
 151             return False
 152
 153     def unmark_deleted(self):
 154         if self.deleted:
 155             self.deleted = False
 156             self.save()
 157             return True
 158         else:
 159             return False
 160
 161 mark_canceled = django.dispatch.Signal(providing_args=['instance'])
 162
 163 class CancelableContent(models.Model):
 164     canceled = models.BooleanField(default=False)
 165
 166     def cancel(self):
 167         if not self.canceled:
 168             self.canceled = True
 169             self.save()
 170             mark_canceled.send(sender=self.__class__, instance=self)
 171             return True
 172
 173         return False
 174
 175     class Meta:
 176         abstract = True
 177         app_label = 'forum'
 178
 179
 180 from node import Node, NodeRevision
 181
 182 class QandA(Node):
 183     wiki                 = models.BooleanField(default=False)
 184     wikified_at          = models.DateTimeField(null=True, blank=True)
 185
 186     class Meta:
 187         abstract = True
 188         app_label = 'forum'
 189
 190     def wikify(self):
 191         if not self.wiki:
 192             self.wiki = True
 193             self.wikified_at = datetime.datetime.now()
 194             self.save()
 195
 196
 197
 198