+# -*- coding: utf-8 -*-
+
from base import *
import re
from tag import Tag
import markdown
+from django.utils.encoding import smart_unicode
from django.utils.translation import ugettext as _
from django.utils.safestring import mark_safe
from django.utils.html import strip_tags
-from forum.utils.userlinking import auto_user_link
from forum.utils.html import sanitize_html
+from forum.utils.userlinking import auto_user_link
+from forum.settings import SUMMARY_LENGTH
from utils import PickledObjectField
class NodeContent(models.Model):
def html(self):
return self.body
+ def rendered(self, content):
+ return auto_user_link(self, self._as_markdown(content, *['auto_linker']))
+
@classmethod
def _as_markdown(cls, content, *extensions):
try:
return ''
def as_markdown(self, *extensions):
- return self._as_markdown(self.body, *extensions)
+ return self._as_markdown(smart_unicode(self.body), *extensions)
@property
def headline(self):
- return self.title
+ title = self.title
+
+ # Replaces multiple spaces with single ones.
+ title = re.sub(' +',' ', title)
+
+ return title
def tagname_list(self):
if self.tagnames:
- t = [name.strip() for name in self.tagnames.split(u' ') if name]
- return [name.strip() for name in self.tagnames.split(u' ') if name]
+ return [name.strip() for name in self.tagnames.split() if name]
else:
return []
class NodeManager(CachedManager):
use_for_related_fields = True
- def get_query_set(self):
+ def get_queryset(self):
qs = NodeQuerySet(self.model)
+ # If the node is an answer, question or comment we filter the Node model by type
if self.model is not Node:
- return qs.filter(node_type=self.model.get_type())
- else:
- return qs
+ qs = qs.filter(node_type=self.model.get_type())
+
+ return qs
def get_for_types(self, types, *args, **kwargs):
kwargs['node_type__in'] = [t.get_type() for t in types]
marked = models.BooleanField(default=False)
comment_count = DenormalizedField("children", node_type="comment", canceled=False)
- flag_count = DenormalizedField("flags")
+ flag_count = DenormalizedField("actions", action_type="flag", canceled=False)
friendly_name = _("post")
@classmethod
def _generate_cache_key(cls, key, group="node"):
return super(Node, cls)._generate_cache_key(key, group)
-
+
@classmethod
def get_type(cls):
return cls.__name__.lower()
@property
def summary(self):
- return strip_tags(self.html)[:300]
+ content = strip_tags(self.html)
+
+ # Remove multiple spaces.
+ content = re.sub(' +',' ', content)
+
+ # Replace line breaks with a space, we don't need them at all.
+ content = content.replace("\n", ' ')
+
+ # Truncate and all an ellipsis if length greater than summary length.
+ content = (content[:SUMMARY_LENGTH] + '...') if len(content) > SUMMARY_LENGTH else content
+
+ return content
+
+ # Can be used to block subscription notifications for a specific node from a module
+ def _is_notifiable(self):
+ return True
+
+ @property
+ def is_notifiable(self):
+ return self._is_notifiable()
@models.permalink
def get_revisions_url(self):
def create_revision(self, user, **kwargs):
number = self.revisions.aggregate(last=models.Max('revision'))['last'] + 1
revision = self._create_revision(user, number, **kwargs)
- self.activate_revision(user, revision, extensions=['urlize'])
+ self.activate_revision(user, revision)
return revision
- def activate_revision(self, user, revision, extensions=['urlize']):
+ def activate_revision(self, user, revision):
self.title = revision.title
self.tagnames = revision.tagnames
- self.body = auto_user_link(self, self._as_markdown(revision.body, *extensions))
+
+ self.body = self.rendered(revision.body)
self.active_revision = revision
- self.update_last_activity(user)
+
+ # Try getting the previous revision
+ try:
+ prev_revision = NodeRevision.objects.get(node=self, revision=revision.revision-1)
+
+ update_activity = True
+
+ # Do not update the activity if only the tags are changed
+ if prev_revision.title == revision.title and prev_revision.body == revision.body \
+ and prev_revision.tagnames != revision.tagnames and not settings.UPDATE_LATEST_ACTIVITY_ON_TAG_EDIT:
+ update_activity = False
+ except NodeRevision.DoesNotExist:
+ update_activity = True
+ finally:
+ if update_activity:
+ self.update_last_activity(user)
self.save()
+ def get_active_users(self, active_users = None):
+ if not active_users:
+ active_users = set()
+
+ active_users.add(self.author)
+
+ for node in self.children.all():
+ if not node.nis.deleted:
+ node.get_active_users(active_users)
+
+ return active_users
+
+ def get_last_edited(self):
+ if not self.last_edited:
+ try:
+ le = self.actions.exclude(action_type__in=('voteup', 'votedown', 'flag'), canceled=True).order_by('-action_date')[0]
+ self.last_edited = le
+ self.save()
+ except:
+ pass
+
+ return self.last_edited
+
+
def _list_changes_in_tags(self):
dirty = self.get_dirty_fields()
return None
else:
if self._original_state['tagnames']:
- old_tags = set(name for name in self._original_state['tagnames'].split(u' '))
+ old_tags = set(self._original_state['tagnames'].split())
else:
old_tags = set()
- new_tags = set(name for name in self.tagnames.split(u' ') if name)
+ new_tags = set(self.tagnames.split())
return dict(
current=list(new_tags),
for name in tag_changes['added']:
try:
tag = Tag.objects.get(name=name)
- except:
+ except Tag.DoesNotExist:
tag = Tag.objects.create(name=name, created_by=self._last_active_user())
if not self.nis.deleted:
tag.save()
def delete(self, *args, **kwargs):
+ for tag in self.tags.all():
+ tag.add_to_usage_count(-1)
+ tag.save()
+
self.active_revision = None
self.save()
tags_changed = self._process_changes_in_tags()
super(Node, self).save(*args, **kwargs)
-
- if tags_changed: self.tags = list(Tag.objects.filter(name__in=self.tagname_list()))
+ if tags_changed:
+ if self.tagnames.strip():
+ self.tags = list(Tag.objects.filter(name__in=self.tagname_list()))
+ else:
+ self.tags = []
class Meta:
app_label = 'forum'