addons/website_blog/models/website_blog.py

   1 # -*- coding: utf-8 -*-
   2
   3 from datetime import datetime
   4 import difflib
   5 import lxml
   6 import random
   7
   8 from openerp import tools
   9 from openerp import SUPERUSER_ID
  10 from openerp.osv import osv, fields
  11 from openerp.tools.translate import _
  12
  13
  14 class Blog(osv.Model):
  15     _name = 'blog.blog'
  16     _description = 'Blogs'
  17     _inherit = ['mail.thread', 'website.seo.metadata']
  18     _order = 'name'
  19     _columns = {
  20         'name': fields.char('Blog Name', required=True),
  21         'subtitle': fields.char('Blog Subtitle'),
  22         'description': fields.text('Description'),
  23     }
  24
  25
  26 class BlogTag(osv.Model):
  27     _name = 'blog.tag'
  28     _description = 'Blog Tag'
  29     _inherit = ['website.seo.metadata']
  30     _order = 'name'
  31     _columns = {
  32         'name': fields.char('Name', required=True),
  33     }
  34
  35
  36 class BlogPost(osv.Model):
  37     _name = "blog.post"
  38     _description = "Blog Post"
  39     _inherit = ['mail.thread', 'website.seo.metadata']
  40     _order = 'id DESC'
  41
  42     def _compute_ranking(self, cr, uid, ids, name, arg, context=None):
  43         res = {}
  44         for blog_post in self.browse(cr, uid, ids, context=context):
  45             age = datetime.now() - datetime.strptime(blog_post.create_date, tools.DEFAULT_SERVER_DATETIME_FORMAT)
  46             res[blog_post.id] = blog_post.visits * (0.5+random.random()) / max(3, age.days)
  47         return res
  48
  49     _columns = {
  50         'name': fields.char('Title', required=True, translate=True),
  51         'subtitle': fields.char('Sub Title', translate=True),
  52         'author_id': fields.many2one('res.partner', 'Author'),
  53         'background_image': fields.binary('Background Image', oldname='content_image'),
  54         'blog_id': fields.many2one(
  55             'blog.blog', 'Blog',
  56             required=True, ondelete='cascade',
  57         ),
  58         'tag_ids': fields.many2many(
  59             'blog.tag', string='Tags',
  60         ),
  61         'content': fields.html('Content', translate=True, sanitize=False),
  62         # website control
  63         'website_published': fields.boolean(
  64             'Publish', help="Publish on the website", copy=False,
  65         ),
  66         'website_message_ids': fields.one2many(
  67             'mail.message', 'res_id',
  68             domain=lambda self: [
  69                 '&', '&', ('model', '=', self._name), ('type', '=', 'comment'), ('path', '=', False)
  70             ],
  71             string='Website Messages',
  72             help="Website communication history",
  73         ),
  74         # creation / update stuff
  75         'create_date': fields.datetime(
  76             'Created on',
  77             select=True, readonly=True,
  78         ),
  79         'create_uid': fields.many2one(
  80             'res.users', 'Author',
  81             select=True, readonly=True,
  82         ),
  83         'write_date': fields.datetime(
  84             'Last Modified on',
  85             select=True, readonly=True,
  86         ),
  87         'write_uid': fields.many2one(
  88             'res.users', 'Last Contributor',
  89             select=True, readonly=True,
  90         ),
  91         'author_avatar': fields.related(
  92             'author_id', 'image_small',
  93             string="Avatar", type="binary"),
  94         'visits': fields.integer('No of Views'),
  95         'ranking': fields.function(_compute_ranking, string='Ranking', type='float'),
  96     }
  97
  98     _defaults = {
  99         'name': _('Blog Post Title'),
 100         'subtitle': _('Subtitle'),
 101         'author_id': lambda self, cr, uid, ctx=None: self.pool['res.users'].browse(cr, uid, uid, context=ctx).partner_id.id,
 102     }
 103
 104     def html_tag_nodes(self, html, attribute=None, tags=None, context=None):
 105         """ Processing of html content to tag paragraphs and set them an unique
 106         ID.
 107         :return result: (html, mappin), where html is the updated html with ID
 108                         and mapping is a list of (old_ID, new_ID), where old_ID
 109                         is None is the paragraph is a new one. """
 110         mapping = []
 111         if not html:
 112             return html, mapping
 113         if tags is None:
 114             tags = ['p']
 115         if attribute is None:
 116             attribute = 'data-unique-id'
 117         counter = 0
 118
 119         # form a tree
 120         root = lxml.html.fragment_fromstring(html, create_parent='div')
 121         if not len(root) and root.text is None and root.tail is None:
 122             return html, mapping
 123
 124         # check all nodes, replace :
 125         # - img src -> check URL
 126         # - a href -> check URL
 127         for node in root.iter():
 128             if not node.tag in tags:
 129                 continue
 130             ancestor_tags = [parent.tag for parent in node.iterancestors()]
 131             if ancestor_tags:
 132                 ancestor_tags.pop()
 133             ancestor_tags.append('counter_%s' % counter)
 134             new_attribute = '/'.join(reversed(ancestor_tags))
 135             old_attribute = node.get(attribute)
 136             node.set(attribute, new_attribute)
 137             mapping.append((old_attribute, counter))
 138             counter += 1
 139
 140         html = lxml.html.tostring(root, pretty_print=False, method='html')
 141         # this is ugly, but lxml/etree tostring want to put everything in a 'div' that breaks the editor -> remove that
 142         if html.startswith('<div>') and html.endswith('</div>'):
 143             html = html[5:-6]
 144         return html, mapping
 145
 146     def _postproces_content(self, cr, uid, id, content=None, context=None):
 147         if content is None:
 148             content = self.browse(cr, uid, id, context=context).content
 149         if content is False:
 150             return content
 151         content, mapping = self.html_tag_nodes(content, attribute='data-chatter-id', tags=['p'], context=context)
 152         for old_attribute, new_attribute in mapping:
 153             if not old_attribute:
 154                 continue
 155             msg_ids = self.pool['mail.message'].search(cr, SUPERUSER_ID, [('path', '=', old_attribute)], context=context)
 156             self.pool['mail.message'].write(cr, SUPERUSER_ID, msg_ids, {'path': new_attribute}, context=context)
 157         return content
 158
 159     def create(self, cr, uid, vals, context=None):
 160         if context is None:
 161             context = {}
 162         if 'content' in vals:
 163             vals['content'] = self._postproces_content(cr, uid, None, vals['content'], context=context)
 164         create_context = dict(context, mail_create_nolog=True)
 165         post_id = super(BlogPost, self).create(cr, uid, vals, context=create_context)
 166         return post_id
 167
 168     def write(self, cr, uid, ids, vals, context=None):
 169         if 'content' in vals:
 170             vals['content'] = self._postproces_content(cr, uid, None, vals['content'], context=context)
 171         result = super(BlogPost, self).write(cr, uid, ids, vals, context)
 172         return result