addons/website/models/ir_ui_view.py

   1 # -*- coding: utf-8 -*-
   2 import copy
   3
   4 from lxml import etree, html
   5
   6 from openerp import SUPERUSER_ID
   7 from openerp.addons.website.models import website
   8 from openerp.http import request
   9 from openerp.osv import osv, fields
  10
  11 class view(osv.osv):
  12     _inherit = "ir.ui.view"
  13     _columns = {
  14         'page': fields.boolean("Whether this view is a web page template (complete)"),
  15         'website_meta_title': fields.char("Website meta title", size=70, translate=True),
  16         'website_meta_description': fields.text("Website meta description", size=160, translate=True),
  17         'website_meta_keywords': fields.char("Website meta keywords", translate=True),
  18         'customize_show': fields.boolean("Show As Optional Inherit"),
  19     }
  20     _defaults = {
  21         'page': False,
  22         'customize_show': False,
  23     }
  24
  25
  26     def _view_obj(self, cr, uid, view_id, context=None):
  27         if isinstance(view_id, basestring):
  28             return self.pool['ir.model.data'].xmlid_to_object(
  29                 cr, uid, view_id, raise_if_not_found=True, context=context
  30             )
  31         elif isinstance(view_id, (int, long)):
  32             return self.browse(cr, uid, view_id, context=context)
  33
  34         # assume it's already a view object (WTF?)
  35         return view_id
  36
  37     # Returns all views (called and inherited) related to a view
  38     # Used by translation mechanism, SEO and optional templates
  39     def _views_get(self, cr, uid, view_id, options=True, bundles=False, context=None, root=True):
  40         """ For a given view ``view_id``, should return:
  41
  42         * the view itself
  43         * all views inheriting from it, enabled or not
  44           - but not the optional children of a non-enabled child
  45         * all views called from it (via t-call)
  46         """
  47         try:
  48             view = self._view_obj(cr, uid, view_id, context=context)
  49         except ValueError:
  50             # Shall we log that ?
  51             return []
  52
  53         while root and view.inherit_id:
  54             view = view.inherit_id
  55
  56         result = [view]
  57
  58         node = etree.fromstring(view.arch)
  59         xpath = "//t[@t-call]"
  60         if bundles:
  61             xpath += "| //t[@t-call-assets]"
  62         for child in node.xpath(xpath):
  63             try:
  64                 called_view = self._view_obj(cr, uid, child.get('t-call', child.get('t-call-assets')), context=context)
  65             except ValueError:
  66                 continue
  67             if called_view not in result:
  68                 result += self._views_get(cr, uid, called_view, options=options, bundles=bundles, context=context)
  69
  70         extensions = view.inherit_children_ids
  71         if not options:
  72             # only active children
  73             extensions = (v for v in view.inherit_children_ids if v.active)
  74
  75         # Keep options in a deterministic order regardless of their applicability
  76         for extension in sorted(extensions, key=lambda v: v.id):
  77             for r in self._views_get(
  78                     cr, uid, extension,
  79                     # only return optional grandchildren if this child is enabled
  80                     options=extension.active,
  81                     context=context, root=False):
  82                 if r not in result:
  83                     result.append(r)
  84         return result
  85
  86     def extract_embedded_fields(self, cr, uid, arch, context=None):
  87         return arch.xpath('//*[@data-oe-model != "ir.ui.view"]')
  88
  89     def save_embedded_field(self, cr, uid, el, context=None):
  90         Model = self.pool[el.get('data-oe-model')]
  91         field = el.get('data-oe-field')
  92
  93         column = Model._all_columns[field].column
  94         converter = self.pool['website.qweb'].get_converter_for(
  95             el.get('data-oe-type'))
  96         value = converter.from_html(cr, uid, Model, column, el)
  97
  98         if value is not None:
  99             # TODO: batch writes?
 100             Model.write(cr, uid, [int(el.get('data-oe-id'))], {
 101                 field: value
 102             }, context=context)
 103
 104     def to_field_ref(self, cr, uid, el, context=None):
 105         # filter out meta-information inserted in the document
 106         attributes = dict((k, v) for k, v in el.items()
 107                           if not k.startswith('data-oe-'))
 108         attributes['t-field'] = el.get('data-oe-expression')
 109
 110         out = html.html_parser.makeelement(el.tag, attrib=attributes)
 111         out.tail = el.tail
 112         return out
 113
 114     def replace_arch_section(self, cr, uid, view_id, section_xpath, replacement, context=None):
 115         # the root of the arch section shouldn't actually be replaced as it's
 116         # not really editable itself, only the content truly is editable.
 117
 118         [view] = self.browse(cr, uid, [view_id], context=context)
 119         arch = etree.fromstring(view.arch.encode('utf-8'))
 120         # => get the replacement root
 121         if not section_xpath:
 122             root = arch
 123         else:
 124             # ensure there's only one match
 125             [root] = arch.xpath(section_xpath)
 126
 127         root.text = replacement.text
 128         root.tail = replacement.tail
 129         # replace all children
 130         del root[:]
 131         for child in replacement:
 132             root.append(copy.deepcopy(child))
 133
 134         return arch
 135
 136     def render(self, cr, uid, id_or_xml_id, values=None, engine='ir.qweb', context=None):
 137         if request and getattr(request, 'website_enabled', False):
 138             engine='website.qweb'
 139
 140             if isinstance(id_or_xml_id, list):
 141                 id_or_xml_id = id_or_xml_id[0]
 142
 143             if not context:
 144                 context = {}
 145
 146             company = self.pool['res.company'].browse(cr, SUPERUSER_ID, request.website.company_id.id, context=context)
 147
 148             qcontext = dict(
 149                 context.copy(),
 150                 website=request.website,
 151                 url_for=website.url_for,
 152                 slug=website.slug,
 153                 res_company=company,
 154                 user_id=self.pool.get("res.users").browse(cr, uid, uid),
 155                 translatable=context.get('lang') != request.website.default_lang_code,
 156                 editable=request.website.is_publisher(),
 157                 menu_data=self.pool['ir.ui.menu'].load_menus_root(cr, uid, context=context) if request.website.is_user() else None,
 158             )
 159
 160             # add some values
 161             if values:
 162                 qcontext.update(values)
 163
 164             # in edit mode ir.ui.view will tag nodes
 165             context = dict(context, inherit_branding=qcontext.get('editable', False))
 166
 167             view_obj = request.website.get_template(id_or_xml_id)
 168             if 'main_object' not in qcontext:
 169                 qcontext['main_object'] = view_obj
 170
 171             values = qcontext
 172
 173         return super(view, self).render(cr, uid, id_or_xml_id, values=values, engine=engine, context=context)
 174
 175     def _pretty_arch(self, arch):
 176         # remove_blank_string does not seem to work on HTMLParser, and
 177         # pretty-printing with lxml more or less requires stripping
 178         # whitespace: http://lxml.de/FAQ.html#why-doesn-t-the-pretty-print-option-reformat-my-xml-output
 179         # so serialize to XML, parse as XML (remove whitespace) then serialize
 180         # as XML (pretty print)
 181         arch_no_whitespace = etree.fromstring(
 182             etree.tostring(arch, encoding='utf-8'),
 183             parser=etree.XMLParser(encoding='utf-8', remove_blank_text=True))
 184         return etree.tostring(
 185             arch_no_whitespace, encoding='unicode', pretty_print=True)
 186
 187     def save(self, cr, uid, res_id, value, xpath=None, context=None):
 188         """ Update a view section. The view section may embed fields to write
 189
 190         :param str model:
 191         :param int res_id:
 192         :param str xpath: valid xpath to the tag to replace
 193         """
 194         res_id = int(res_id)
 195
 196         arch_section = html.fromstring(
 197             value, parser=html.HTMLParser(encoding='utf-8'))
 198
 199         if xpath is None:
 200             # value is an embedded field on its own, not a view section
 201             self.save_embedded_field(cr, uid, arch_section, context=context)
 202             return
 203
 204         for el in self.extract_embedded_fields(cr, uid, arch_section, context=context):
 205             self.save_embedded_field(cr, uid, el, context=context)
 206
 207             # transform embedded field back to t-field
 208             el.getparent().replace(el, self.to_field_ref(cr, uid, el, context=context))
 209
 210         arch = self.replace_arch_section(cr, uid, res_id, xpath, arch_section, context=context)
 211         self.write(cr, uid, res_id, {
 212             'arch': self._pretty_arch(arch)
 213         }, context=context)
 214
 215         view = self.browse(cr, SUPERUSER_ID, res_id, context=context)
 216         if view.model_data_id:
 217             view.model_data_id.write({'noupdate': True})
 218