addons/website_blog/controllers/main.py

   1 # -*- coding: utf-8 -*-
   2
   3 import datetime
   4 import werkzeug
   5
   6 from openerp import tools
   7 from openerp.addons.web import http
   8 from openerp.addons.web.http import request
   9 from openerp.addons.website.models.website import slug
  10 from openerp.osv.orm import browse_record
  11 from openerp.tools.translate import _
  12 from openerp import SUPERUSER_ID
  13 from openerp.tools import html2plaintext
  14
  15
  16 class QueryURL(object):
  17     def __init__(self, path='', path_args=None, **args):
  18         self.path = path
  19         self.args = args
  20         self.path_args = set(path_args or [])
  21
  22     def __call__(self, path=None, path_args=None, **kw):
  23         path = path or self.path
  24         for k, v in self.args.items():
  25             kw.setdefault(k, v)
  26         path_args = set(path_args or []).union(self.path_args)
  27         paths, fragments = [], []
  28         for key, value in kw.items():
  29             if value and key in path_args:
  30                 if isinstance(value, browse_record):
  31                     paths.append((key, slug(value)))
  32                 else:
  33                     paths.append((key, value))
  34             elif value:
  35                 if isinstance(value, list) or isinstance(value, set):
  36                     fragments.append(werkzeug.url_encode([(key, item) for item in value]))
  37                 else:
  38                     fragments.append(werkzeug.url_encode([(key, value)]))
  39         for key, value in paths:
  40             path += '/' + key + '/%s' % value
  41         if fragments:
  42             path += '?' + '&'.join(fragments)
  43         return path
  44
  45
  46 class WebsiteBlog(http.Controller):
  47     _blog_post_per_page = 20
  48     _post_comment_per_page = 10
  49
  50     def nav_list(self):
  51         blog_post_obj = request.registry['blog.post']
  52         groups = blog_post_obj.read_group(
  53             request.cr, request.uid, [], ['name', 'create_date'],
  54             groupby="create_date", orderby="create_date desc", context=request.context)
  55         for group in groups:
  56             begin_date = datetime.datetime.strptime(group['__domain'][0][2], tools.DEFAULT_SERVER_DATETIME_FORMAT).date()
  57             end_date = datetime.datetime.strptime(group['__domain'][1][2], tools.DEFAULT_SERVER_DATETIME_FORMAT).date()
  58             group['date_begin'] = '%s' % datetime.date.strftime(begin_date, tools.DEFAULT_SERVER_DATE_FORMAT)
  59             group['date_end'] = '%s' % datetime.date.strftime(end_date, tools.DEFAULT_SERVER_DATE_FORMAT)
  60         return groups
  61
  62     @http.route([
  63         '/blog',
  64         '/blog/page/<int:page>',
  65     ], type='http', auth="public", website=True)
  66     def blogs(self, page=1, **post):
  67         cr, uid, context = request.cr, request.uid, request.context
  68         blog_obj = request.registry['blog.post']
  69         total = blog_obj.search(cr, uid, [], count=True, context=context)
  70         pager = request.website.pager(
  71             url='/blog',
  72             total=total,
  73             page=page,
  74             step=self._blog_post_per_page,
  75         )
  76         post_ids = blog_obj.search(cr, uid, [], offset=(page-1)*self._blog_post_per_page, limit=self._blog_post_per_page, context=context)
  77         posts = blog_obj.browse(cr, uid, post_ids, context=context)
  78         blog_url = QueryURL('', ['blog', 'tag'])
  79         return request.website.render("website_blog.latest_blogs", {
  80             'posts': posts,
  81             'pager': pager,
  82             'blog_url': blog_url,
  83         })
  84
  85     @http.route([
  86         '/blog/<model("blog.blog"):blog>',
  87         '/blog/<model("blog.blog"):blog>/page/<int:page>',
  88         '/blog/<model("blog.blog"):blog>/tag/<model("blog.tag"):tag>',
  89         '/blog/<model("blog.blog"):blog>/tag/<model("blog.tag"):tag>/page/<int:page>',
  90     ], type='http', auth="public", website=True)
  91     def blog(self, blog=None, tag=None, page=1, **opt):
  92         """ Prepare all values to display the blog.
  93
  94         :return dict values: values for the templates, containing
  95
  96          - 'blog': current blog
  97          - 'blogs': all blogs for navigation
  98          - 'pager': pager of posts
  99          - 'tag': current tag
 100          - 'tags': all tags, for navigation
 101          - 'nav_list': a dict [year][month] for archives navigation
 102          - 'date': date_begin optional parameter, used in archives navigation
 103          - 'blog_url': help object to create URLs
 104         """
 105         date_begin, date_end = opt.get('date_begin'), opt.get('date_end')
 106
 107         cr, uid, context = request.cr, request.uid, request.context
 108         blog_post_obj = request.registry['blog.post']
 109
 110         blog_obj = request.registry['blog.blog']
 111         blog_ids = blog_obj.search(cr, uid, [], order="create_date asc", context=context)
 112         blogs = blog_obj.browse(cr, uid, blog_ids, context=context)
 113
 114         domain = []
 115         if blog:
 116             domain += [('blog_id', '=', blog.id)]
 117         if tag:
 118             domain += [('tag_ids', 'in', tag.id)]
 119         if date_begin and date_end:
 120             domain += [("create_date", ">=", date_begin), ("create_date", "<=", date_end)]
 121
 122         blog_url = QueryURL('', ['blog', 'tag'], blog=blog, tag=tag, date_begin=date_begin, date_end=date_end)
 123         post_url = QueryURL('', ['blogpost'], tag_id=tag and tag.id or None, date_begin=date_begin, date_end=date_end)
 124
 125         blog_post_ids = blog_post_obj.search(cr, uid, domain, order="create_date desc", context=context)
 126         blog_posts = blog_post_obj.browse(cr, uid, blog_post_ids, context=context)
 127
 128         pager = request.website.pager(
 129             url=blog_url(),
 130             total=len(blog_posts),
 131             page=page,
 132             step=self._blog_post_per_page,
 133         )
 134         pager_begin = (page - 1) * self._blog_post_per_page
 135         pager_end = page * self._blog_post_per_page
 136         blog_posts = blog_posts[pager_begin:pager_end]
 137
 138         tags = blog.all_tags()[blog.id]
 139
 140         values = {
 141             'blog': blog,
 142             'blogs': blogs,
 143             'tags': tags,
 144             'tag': tag,
 145             'blog_posts': blog_posts,
 146             'pager': pager,
 147             'nav_list': self.nav_list(),
 148             'blog_url': blog_url,
 149             'post_url': post_url,
 150             'date': date_begin,
 151         }
 152         response = request.website.render("website_blog.blog_post_short", values)
 153         return response
 154
 155     @http.route([
 156             '''/blog/<model("blog.blog"):blog>/post/<model("blog.post", "[('blog_id','=',blog[0])]"):blog_post>''',
 157     ], type='http', auth="public", website=True)
 158     def blog_post(self, blog, blog_post, tag_id=None, page=1, enable_editor=None, **post):
 159         """ Prepare all values to display the blog.
 160
 161         :return dict values: values for the templates, containing
 162
 163          - 'blog_post': browse of the current post
 164          - 'blog': browse of the current blog
 165          - 'blogs': list of browse records of blogs
 166          - 'tag': current tag, if tag_id in parameters
 167          - 'tags': all tags, for tag-based navigation
 168          - 'pager': a pager on the comments
 169          - 'nav_list': a dict [year][month] for archives navigation
 170          - 'next_post': next blog post, to direct the user towards the next interesting post
 171         """
 172         cr, uid, context = request.cr, request.uid, request.context
 173         tag_obj = request.registry['blog.tag']
 174         blog_post_obj = request.registry['blog.post']
 175         date_begin, date_end = post.get('date_begin'), post.get('date_end')
 176
 177         pager_url = "/blogpost/%s" % blog_post.id
 178
 179         pager = request.website.pager(
 180             url=pager_url,
 181             total=len(blog_post.website_message_ids),
 182             page=page,
 183             step=self._post_comment_per_page,
 184             scope=7
 185         )
 186         pager_begin = (page - 1) * self._post_comment_per_page
 187         pager_end = page * self._post_comment_per_page
 188         comments = blog_post.website_message_ids[pager_begin:pager_end]
 189
 190         tag = None
 191         if tag_id:
 192             tag = request.registry['blog.tag'].browse(request.cr, request.uid, int(tag_id), context=request.context)
 193         post_url = QueryURL('', ['blogpost'], blogpost=blog_post, tag_id=tag_id, date_begin=date_begin, date_end=date_end)
 194         blog_url = QueryURL('', ['blog', 'tag'], blog=blog_post.blog_id, tag=tag, date_begin=date_begin, date_end=date_end)
 195
 196         if not blog_post.blog_id.id == blog.id:
 197             return request.redirect("/blog/%s/post/%s" % (slug(blog_post.blog_id), slug(blog_post)))
 198
 199         tags = tag_obj.browse(cr, uid, tag_obj.search(cr, uid, [], context=context), context=context)
 200
 201         # Find next Post
 202         visited_blogs = request.httprequest.cookies.get('visited_blogs') or ''
 203         visited_ids = filter(None, visited_blogs.split(','))
 204         visited_ids = map(lambda x: int(x), visited_ids)
 205         if blog_post.id not in visited_ids:
 206             visited_ids.append(blog_post.id)
 207         next_post_id = blog_post_obj.search(cr, uid, [
 208             ('id', 'not in', visited_ids),
 209         ], order='ranking desc', limit=1, context=context)
 210         if not next_post_id:
 211             next_post_id = blog_post_obj.search(cr, uid, [('id', '!=', blog.id)], order='ranking desc', limit=1, context=context)
 212         next_post = next_post_id and blog_post_obj.browse(cr, uid, next_post_id[0], context=context) or False
 213
 214         values = {
 215             'tags': tags,
 216             'tag': tag,
 217             'blog': blog,
 218             'blog_post': blog_post,
 219             'main_object': blog_post,
 220             'nav_list': self.nav_list(),
 221             'enable_editor': enable_editor,
 222             'next_post': next_post,
 223             'date': date_begin,
 224             'post_url': post_url,
 225             'blog_url': blog_url,
 226             'pager': pager,
 227             'comments': comments,
 228         }
 229         response = request.website.render("website_blog.blog_post_complete", values)
 230         response.set_cookie('visited_blogs', ','.join(map(str, visited_ids)))
 231
 232         request.session[request.session_id] = request.session.get(request.session_id, [])
 233         if not (blog_post.id in request.session[request.session_id]):
 234             request.session[request.session_id].append(blog_post.id)
 235             # Increase counter
 236             blog_post_obj.write(cr, SUPERUSER_ID, [blog_post.id], {
 237                 'visits': blog_post.visits+1,
 238             },context=context)
 239         return response
 240
 241     def _blog_post_message(self, user, blog_post_id=0, **post):
 242         cr, uid, context = request.cr, request.uid, request.context
 243         blog_post = request.registry['blog.post']
 244         partner_obj = request.registry['res.partner']
 245
 246         if uid != request.website.user_id.id:
 247             partner_ids = [user.partner_id.id]
 248         else:
 249             partner_ids = blog_post._find_partner_from_emails(
 250                 cr, SUPERUSER_ID, 0, [post.get('email')], context=context)
 251             if not partner_ids or not partner_ids[0]:
 252                 partner_ids = [partner_obj.create(cr, SUPERUSER_ID, {'name': post.get('name'), 'email': post.get('email')}, context=context)]
 253
 254         message_id = blog_post.message_post(
 255             cr, SUPERUSER_ID, int(blog_post_id),
 256             body=post.get('comment'),
 257             type='comment',
 258             subtype='mt_comment',
 259             author_id=partner_ids[0],
 260             path=post.get('path', False),
 261             context=context)
 262         return message_id
 263
 264     @http.route(['/blogpost/comment'], type='http', auth="public", methods=['POST'], website=True)
 265     def blog_post_comment(self, blog_post_id=0, **post):
 266         cr, uid, context = request.cr, request.uid, request.context
 267         if post.get('comment'):
 268             user = request.registry['res.users'].browse(cr, uid, uid, context=context)
 269             blog_post = request.registry['blog.post']
 270             blog_post.check_access_rights(cr, uid, 'read')
 271             self._blog_post_message(user, blog_post_id, **post)
 272         return werkzeug.utils.redirect(request.httprequest.referrer + "#comments")
 273
 274     def _get_discussion_detail(self, ids, publish=False, **post):
 275         cr, uid, context = request.cr, request.uid, request.context
 276         values = []
 277         mail_obj = request.registry.get('mail.message')
 278         for message in mail_obj.browse(cr, SUPERUSER_ID, ids, context=context):
 279             values.append({
 280                 "id": message.id,
 281                 "author_name": message.author_id.name,
 282                 "author_image": message.author_id.image and \
 283                     ("data:image/png;base64,%s" % message.author_id.image) or \
 284                     '/website_blog/static/src/img/anonymous.png',
 285                 "date": message.date,
 286                 'body': html2plaintext(message.body),
 287                 'website_published' : message.website_published,
 288                 'publish' : publish,
 289             })
 290         return values
 291
 292     @http.route(['/blogpost/post_discussion'], type='json', auth="public", website=True)
 293     def post_discussion(self, blog_post_id, **post):
 294         cr, uid, context = request.cr, request.uid, request.context
 295         publish = request.registry['res.users'].has_group(cr, uid, 'base.group_website_publisher')
 296         user = request.registry['res.users'].browse(cr, uid, uid, context=context)
 297         id = self._blog_post_message(user, blog_post_id, **post)
 298         return self._get_discussion_detail([id], publish, **post)
 299
 300     @http.route('/blogpost/new', type='http', auth="public", website=True)
 301     def blog_post_create(self, blog_id, **post):
 302         cr, uid, context = request.cr, request.uid, request.context
 303         new_blog_post_id = request.registry['blog.post'].create(cr, uid, {
 304             'blog_id': blog_id,
 305             'name': _("Blog Post Title"),
 306             'subtitle': _("Subtitle"),
 307             'content': '',
 308             'website_published': False,
 309         }, context=context)
 310         new_blog_post = request.registry['blog.post'].browse(cr, uid, new_blog_post_id, context=context)
 311         return werkzeug.utils.redirect("/blog/%s/post/%s?enable_editor=1" % (slug(new_blog_post.blog_id), slug(new_blog_post)))
 312
 313     @http.route('/blogpost/duplicate', type='http', auth="public", website=True)
 314     def blog_post_copy(self, blog_post_id, **post):
 315         """ Duplicate a blog.
 316
 317         :param blog_post_id: id of the blog post currently browsed.
 318
 319         :return redirect to the new blog created
 320         """
 321         cr, uid, context = request.cr, request.uid, request.context
 322         create_context = dict(context, mail_create_nosubscribe=True)
 323         nid = request.registry['blog.post'].copy(cr, uid, blog_post_id, {}, context=create_context)
 324         new_blog_post = request.registry['blog.post'].browse(cr, uid, nid, context=context)
 325         post = request.registry['blog.post'].browse(cr, uid, nid, context)
 326         return werkzeug.utils.redirect("/blog/%s/post/%s?enable_editor=1" % (slug(post.blog_id), slug(new_blog_post)))
 327
 328     @http.route('/blogpost/get_discussion/', type='json', auth="public", website=True)
 329     def discussion(self, post_id=0, path=None, count=False, **post):
 330         cr, uid, context = request.cr, request.uid, request.context
 331         mail_obj = request.registry.get('mail.message')
 332         domain = [('res_id', '=', int(post_id)), ('model', '=', 'blog.post'), ('path', '=', path)]
 333         #check current user belongs to website publisher group
 334         publish = request.registry['res.users'].has_group(cr, uid, 'base.group_website_publisher')
 335         if not publish:
 336             domain.append(('website_published', '=', True))
 337         ids = mail_obj.search(cr, SUPERUSER_ID, domain, count=count)
 338         if count:
 339             return ids
 340         return self._get_discussion_detail(ids, publish, **post)
 341
 342     @http.route('/blogpost/get_discussions/', type='json', auth="public", website=True)
 343     def discussions(self, post_id=0, paths=None, count=False, **post):
 344         ret = []
 345         for path in paths:
 346             result = self.discussion(post_id=post_id, path=path, count=count, **post)
 347             ret.append({"path": path, "val": result})
 348         return ret
 349
 350     @http.route('/blogpost/change_background', type='json', auth="public", website=True)
 351     def change_bg(self, post_id=0, image=None, **post):
 352         if not post_id:
 353             return False
 354         return request.registry['blog.post'].write(request.cr, request.uid, [int(post_id)], {'background_image': image}, request.context)
 355
 356     @http.route('/blog/get_user/', type='json', auth="public", website=True)
 357     def get_user(self, **post):
 358         return [False if request.session.uid else True]