226 lines
8.5 KiB
Python
226 lines
8.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from datetime import datetime
|
|
import difflib
|
|
import lxml
|
|
import random
|
|
|
|
from openerp import tools
|
|
from openerp import SUPERUSER_ID
|
|
from openerp.osv import osv, fields
|
|
from openerp.tools.translate import _
|
|
|
|
|
|
class Blog(osv.Model):
|
|
_name = 'blog.blog'
|
|
_description = 'Blogs'
|
|
_inherit = ['mail.thread', 'website.seo.metadata']
|
|
_order = 'name'
|
|
_columns = {
|
|
'name': fields.char('Blog Name', required=True),
|
|
'subtitle': fields.char('Blog Subtitle'),
|
|
'description': fields.text('Description'),
|
|
}
|
|
|
|
|
|
class BlogTag(osv.Model):
|
|
_name = 'blog.tag'
|
|
_description = 'Blog Tag'
|
|
_inherit = ['website.seo.metadata']
|
|
_order = 'name'
|
|
_columns = {
|
|
'name': fields.char('Name', required=True),
|
|
}
|
|
|
|
|
|
class BlogPost(osv.Model):
|
|
_name = "blog.post"
|
|
_description = "Blog Post"
|
|
_inherit = ['mail.thread', 'website.seo.metadata']
|
|
_order = 'id DESC'
|
|
|
|
def _compute_ranking(self, cr, uid, ids, name, arg, context=None):
|
|
res = {}
|
|
for blog_post in self.browse(cr, uid, ids, context=context):
|
|
age = datetime.now() - datetime.strptime(blog_post.create_date, tools.DEFAULT_SERVER_DATETIME_FORMAT)
|
|
res[blog_post.id] = blog_post.visits * (0.5+random.random()) / max(3, age.days)
|
|
return res
|
|
|
|
_columns = {
|
|
'name': fields.char('Title', required=True, translate=True),
|
|
'subtitle': fields.char('Sub Title', translate=True),
|
|
'author_id': fields.many2one('res.partner', 'Author'),
|
|
'background_image': fields.binary('Background Image', oldname='content_image'),
|
|
'blog_id': fields.many2one(
|
|
'blog.blog', 'Blog',
|
|
required=True, ondelete='cascade',
|
|
),
|
|
'tag_ids': fields.many2many(
|
|
'blog.tag', string='Tags',
|
|
),
|
|
'content': fields.html('Content', translate=True),
|
|
# website control
|
|
'website_published': fields.boolean(
|
|
'Publish', help="Publish on the website"
|
|
),
|
|
'website_message_ids': fields.one2many(
|
|
'mail.message', 'res_id',
|
|
domain=lambda self: [
|
|
'&', '&', ('model', '=', self._name), ('type', '=', 'comment'), ('path', '=', False)
|
|
],
|
|
string='Website Messages',
|
|
help="Website communication history",
|
|
),
|
|
'history_ids': fields.one2many(
|
|
'blog.post.history', 'post_id',
|
|
'History', help='Last post modifications',
|
|
deprecated='This field will be removed for OpenERP v9.'
|
|
),
|
|
# creation / update stuff
|
|
'create_date': fields.datetime(
|
|
'Created on',
|
|
select=True, readonly=True,
|
|
),
|
|
'create_uid': fields.many2one(
|
|
'res.users', 'Author',
|
|
select=True, readonly=True,
|
|
),
|
|
'write_date': fields.datetime(
|
|
'Last Modified on',
|
|
select=True, readonly=True,
|
|
),
|
|
'write_uid': fields.many2one(
|
|
'res.users', 'Last Contributor',
|
|
select=True, readonly=True,
|
|
),
|
|
'visits': fields.integer('No of Views'),
|
|
'ranking': fields.function(_compute_ranking, string='Ranking', type='float'),
|
|
}
|
|
|
|
_defaults = {
|
|
'name': _('Blog Post Title'),
|
|
'subtitle': _('Subtitle'),
|
|
'author_id': lambda self, cr, uid, ctx=None: self.pool['res.users'].browse(cr, uid, uid, context=ctx).partner_id.id,
|
|
}
|
|
|
|
def html_tag_nodes(self, html, attribute=None, tags=None, context=None):
|
|
""" Processing of html content to tag paragraphs and set them an unique
|
|
ID.
|
|
:return result: (html, mappin), where html is the updated html with ID
|
|
and mapping is a list of (old_ID, new_ID), where old_ID
|
|
is None is the paragraph is a new one. """
|
|
mapping = []
|
|
if not html:
|
|
return html, mapping
|
|
if tags is None:
|
|
tags = ['p']
|
|
if attribute is None:
|
|
attribute = 'data-unique-id'
|
|
counter = 0
|
|
|
|
# form a tree
|
|
root = lxml.html.fragment_fromstring(html, create_parent='div')
|
|
if not len(root) and root.text is None and root.tail is None:
|
|
return html, mapping
|
|
|
|
# check all nodes, replace :
|
|
# - img src -> check URL
|
|
# - a href -> check URL
|
|
for node in root.iter():
|
|
if not node.tag in tags:
|
|
continue
|
|
ancestor_tags = [parent.tag for parent in node.iterancestors()]
|
|
if ancestor_tags:
|
|
ancestor_tags.pop()
|
|
ancestor_tags.append('counter_%s' % counter)
|
|
new_attribute = '/'.join(reversed(ancestor_tags))
|
|
old_attribute = node.get(attribute)
|
|
node.set(attribute, new_attribute)
|
|
mapping.append((old_attribute, counter))
|
|
counter += 1
|
|
|
|
html = lxml.html.tostring(root, pretty_print=False, method='html')
|
|
# this is ugly, but lxml/etree tostring want to put everything in a 'div' that breaks the editor -> remove that
|
|
if html.startswith('<div>') and html.endswith('</div>'):
|
|
html = html[5:-6]
|
|
return html, mapping
|
|
|
|
def _postproces_content(self, cr, uid, id, content=None, context=None):
|
|
if content is None:
|
|
content = self.browse(cr, uid, id, context=context).content
|
|
if content is False:
|
|
return content
|
|
content, mapping = self.html_tag_nodes(content, attribute='data-chatter-id', tags=['p'], context=context)
|
|
for old_attribute, new_attribute in mapping:
|
|
if not old_attribute:
|
|
continue
|
|
msg_ids = self.pool['mail.message'].search(cr, SUPERUSER_ID, [('path', '=', old_attribute)], context=context)
|
|
self.pool['mail.message'].write(cr, SUPERUSER_ID, msg_ids, {'path': new_attribute}, context=context)
|
|
return content
|
|
|
|
def create_history(self, cr, uid, ids, vals, context=None):
|
|
for i in ids:
|
|
history = self.pool.get('blog.post.history')
|
|
if vals.get('content'):
|
|
res = {
|
|
'content': vals.get('content', ''),
|
|
'post_id': i,
|
|
}
|
|
history.create(cr, uid, res)
|
|
|
|
def create(self, cr, uid, vals, context=None):
|
|
if context is None:
|
|
context = {}
|
|
if 'content' in vals:
|
|
vals['content'] = self._postproces_content(cr, uid, None, vals['content'], context=context)
|
|
create_context = dict(context, mail_create_nolog=True)
|
|
post_id = super(BlogPost, self).create(cr, uid, vals, context=create_context)
|
|
self.create_history(cr, uid, [post_id], vals, context)
|
|
return post_id
|
|
|
|
def write(self, cr, uid, ids, vals, context=None):
|
|
if 'content' in vals:
|
|
vals['content'] = self._postproces_content(cr, uid, None, vals['content'], context=context)
|
|
result = super(BlogPost, self).write(cr, uid, ids, vals, context)
|
|
self.create_history(cr, uid, ids, vals, context)
|
|
return result
|
|
|
|
def copy(self, cr, uid, id, default=None, context=None):
|
|
if default is None:
|
|
default = {}
|
|
default.update({
|
|
'website_message_ids': [],
|
|
'website_published': False,
|
|
'website_published_datetime': False,
|
|
})
|
|
return super(BlogPost, self).copy(cr, uid, id, default=default, context=context)
|
|
|
|
|
|
class BlogPostHistory(osv.Model):
|
|
_name = "blog.post.history"
|
|
_description = "Blog Post History"
|
|
_order = 'id DESC'
|
|
_rec_name = "create_date"
|
|
|
|
_columns = {
|
|
'post_id': fields.many2one('blog.post', 'Blog Post'),
|
|
'summary': fields.char('Summary', size=256, select=True),
|
|
'content': fields.text("Content"),
|
|
'create_date': fields.datetime("Date"),
|
|
'create_uid': fields.many2one('res.users', "Modified By"),
|
|
}
|
|
|
|
def getDiff(self, cr, uid, v1, v2, context=None):
|
|
history_pool = self.pool.get('blog.post.history')
|
|
text1 = history_pool.read(cr, uid, [v1], ['content'])[0]['content']
|
|
text2 = history_pool.read(cr, uid, [v2], ['content'])[0]['content']
|
|
line1 = line2 = ''
|
|
if text1:
|
|
line1 = text1.splitlines(1)
|
|
if text2:
|
|
line2 = text2.splitlines(1)
|
|
if (not line1 and not line2) or (line1 == line2):
|
|
raise osv.except_osv(_('Warning!'), _('There are no changes in revisions.'))
|
|
diff = difflib.HtmlDiff()
|
|
return diff.make_table(line1, line2, "Revision-%s" % (v1), "Revision-%s" % (v2), context=True)
|