forum/actions/node.py

   1 from django.utils.html import strip_tags\r
   2 from django.utils.translation import ugettext as _\r
   3 from forum.models.action import ActionProxy\r
   4 from forum.models import Comment, Question, Answer, NodeRevision\r
   5 \r
   6 class NodeEditAction(ActionProxy):\r
   7     def create_revision_data(self, initial=False, **data):\r
   8         revision_data = dict(summary=data.get('summary', (initial and _('Initial revision') or '')), body=data['text'])\r
   9 \r
  10         if data.get('title', None):\r
  11             revision_data['title'] = strip_tags(data['title'].strip())\r
  12 \r
  13         if data.get('tags', None):\r
  14             revision_data['tagnames'] = data['tags'].strip()\r
  15 \r
  16         return revision_data\r
  17 \r
  18 class AskAction(NodeEditAction):\r
  19     verb = _("asked")\r
  20 \r
  21     def process_data(self, **data):\r
  22         question = Question(author=self.user, **self.create_revision_data(True, **data))\r
  23         question.save()\r
  24         self.node = question\r
  25 \r
  26     def describe(self, viewer=None):\r
  27         return _("%(user)s asked %(question)s") % {\r
  28             'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)),\r
  29             'question': self.hyperlink(self.node.get_absolute_url(), self.node.title)\r
  30         }\r
  31 \r
  32 class AnswerAction(NodeEditAction):\r
  33     verb = _("answered")\r
  34 \r
  35     def process_data(self, **data):\r
  36         answer = Answer(author=self.user, parent=data['question'], **self.create_revision_data(True, **data))\r
  37         answer.save()\r
  38         self.node = answer\r
  39 \r
  40     def process_action(self):\r
  41         self.node.question.reset_answer_count_cache()\r
  42 \r
  43     def describe(self, viewer=None):\r
  44         question = self.node.parent\r
  45         return _("%(user)s answered %(asker)s %(question)s") % {\r
  46             'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)),\r
  47             'asker': self.hyperlink(question.author.get_profile_url(), self.friendly_username(viewer, question.author)),\r
  48             'question': self.hyperlink(self.node.get_absolute_url(), question.title)\r
  49         }\r
  50 \r
  51 class CommentAction(ActionProxy):\r
  52     verb = _("commented")\r
  53 \r
  54     def process_data(self, text='', parent=None):\r
  55         comment = Comment(author=self.user, parent=parent, body=text)\r
  56         comment.save()\r
  57         self.node = comment\r
  58 \r
  59     def describe(self, viewer=None):\r
  60         return _("%(user)s commented on %(post_desc)s") % {\r
  61             'user': self.hyperlink(self.node.author.get_profile_url(), self.friendly_username(viewer, self.node.author)),\r
  62             'post_desc': self.describe_node(viewer, self.node.parent)\r
  63         }\r
  64 \r
  65 class ReviseAction(NodeEditAction):\r
  66     verb = _("edited")\r
  67 \r
  68     def process_data(self, **data):\r
  69         revision_data = self.create_revision_data(**data)\r
  70         revision = self.node.create_revision(self.user, action=self, **revision_data)\r
  71         self.extra = revision.revision\r
  72 \r
  73     def describe(self, viewer=None):\r
  74         return _("%(user)s edited %(post_desc)s") % {\r
  75             'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)),\r
  76             'post_desc': self.describe_node(viewer, self.node)\r
  77         }\r
  78 \r
  79     def get_absolute_url(self):\r
  80         return self.node.get_revisions_url()\r
  81 \r
  82 class RetagAction(ActionProxy):\r
  83     verb = _("retagged")\r
  84 \r
  85     def process_data(self, tagnames=''):\r
  86         active = self.node.active_revision\r
  87         revision_data = dict(summary=_('Retag'), title=active.title, tagnames=strip_tags(tagnames.strip()), body=active.body)\r
  88         self.node.create_revision(self.user, action=self, **revision_data)\r
  89 \r
  90     def describe(self, viewer=None):\r
  91         return _("%(user)s retagged %(post_desc)s") % {\r
  92             'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)),\r
  93             'post_desc': self.describe_node(viewer, self.node)\r
  94         }\r
  95 \r
  96     def get_absolute_url(self):\r
  97         return self.node.get_revisions_url()\r
  98 \r
  99 class RollbackAction(ActionProxy):\r
 100     verb = _("reverted")\r
 101 \r
 102     def process_data(self, activate=None):\r
 103         previous = self.node.active_revision\r
 104         self.node.activate_revision(self.user, activate, self)\r
 105         self.extra = "%d:%d" % (previous.revision, activate.revision)\r
 106 \r
 107     def describe(self, viewer=None):\r
 108         revisions = [NodeRevision.objects.get(node=self.node, revision=int(n)) for n in self.extra.split(':')]\r
 109 \r
 110         return _("%(user)s reverted %(post_desc)s from revision %(initial)d (%(initial_sum)s) to revision %(final)d (%(final_sum)s)") % {\r
 111             'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)),\r
 112             'post_desc': self.describe_node(viewer, self.node),\r
 113             'initial': revisions[0].revision, 'initial_sum': revisions[0].summary,\r
 114             'final': revisions[1].revision, 'final_sum': revisions[1].summary,\r
 115         }\r
 116 \r
 117     def get_absolute_url(self):\r
 118         return self.node.get_revisions_url()\r
 119 \r
 120 class CloseAction(ActionProxy):\r
 121     verb = _("closed")\r
 122 \r
 123     def process_action(self):\r
 124         self.node.extra_action = self\r
 125         self.node.marked = True\r
 126         self.node.save()\r
 127 \r
 128     def cancel_action(self):\r
 129         self.node.extra_action = None\r
 130         self.node.marked = False\r
 131         self.node.save()\r
 132 \r
 133     def describe(self, viewer=None):\r
 134         return _("%(user)s closed %(post_desc)s: %(reason)s") % {\r
 135             'user': self.hyperlink(self.user.get_profile_url(), self.friendly_username(viewer, self.user)),\r
 136             'post_desc': self.describe_node(viewer, self.node),\r
 137             'reason': self.extra\r
 138         }