-import datetime\r
-import hashlib\r
-from urllib import quote_plus, urlencode\r
-from django.db import models, IntegrityError, connection, transaction\r
-from django.utils.http import urlquote as django_urlquote\r
-from django.utils.html import strip_tags\r
-from django.core.urlresolvers import reverse\r
-from django.contrib.auth.models import User\r
-from django.contrib.contenttypes import generic\r
-from django.contrib.contenttypes.models import ContentType\r
-from django.template.defaultfilters import slugify\r
-from django.db.models.signals import post_delete, post_save, pre_save\r
-from django.utils.translation import ugettext as _\r
-from django.utils.safestring import mark_safe\r
-from django.contrib.sitemaps import ping_google\r
-import django.dispatch\r
-from django.conf import settings\r
-import logging\r
-\r
-if settings.USE_SPHINX_SEARCH == True:\r
- from djangosphinx.models import SphinxSearch\r
-\r
-from forum.const import *\r
-\r
-class MetaContent(models.Model):\r
- """\r
- Base class for Vote, Comment and FlaggedItem\r
- """\r
- content_type = models.ForeignKey(ContentType)\r
- object_id = models.PositiveIntegerField()\r
- content_object = generic.GenericForeignKey('content_type', 'object_id')\r
- user = models.ForeignKey(User, related_name='%(class)ss')\r
-\r
- class Meta:\r
- abstract = True\r
- app_label = 'forum'\r
-\r
-\r
-class DeletableContent(models.Model):\r
- deleted = models.BooleanField(default=False)\r
- deleted_at = models.DateTimeField(null=True, blank=True)\r
- deleted_by = models.ForeignKey(User, null=True, blank=True, related_name='deleted_%(class)ss')\r
-\r
- class Meta:\r
- abstract = True\r
- app_label = 'forum'\r
-\r
-\r
-class ContentRevision(models.Model):\r
- """\r
- Base class for QuestionRevision and AnswerRevision\r
- """\r
- revision = models.PositiveIntegerField()\r
- author = models.ForeignKey(User, related_name='%(class)ss')\r
- revised_at = models.DateTimeField()\r
- summary = models.CharField(max_length=300, blank=True)\r
- text = models.TextField()\r
-\r
- class Meta:\r
- abstract = True\r
- app_label = 'forum'\r
-\r
-\r
-class AnonymousContent(models.Model):\r
- """\r
- Base class for AnonymousQuestion and AnonymousAnswer\r
- """\r
- session_key = models.CharField(max_length=40) #session id for anonymous questions\r
- wiki = models.BooleanField(default=False)\r
- added_at = models.DateTimeField(default=datetime.datetime.now)\r
- ip_addr = models.IPAddressField(max_length=21) #allow high port numbers\r
- author = models.ForeignKey(User,null=True)\r
- text = models.TextField()\r
- summary = models.CharField(max_length=180)\r
-\r
- class Meta:\r
- abstract = True\r
- app_label = 'forum'\r
-\r
-\r
-from meta import Comment, Vote, FlaggedItem\r
-\r
-class Content(models.Model):\r
- """\r
- Base class for Question and Answer\r
- """\r
- author = models.ForeignKey(User, related_name='%(class)ss')\r
- added_at = models.DateTimeField(default=datetime.datetime.now)\r
-\r
- wiki = models.BooleanField(default=False)\r
- wikified_at = models.DateTimeField(null=True, blank=True)\r
-\r
- locked = models.BooleanField(default=False)\r
- locked_by = models.ForeignKey(User, null=True, blank=True, related_name='locked_%(class)ss')\r
- locked_at = models.DateTimeField(null=True, blank=True)\r
-\r
- score = models.IntegerField(default=0)\r
- vote_up_count = models.IntegerField(default=0)\r
- vote_down_count = models.IntegerField(default=0)\r
-\r
- comment_count = models.PositiveIntegerField(default=0)\r
- offensive_flag_count = models.SmallIntegerField(default=0)\r
-\r
- last_edited_at = models.DateTimeField(null=True, blank=True)\r
- last_edited_by = models.ForeignKey(User, null=True, blank=True, related_name='last_edited_%(class)ss')\r
-\r
- html = models.TextField()\r
- comments = generic.GenericRelation(Comment)\r
- votes = generic.GenericRelation(Vote)\r
- flagged_items = generic.GenericRelation(FlaggedItem)\r
-\r
- class Meta:\r
- abstract = True\r
- app_label = 'forum'\r
-\r
- def save(self,**kwargs):\r
- super(Content,self).save(**kwargs)\r
- try:\r
- ping_google()\r
- except Exception:\r
- logging.debug('problem pinging google did you register you sitemap with google?')\r
-\r
- def get_object_comments(self):\r
- comments = self.comments.all().order_by('id')\r
- return comments\r
-\r
- def post_get_last_update_info(self):\r
- when = self.added_at\r
- who = self.author\r
- if self.last_edited_at and self.last_edited_at > when:\r
- when = self.last_edited_at\r
- who = self.last_edited_by\r
- comments = self.comments.all()\r
- if len(comments) > 0:\r
- for c in comments:\r
- if c.added_at > when:\r
- when = c.added_at\r
- who = c.user\r
- return when, who
\ No newline at end of file
+import datetime
+import hashlib
+from urllib import quote_plus, urlencode
+from django.db import models, IntegrityError, connection, transaction
+from django.utils.http import urlquote as django_urlquote
+from django.utils.html import strip_tags
+from django.core.urlresolvers import reverse
+from django.contrib.contenttypes import generic
+from django.contrib.contenttypes.models import ContentType
+from django.core.cache import cache
+from django.template.defaultfilters import slugify
+from django.db.models.signals import post_delete, post_save, pre_save, pre_delete
+from django.utils.translation import ugettext as _
+from django.utils.safestring import mark_safe
+from django.contrib.sitemaps import ping_google
+import django.dispatch
+from django.conf import settings
+from forum import const
+import logging
+
+from forum.const import *
+
+class CachedManager(models.Manager):
+ use_for_related_fields = True
+
+ def get(self, *args, **kwargs):
+ try:
+ pk = [v for (k,v) in kwargs.items() if k in ('pk', 'pk__exact', 'id', 'id__exact') or k.endswith('_ptr__pk')][0]
+ except:
+ pk = None
+
+ if pk is not None:
+ key = self.model.cache_key(pk)
+ obj = cache.get(key)
+
+ if obj is None:
+ obj = super(CachedManager, self).get(*args, **kwargs)
+ cache.set(key, obj, 60 * 60)
+
+ return obj
+
+ return super(CachedManager, self).get(*args, **kwargs)
+
+ def get_or_create(self, *args, **kwargs):
+ try:
+ return self.get(*args, **kwargs)
+ except:
+ return super(CachedManager, self).get_or_create(*args, **kwargs)
+
+
+class BaseModel(models.Model):
+ objects = CachedManager()
+
+ class Meta:
+ abstract = True
+ app_label = 'forum'
+
+ def __init__(self, *args, **kwargs):
+ super(BaseModel, self).__init__(*args, **kwargs)
+ self._original_state = dict([(k, v) for k,v in self.__dict__.items() if not k in kwargs])
+
+ @classmethod
+ def cache_key(cls, pk):
+ return '%s.%s:%s' % (settings.APP_URL, cls.__name__, pk)
+
+ def get_dirty_fields(self):
+ missing = object()
+ return dict([(k, self._original_state.get(k, None)) for k,v in self.__dict__.items()
+ if self._original_state.get(k, missing) == missing or self._original_state[k] != v])
+
+ def save(self, *args, **kwargs):
+ super(BaseModel, self).save(*args, **kwargs)
+ self._original_state = dict(self.__dict__)
+ cache.set(self.cache_key(self.pk), self, 86400)
+
+ def delete(self):
+ cache.delete(self.cache_key(self.pk))
+ super(BaseModel, self).delete()
+
+
+class ActiveObjectManager(models.Manager):
+ def get_query_set(self):
+ return super(ActiveObjectManager, self).get_query_set().filter(canceled=False)
+
+class UndeletedObjectManager(models.Manager):
+ def get_query_set(self):
+ return super(UndeletedObjectManager, self).get_query_set().filter(deleted=False)
+
+class GenericContent(BaseModel):
+ """
+ Base class for Vote, Comment and FlaggedItem
+ """
+ content_type = models.ForeignKey(ContentType)
+ object_id = models.PositiveIntegerField()
+ content_object = generic.GenericForeignKey('content_type', 'object_id')
+
+ class Meta:
+ abstract = True
+ app_label = 'forum'
+
+class MetaContent(BaseModel):
+ node = models.ForeignKey('Node', null=True, related_name='%(class)ss')
+
+ def __init__(self, *args, **kwargs):
+ if 'content_object' in kwargs:
+ kwargs['node'] = kwargs['content_object']
+ del kwargs['content_object']
+
+ super (MetaContent, self).__init__(*args, **kwargs)
+
+ @property
+ def content_object(self):
+ return self.node.leaf
+
+ class Meta:
+ abstract = True
+ app_label = 'forum'
+
+from user import User
+
+class UserContent(models.Model):
+ user = models.ForeignKey(User, related_name='%(class)ss')
+
+ class Meta:
+ abstract = True
+ app_label = 'forum'
+
+
+marked_deleted = django.dispatch.Signal(providing_args=["instance", "deleted_by"])
+
+class DeletableContent(models.Model):
+ deleted = models.BooleanField(default=False)
+ deleted_at = models.DateTimeField(null=True, blank=True)
+ deleted_by = models.ForeignKey(User, null=True, blank=True, related_name='deleted_%(class)ss')
+
+ active = UndeletedObjectManager()
+
+ class Meta:
+ abstract = True
+ app_label = 'forum'
+
+ def mark_deleted(self, user):
+ if not self.deleted:
+ self.deleted = True
+ self.deleted_at = datetime.datetime.now()
+ self.deleted_by = user
+ self.save()
+ marked_deleted.send(sender=self.__class__, instance=self, deleted_by=user)
+ return True
+ else:
+ return False
+
+ def unmark_deleted(self):
+ if self.deleted:
+ self.deleted = False
+ self.save()
+ return True
+ else:
+ return False
+
+
+from meta import Comment, Vote, FlaggedItem
+from node import Node, NodeRevision
+
+class QandA(Node):
+ wiki = models.BooleanField(default=False)
+ wikified_at = models.DateTimeField(null=True, blank=True)
+
+ class Meta:
+ abstract = True
+ app_label = 'forum'
+
+ def wikify(self):
+ if not self.wiki:
+ self.wiki = True
+ self.wikified_at = datetime.datetime.now()
+ self.save()
+
+ def save(self, *args, **kwargs):
+ self.__dict__['score'] = self.__dict__['vote_up_count'] - self.__dict__['vote_down_count']
+ super(QandA,self).save(*args, **kwargs)
+
+ try:
+ ping_google()
+ except Exception:
+ logging.debug('problem pinging google did you register you sitemap with google?')
+
+
+
projects / osqa.git / blobdiff