addons/website_forum/models/forum.py

   1 # -*- coding: utf-8 -*-
   2
   3 from datetime import datetime
   4 import math
   5 import uuid
   6 from werkzeug.exceptions import Forbidden
   7
   8 from openerp import _
   9 from openerp import api, fields, models
  10 from openerp import modules
  11 from openerp import tools
  12 from openerp import SUPERUSER_ID
  13 from openerp.addons.website.models.website import slug
  14 from openerp.exceptions import Warning
  15
  16
  17 class KarmaError(Forbidden):
  18     """ Karma-related error, used for forum and posts. """
  19     pass
  20
  21
  22 class Forum(models.Model):
  23     _name = 'forum.forum'
  24     _description = 'Forum'
  25     _inherit = ['mail.thread', 'website.seo.metadata']
  26
  27     def init(self, cr):
  28         """ Add forum uuid for user email validation.
  29
  30         TDE TODO: move me somewhere else, auto_init ? """
  31         forum_uuids = self.pool['ir.config_parameter'].search(cr, SUPERUSER_ID, [('key', '=', 'website_forum.uuid')])
  32         if not forum_uuids:
  33             self.pool['ir.config_parameter'].set_param(cr, SUPERUSER_ID, 'website_forum.uuid', str(uuid.uuid4()), ['base.group_system'])
  34
  35     @api.model
  36     def _get_default_faq(self):
  37         fname = modules.get_module_resource('website_forum', 'data', 'forum_default_faq.html')
  38         with open(fname, 'r') as f:
  39             return f.read()
  40         return False
  41
  42     # description and use
  43     name = fields.Char('Forum Name', required=True, translate=True)
  44     faq = fields.Html('Guidelines', default=_get_default_faq, translate=True)
  45     description = fields.Html(
  46         'Description',
  47         default='<p> This community is for professionals and enthusiasts of our products and services.'
  48                 'Share and discuss the best content and new marketing ideas,'
  49                 'build your professional profile and become a better marketer together.</p>')
  50     default_order = fields.Selection([
  51         ('create_date desc', 'Newest'),
  52         ('write_date desc', 'Last Updated'),
  53         ('vote_count desc', 'Most Voted'),
  54         ('relevancy desc', 'Relevancy'),
  55         ('child_count desc', 'Answered')],
  56         string='Default Order', required=True, default='write_date desc')
  57     relevancy_post_vote = fields.Float('First Relevancy Parameter', default=0.8)
  58     relevancy_time_decay = fields.Float('Second Relevancy Parameter', default=1.8)
  59     default_post_type = fields.Selection([
  60         ('question', 'Question'),
  61         ('discussion', 'Discussion'),
  62         ('link', 'Link')],
  63         string='Default Post', required=True, default='question')
  64     allow_question = fields.Boolean('Questions', help="Users can answer only once per question. Contributors can edit answers and mark the right ones.", default=True)
  65     allow_discussion = fields.Boolean('Discussions', default=False)
  66     allow_link = fields.Boolean('Links', help="When clicking on the post, it redirects to an external link", default=False)
  67     # karma generation
  68     karma_gen_question_new = fields.Integer(string='Asking a question', default=2)
  69     karma_gen_question_upvote = fields.Integer(string='Question upvoted', default=5)
  70     karma_gen_question_downvote = fields.Integer(string='Question downvoted', default=-2)
  71     karma_gen_answer_upvote = fields.Integer(string='Answer upvoted', default=10)
  72     karma_gen_answer_downvote = fields.Integer(string='Answer downvoted', default=-2)
  73     karma_gen_answer_accept = fields.Integer(string='Accepting an answer', default=2)
  74     karma_gen_answer_accepted = fields.Integer(string='Answer accepted', default=15)
  75     karma_gen_answer_flagged = fields.Integer(string='Answer flagged', default=-100)
  76     # karma-based actions
  77     karma_ask = fields.Integer(string='Ask a new question', default=3)
  78     karma_answer = fields.Integer(string='Answer a question', default=3)
  79     karma_edit_own = fields.Integer(string='Edit its own posts', default=1)
  80     karma_edit_all = fields.Integer(string='Edit all posts', default=300)
  81     karma_close_own = fields.Integer(string='Close its own posts', default=100)
  82     karma_close_all = fields.Integer(string='Close all posts', default=500)
  83     karma_unlink_own = fields.Integer(string='Delete its own posts', default=500)
  84     karma_unlink_all = fields.Integer(string='Delete all posts', default=1000)
  85     karma_upvote = fields.Integer(string='Upvote', default=5)
  86     karma_downvote = fields.Integer(string='Downvote', default=50)
  87     karma_answer_accept_own = fields.Integer(string='Accept an answer on its own questions', default=20)
  88     karma_answer_accept_all = fields.Integer(string='Accept an answers to all questions', default=500)
  89     karma_editor_link_files = fields.Integer(string='Linking files (Editor)', default=20)
  90     karma_editor_clickable_link = fields.Integer(string='Add clickable links (Editor)', default=20)
  91     karma_comment_own = fields.Integer(string='Comment its own posts', default=1)
  92     karma_comment_all = fields.Integer(string='Comment all posts', default=1)
  93     karma_comment_convert_own = fields.Integer(string='Convert its own answers to comments and vice versa', default=50)
  94     karma_comment_convert_all = fields.Integer(string='Convert all answers to answers and vice versa', default=500)
  95     karma_comment_unlink_own = fields.Integer(string='Unlink its own comments', default=50)
  96     karma_comment_unlink_all = fields.Integer(string='Unlinnk all comments', default=500)
  97     karma_retag = fields.Integer(string='Change question tags', default=75)
  98     karma_flag = fields.Integer(string='Flag a post as offensive', default=500)
  99
 100     @api.model
 101     def create(self, values):
 102         return super(Forum, self.with_context(mail_create_nolog=True)).create(values)
 103
 104
 105 class Post(models.Model):
 106     _name = 'forum.post'
 107     _description = 'Forum Post'
 108     _inherit = ['mail.thread', 'website.seo.metadata']
 109     _order = "is_correct DESC, vote_count DESC, write_date DESC"
 110
 111     name = fields.Char('Title')
 112     forum_id = fields.Many2one('forum.forum', string='Forum', required=True)
 113     content = fields.Html('Content')
 114     content_link = fields.Char('URL', help="URL of Link Articles")
 115     tag_ids = fields.Many2many('forum.tag', 'forum_tag_rel', 'forum_id', 'forum_tag_id', string='Tags')
 116     state = fields.Selection([('active', 'Active'), ('close', 'Close'), ('offensive', 'Offensive')], string='Status', default='active')
 117     views = fields.Integer('Number of Views', default=0)
 118     active = fields.Boolean('Active', default=True)
 119     post_type = fields.Selection([
 120         ('question', 'Question'),
 121         ('link', 'Article'),
 122         ('discussion', 'Discussion')],
 123         string='Type', default='question')
 124     website_message_ids = fields.One2many(
 125         'mail.message', 'res_id',
 126         domain=lambda self: ['&', ('model', '=', self._name), ('type', 'in', ['email', 'comment'])],
 127         string='Post Messages', help="Comments on forum post",
 128     )
 129     # history
 130     create_date = fields.Datetime('Asked on', select=True, readonly=True)
 131     create_uid = fields.Many2one('res.users', string='Created by', select=True, readonly=True)
 132     write_date = fields.Datetime('Update on', select=True, readonly=True)
 133     write_uid = fields.Many2one('res.users', string='Updated by', select=True, readonly=True)
 134     relevancy = fields.Float('Relevancy', compute="_compute_relevancy", store=True)
 135
 136     @api.one
 137     @api.depends('vote_count', 'forum_id.relevancy_post_vote', 'forum_id.relevancy_time_decay')
 138     def _compute_relevancy(self):
 139         days = (datetime.today() - datetime.strptime(self.create_date, tools.DEFAULT_SERVER_DATETIME_FORMAT)).days
 140         self.relevancy = math.copysign(1, self.vote_count) * (abs(self.vote_count - 1) ** self.forum_id.relevancy_post_vote / (days + 2) ** self.forum_id.relevancy_time_decay)
 141
 142     # vote
 143     vote_ids = fields.One2many('forum.post.vote', 'post_id', string='Votes')
 144     user_vote = fields.Integer('My Vote', compute='_get_user_vote')
 145     vote_count = fields.Integer('Votes', compute='_get_vote_count', store=True)
 146
 147     @api.multi
 148     def _get_user_vote(self):
 149         votes = self.env['forum.post.vote'].search_read([('post_id', 'in', self._ids), ('user_id', '=', self._uid)], ['vote', 'post_id'])
 150         mapped_vote = dict([(v['post_id'][0], v['vote']) for v in votes])
 151         for vote in self:
 152             vote.user_vote = mapped_vote.get(vote.id, 0)
 153
 154     @api.multi
 155     @api.depends('vote_ids')
 156     def _get_vote_count(self):
 157         read_group_res = self.env['forum.post.vote'].read_group([('post_id', 'in', self._ids)], ['post_id', 'vote'], ['post_id', 'vote'], lazy=False)
 158         result = dict.fromkeys(self._ids, 0)
 159         for data in read_group_res:
 160             result[data['post_id'][0]] += data['__count'] * int(data['vote'])
 161         for post in self:
 162             post.vote_count = result[post.id]
 163
 164     # favorite
 165     favourite_ids = fields.Many2many('res.users', string='Favourite')
 166     user_favourite = fields.Boolean('Is Favourite', compute='_get_user_favourite')
 167     favourite_count = fields.Integer('Favorite Count', compute='_get_favorite_count', store=True)
 168
 169     @api.one
 170     def _get_user_favourite(self):
 171         self.user_favourite = self._uid in self.favourite_ids.ids
 172
 173     @api.one
 174     @api.depends('favourite_ids')
 175     def _get_favorite_count(self):
 176         self.favourite_count = len(self.favourite_ids)
 177
 178     # hierarchy
 179     is_correct = fields.Boolean('Correct', help='Correct answer or answer accepted')
 180     parent_id = fields.Many2one('forum.post', string='Question', ondelete='cascade')
 181     self_reply = fields.Boolean('Reply to own question', compute='_is_self_reply', store=True)
 182     child_ids = fields.One2many('forum.post', 'parent_id', string='Answers')
 183     child_count = fields.Integer('Number of answers', compute='_get_child_count', store=True)
 184     uid_has_answered = fields.Boolean('Has Answered', compute='_get_uid_has_answered')
 185     has_validated_answer = fields.Boolean('Is answered', compute='_get_has_validated_answer', store=True)
 186
 187     @api.multi
 188     @api.depends('create_uid', 'parent_id')
 189     def _is_self_reply(self):
 190         self_replies = self.search([('parent_id.create_uid', '=', self._uid)])
 191         for post in self:
 192             post.is_self_reply = post in self_replies
 193
 194     @api.one
 195     @api.depends('child_ids')
 196     def _get_child_count(self):
 197         self.child_count = len(self.child_ids)
 198
 199     @api.one
 200     def _get_uid_has_answered(self):
 201         self.uid_has_answered = any(answer.create_uid.id == self._uid for answer in self.child_ids)
 202
 203     @api.multi
 204     @api.depends('child_ids', 'is_correct')
 205     def _get_has_validated_answer(self):
 206         correct_posts = [ans.parent_id for ans in self.search([('parent_id', 'in', self._ids), ('is_correct', '=', True)])]
 207         for post in self:
 208             post.is_correct = post in correct_posts
 209
 210     # closing
 211     closed_reason_id = fields.Many2one('forum.post.reason', string='Reason')
 212     closed_uid = fields.Many2one('res.users', string='Closed by', select=1)
 213     closed_date = fields.Datetime('Closed on', readonly=True)
 214     # karma calculation and access
 215     karma_accept = fields.Integer('Convert comment to answer', compute='_get_post_karma_rights')
 216     karma_edit = fields.Integer('Karma to edit', compute='_get_post_karma_rights')
 217     karma_close = fields.Integer('Karma to close', compute='_get_post_karma_rights')
 218     karma_unlink = fields.Integer('Karma to unlink', compute='_get_post_karma_rights')
 219     karma_comment = fields.Integer('Karma to comment', compute='_get_post_karma_rights')
 220     karma_comment_convert = fields.Integer('Karma to convert comment to answer', compute='_get_post_karma_rights')
 221     can_ask = fields.Boolean('Can Ask', compute='_get_post_karma_rights')
 222     can_answer = fields.Boolean('Can Answer', compute='_get_post_karma_rights')
 223     can_accept = fields.Boolean('Can Accept', compute='_get_post_karma_rights')
 224     can_edit = fields.Boolean('Can Edit', compute='_get_post_karma_rights')
 225     can_close = fields.Boolean('Can Close', compute='_get_post_karma_rights')
 226     can_unlink = fields.Boolean('Can Unlink', compute='_get_post_karma_rights')
 227     can_upvote = fields.Boolean('Can Upvote', compute='_get_post_karma_rights')
 228     can_downvote = fields.Boolean('Can Downvote', compute='_get_post_karma_rights')
 229     can_comment = fields.Boolean('Can Comment', compute='_get_post_karma_rights')
 230     can_comment_convert = fields.Boolean('Can Convert to Comment', compute='_get_post_karma_rights')
 231
 232     @api.one
 233     def _get_post_karma_rights(self):
 234         user = self.env.user
 235
 236         self.karma_accept = self.parent_id and self.parent_id.create_uid.id == self._uid and self.forum_id.karma_answer_accept_own or self.forum_id.karma_answer_accept_all
 237         self.karma_edit = self.create_uid.id == self._uid and self.forum_id.karma_edit_own or self.forum_id.karma_edit_all
 238         self.karma_close = self.create_uid.id == self._uid and self.forum_id.karma_close_own or self.forum_id.karma_close_all
 239         self.karma_unlink = self.create_uid.id == self._uid and self.forum_id.karma_unlink_own or self.forum_id.karma_unlink_all
 240         self.karma_comment = self.create_uid.id == self._uid and self.forum_id.karma_comment_own or self.forum_id.karma_comment_all
 241         self.karma_comment_convert = self.create_uid.id == self._uid and self.forum_id.karma_comment_convert_own or self.forum_id.karma_comment_convert_all
 242
 243         self.can_ask = user.karma >= self.forum_id.karma_ask
 244         self.can_answer = user.karma >= self.forum_id.karma_answer
 245         self.can_accept = user.karma >= self.karma_accept
 246         self.can_edit = user.karma >= self.karma_edit
 247         self.can_close = user.karma >= self.karma_close
 248         self.can_unlink = user.karma >= self.karma_unlink
 249         self.can_upvote = user.karma >= self.forum_id.karma_upvote
 250         self.can_downvote = user.karma >= self.forum_id.karma_downvote
 251         self.can_comment = user.karma >= self.karma_comment
 252         self.can_comment_convert = user.karma >= self.karma_comment_convert
 253
 254     @api.model
 255     def create(self, vals):
 256         post = super(Post, self.with_context(mail_create_nolog=True)).create(vals)
 257         # deleted or closed questions
 258         if post.parent_id and (post.parent_id.state == 'close' or post.parent_id.active is False):
 259             raise Warning(_('Posting answer on a [Deleted] or [Closed] question is not possible'))
 260         # karma-based access
 261         if not post.parent_id and not post.can_ask:
 262             raise KarmaError('Not enough karma to create a new question')
 263         elif post.parent_id and not post.can_answer:
 264             raise KarmaError('Not enough karma to answer to a question')
 265         # messaging and chatter
 266         base_url = self.env['ir.config_parameter'].get_param('web.base.url')
 267         if post.parent_id:
 268             body = _(
 269                 '<p>A new answer for <i>%s</i> has been posted. <a href="%s/forum/%s/question/%s">Click here to access the post.</a></p>' %
 270                 (post.parent_id.name, base_url, slug(post.parent_id.forum_id), slug(post.parent_id))
 271             )
 272             post.parent_id.message_post(subject=_('Re: %s') % post.parent_id.name, body=body, subtype='website_forum.mt_answer_new')
 273         else:
 274             body = _(
 275                 '<p>A new question <i>%s</i> has been asked on %s. <a href="%s/forum/%s/question/%s">Click here to access the question.</a></p>' %
 276                 (post.name, post.forum_id.name, base_url, slug(post.forum_id), slug(post))
 277             )
 278             post.message_post(subject=post.name, body=body, subtype='website_forum.mt_question_new')
 279             self.env.user.sudo().add_karma(post.forum_id.karma_gen_question_new)
 280         return post
 281
 282     @api.multi
 283     def write(self, vals):
 284         if 'state' in vals:
 285             if vals['state'] in ['active', 'close'] and any(not post.can_close for post in self):
 286                 raise KarmaError('Not enough karma to close or reopen a post.')
 287         if 'active' in vals:
 288             if any(not post.can_unlink for post in self):
 289                 raise KarmaError('Not enough karma to delete or reactivate a post')
 290         if 'is_correct' in vals:
 291             if any(not post.can_accept for post in self):
 292                 raise KarmaError('Not enough karma to accept or refuse an answer')
 293             # update karma except for self-acceptance
 294             mult = 1 if vals['is_correct'] else -1
 295             for post in self:
 296                 if vals['is_correct'] != post.is_correct and post.create_uid.id != self._uid:
 297                     post.create_uid.sudo().add_karma(post.forum_id.karma_gen_answer_accepted * mult)
 298                     self.env.user.sudo().add_karma(post.forum_id.karma_gen_answer_accept * mult)
 299         if any(key not in ['state', 'active', 'is_correct', 'closed_uid', 'closed_date', 'closed_reason_id'] for key in vals.keys()) and any(not post.can_edit for post in self):
 300             raise KarmaError('Not enough karma to edit a post.')
 301
 302         res = super(Post, self).write(vals)
 303         # if post content modify, notify followers
 304         if 'content' in vals or 'name' in vals:
 305             for post in self:
 306                 if post.parent_id:
 307                     body, subtype = _('Answer Edited'), 'website_forum.mt_answer_edit'
 308                     obj_id = post.parent_id
 309                 else:
 310                     body, subtype = _('Question Edited'), 'website_forum.mt_question_edit'
 311                     obj_id = post
 312                 obj_id.message_post(body=body, subtype=subtype)
 313         return res
 314
 315     @api.multi
 316     def close(self, reason_id):
 317         if any(post.parent_id for post in self):
 318             return False
 319         self.write({
 320             'state': 'close',
 321             'closed_uid': self._uid,
 322             'closed_date': datetime.today().strftime(tools.DEFAULT_SERVER_DATETIME_FORMAT),
 323             'closed_reason_id': reason_id,
 324         })
 325         return True
 326
 327     @api.multi
 328     def unlink(self):
 329         if any(not post.can_unlink for post in self):
 330             raise KarmaError('Not enough karma to unlink a post')
 331         # if unlinking an answer with accepted answer: remove provided karma
 332         for post in self:
 333             if post.is_correct:
 334                 post.create_uid.sudo().add_karma(post.forum_id.karma_gen_answer_accepted * -1)
 335                 self.env.user.sudo().add_karma(post.forum_id.karma_gen_answer_accepted * -1)
 336         return super(Post, self).unlink()
 337
 338     @api.multi
 339     def vote(self, upvote=True):
 340         Vote = self.env['forum.post.vote']
 341         vote_ids = Vote.search([('post_id', 'in', self._ids), ('user_id', '=', self._uid)])
 342         new_vote = '1' if upvote else '-1'
 343         voted_forum_ids = set()
 344         if vote_ids:
 345             for vote in vote_ids:
 346                 if upvote:
 347                     new_vote = '0' if vote.vote == '-1' else '1'
 348                 else:
 349                     new_vote = '0' if vote.vote == '1' else '-1'
 350                 vote.vote = new_vote
 351                 voted_forum_ids.add(vote.post_id.id)
 352         for post_id in set(self._ids) - voted_forum_ids:
 353             for post_id in self._ids:
 354                 Vote.create({'post_id': post_id, 'vote': new_vote})
 355         return {'vote_count': self.vote_count, 'user_vote': new_vote}
 356
 357     @api.one
 358     def convert_answer_to_comment(self):
 359         """ Tools to convert an answer (forum.post) to a comment (mail.message).
 360         The original post is unlinked and a new comment is posted on the question
 361         using the post create_uid as the comment's author. """
 362         if not self.parent_id:
 363             return False
 364
 365         # karma-based action check: use the post field that computed own/all value
 366         if not self.can_comment_convert:
 367             raise KarmaError('Not enough karma to convert an answer to a comment')
 368
 369         # post the message
 370         question = self.parent_id
 371         values = {
 372             'author_id': self.create_uid.partner_id.id,
 373             'body': tools.html2plaintext(self.content),
 374             'type': 'comment',
 375             'subtype': 'mail.mt_comment',
 376             'date': self.create_date,
 377         }
 378         new_message = self.browse(question.id).with_context(mail_create_nosubcribe=True).message_post(**values)
 379
 380         # unlink the original answer, using SUPERUSER_ID to avoid karma issues
 381         self.sudo().unlink()
 382
 383         return new_message
 384
 385     @api.model
 386     def convert_comment_to_answer(self, message_id, default=None):
 387         """ Tool to convert a comment (mail.message) into an answer (forum.post).
 388         The original comment is unlinked and a new answer from the comment's author
 389         is created. Nothing is done if the comment's author already answered the
 390         question. """
 391         comment = self.env['mail.message'].sudo().browse(message_id)
 392         post = self.browse(comment.res_id)
 393         if not comment.author_id or not comment.author_id.user_ids:  # only comment posted by users can be converted
 394             return False
 395
 396         # karma-based action check: must check the message's author to know if own / all
 397         karma_convert = comment.author_id.id == self.env.user.partner_id.id and post.forum_id.karma_comment_convert_own or post.forum_id.karma_comment_convert_all
 398         can_convert = self.env.user.karma >= karma_convert
 399         if not can_convert:
 400             raise KarmaError('Not enough karma to convert a comment to an answer')
 401
 402         # check the message's author has not already an answer
 403         question = post.parent_id if post.parent_id else post
 404         post_create_uid = comment.author_id.user_ids[0]
 405         if any(answer.create_uid.id == post_create_uid.id for answer in question.child_ids):
 406             return False
 407
 408         # create the new post
 409         post_values = {
 410             'forum_id': question.forum_id.id,
 411             'content': comment.body,
 412             'parent_id': question.id,
 413         }
 414         # done with the author user to have create_uid correctly set
 415         new_post = self.sudo(post_create_uid.id).create(post_values)
 416
 417         # delete comment
 418         comment.unlink()
 419
 420         return new_post
 421
 422     @api.one
 423     def unlink_comment(self, message_id):
 424         user = self.env.user
 425         comment = self.env['mail.message'].sudo().browse(message_id)
 426         if not comment.model == 'forum.post' or not comment.res_id == self.id:
 427             return False
 428         # karma-based action check: must check the message's author to know if own or all
 429         karma_unlink = comment.author_id.id == user.partner_id.id and self.forum_id.karma_comment_unlink_own or self.forum_id.karma_comment_unlink_all
 430         can_unlink = user.karma >= karma_unlink
 431         if not can_unlink:
 432             raise KarmaError('Not enough karma to unlink a comment')
 433         return comment.unlink()
 434
 435     @api.multi
 436     def set_viewed(self):
 437         self._cr.execute("""UPDATE forum_post SET views = views+1 WHERE id IN %s""", (self._ids,))
 438         return True
 439
 440     @api.model
 441     def _get_access_link(self, mail, partner):
 442         post = self.browse(mail.res_id)
 443         res_id = post.parent_id and "%s#answer-%s" % (post.parent_id.id, post.id) or post.id
 444         return "/forum/%s/question/%s" % (post.forum_id.id, res_id)
 445
 446     @api.cr_uid_ids_context
 447     def message_post(self, cr, uid, thread_id, type='notification', subtype=None, context=None, **kwargs):
 448         if thread_id and type == 'comment':  # user comments have a restriction on karma
 449             if isinstance(thread_id, (list, tuple)):
 450                 post_id = thread_id[0]
 451             else:
 452                 post_id = thread_id
 453             post = self.browse(cr, uid, post_id, context=context)
 454             # TDE FIXME: trigger browse because otherwise the function field is not compted - check with RCO
 455             tmp1, tmp2 = post.karma_comment, post.can_comment
 456             user = self.pool['res.users'].browse(cr, uid, uid)
 457             tmp3 = user.karma
 458             # TDE END FIXME
 459             if not post.can_comment:
 460                 raise KarmaError('Not enough karma to comment')
 461         return super(Post, self).message_post(cr, uid, thread_id, type=type, subtype=subtype, context=context, **kwargs)
 462
 463
 464 class PostReason(models.Model):
 465     _name = "forum.post.reason"
 466     _description = "Post Closing Reason"
 467     _order = 'name'
 468
 469     name = fields.Char(string='Closing Reason', required=True, translate=True)
 470
 471
 472 class Vote(models.Model):
 473     _name = 'forum.post.vote'
 474     _description = 'Vote'
 475
 476     post_id = fields.Many2one('forum.post', string='Post', ondelete='cascade', required=True)
 477     user_id = fields.Many2one('res.users', string='User', required=True, default=lambda self: self._uid)
 478     vote = fields.Selection([('1', '1'), ('-1', '-1'), ('0', '0')], string='Vote', required=True, default='1')
 479     create_date = fields.Datetime('Create Date', select=True, readonly=True)
 480     forum_id = fields.Many2one('forum.forum', string='Forum', related="post_id.forum_id", store=True)
 481     recipient_id = fields.Many2one('res.users', string='To', related="post_id.create_uid", store=True)
 482
 483     def _get_karma_value(self, old_vote, new_vote, up_karma, down_karma):
 484         _karma_upd = {
 485             '-1': {'-1': 0, '0': -1 * down_karma, '1': -1 * down_karma + up_karma},
 486             '0': {'-1': 1 * down_karma, '0': 0, '1': up_karma},
 487             '1': {'-1': -1 * up_karma + down_karma, '0': -1 * up_karma, '1': 0}
 488         }
 489         return _karma_upd[old_vote][new_vote]
 490
 491     @api.model
 492     def create(self, vals):
 493         vote = super(Vote, self).create(vals)
 494
 495         # own post check
 496         if vote.user_id.id == vote.post_id.create_uid.id:
 497             raise Warning('Not allowed to vote for its own post')
 498         # karma check
 499         if vote.vote == '1' and not vote.post_id.can_upvote:
 500             raise KarmaError('Not enough karma to upvote.')
 501         elif vote.vote == '-1' and not vote.post_id.can_downvote:
 502             raise KarmaError('Not enough karma to downvote.')
 503
 504         if vote.post_id.parent_id:
 505             karma_value = self._get_karma_value('0', vote.vote, vote.forum_id.karma_gen_answer_upvote, vote.forum_id.karma_gen_answer_downvote)
 506         else:
 507             karma_value = self._get_karma_value('0', vote.vote, vote.forum_id.karma_gen_question_upvote, vote.forum_id.karma_gen_question_downvote)
 508         vote.recipient_id.sudo().add_karma(karma_value)
 509         return vote
 510
 511     @api.multi
 512     def write(self, values):
 513         if 'vote' in values:
 514             for vote in self:
 515                 # own post check
 516                 if vote.user_id.id == vote.post_id.create_uid.id:
 517                     raise Warning('Not allowed to vote for its own post')
 518                 # karma check
 519                 if (values['vote'] == '1' or vote.vote == '-1' and values['vote'] == '0') and not vote.post_id.can_upvote:
 520                     raise KarmaError('Not enough karma to upvote.')
 521                 elif (values['vote'] == '-1' or vote.vote == '1' and values['vote'] == '0') and not vote.post_id.can_downvote:
 522                     raise KarmaError('Not enough karma to downvote.')
 523
 524                 # karma update
 525                 if vote.post_id.parent_id:
 526                     karma_value = self._get_karma_value(vote.vote, values['vote'], vote.forum_id.karma_gen_answer_upvote, vote.forum_id.karma_gen_answer_downvote)
 527                 else:
 528                     karma_value = self._get_karma_value(vote.vote, values['vote'], vote.forum_id.karma_gen_question_upvote, vote.forum_id.karma_gen_question_downvote)
 529                 vote.recipient_id.sudo().add_karma(karma_value)
 530         res = super(Vote, self).write(values)
 531         return res
 532
 533
 534 class Tags(models.Model):
 535     _name = "forum.tag"
 536     _description = "Forum Tag"
 537     _inherit = ['website.seo.metadata']
 538
 539     name = fields.Char('Name', required=True)
 540     create_uid = fields.Many2one('res.users', string='Created by', readonly=True)
 541     forum_id = fields.Many2one('forum.forum', string='Forum', required=True)
 542     post_ids = fields.Many2many('forum.post', 'forum_tag_rel', 'forum_tag_id', 'forum_id', string='Posts')
 543     posts_count = fields.Integer('Number of Posts', compute='_get_posts_count', store=True)
 544
 545     @api.multi
 546     @api.depends("post_ids.tag_ids")
 547     def _get_posts_count(self):
 548         for tag in self:
 549             tag.posts_count = len(tag.post_ids)