openerp/addons/base/ir/ir_ui_view.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 from osv import fields,osv
  23 from lxml import etree
  24 from tools import graph
  25 from tools.safe_eval import safe_eval as eval
  26 import tools
  27 import os
  28 import logging
  29
  30 class view_custom(osv.osv):
  31     _name = 'ir.ui.view.custom'
  32     _order = 'create_date desc'  # search(limit=1) should return the last customization
  33     _columns = {
  34         'ref_id': fields.many2one('ir.ui.view', 'Original View', select=True, required=True, ondelete='cascade'),
  35         'user_id': fields.many2one('res.users', 'User', select=True, required=True, ondelete='cascade'),
  36         'arch': fields.text('View Architecture', required=True),
  37     }
  38
  39     def _auto_init(self, cr, context=None):
  40         super(view_custom, self)._auto_init(cr, context)
  41         cr.execute('SELECT indexname FROM pg_indexes WHERE indexname = \'ir_ui_view_custom_user_id_ref_id\'')
  42         if not cr.fetchone():
  43             cr.execute('CREATE INDEX ir_ui_view_custom_user_id_ref_id ON ir_ui_view_custom (user_id, ref_id)')
  44 view_custom()
  45
  46 class view(osv.osv):
  47     _name = 'ir.ui.view'
  48     _columns = {
  49         'name': fields.char('View Name',size=64,  required=True),
  50         'model': fields.char('Object', size=64, required=True, select=True),
  51         'priority': fields.integer('Sequence', required=True),
  52         'type': fields.selection((
  53             ('tree','Tree'),
  54             ('form','Form'),
  55             ('mdx','mdx'),
  56             ('graph', 'Graph'),
  57             ('calendar', 'Calendar'),
  58             ('diagram','Diagram'),
  59             ('gantt', 'Gantt'),
  60             ('kanban', 'Kanban'),
  61             ('search','Search')), 'View Type', required=True, select=True),
  62         'arch': fields.text('View Architecture', required=True),
  63         'inherit_id': fields.many2one('ir.ui.view', 'Inherited View', ondelete='cascade', select=True),
  64         'field_parent': fields.char('Child Field',size=64),
  65         'xml_id': fields.function(osv.osv.get_xml_id, type='char', size=128, string="External ID",
  66                                   help="ID of the view defined in xml file"),
  67     }
  68     _defaults = {
  69         'arch': '<?xml version="1.0"?>\n<tree string="My view">\n\t<field name="name"/>\n</tree>',
  70         'priority': 16
  71     }
  72     _order = "priority,name"
  73
  74     def _check_xml(self, cr, uid, ids, context=None):
  75         logger = logging.getLogger('init')
  76         for view in self.browse(cr, uid, ids, context):
  77             eview = etree.fromstring(view.arch.encode('utf8'))
  78             frng = tools.file_open(os.path.join('base','rng','view.rng'))
  79             try:
  80                 relaxng_doc = etree.parse(frng)
  81                 relaxng = etree.RelaxNG(relaxng_doc)
  82                 if not relaxng.validate(eview):
  83                     for error in relaxng.error_log:
  84                         logger.error(tools.ustr(error))
  85                     return False
  86             finally:
  87                 frng.close()
  88         return True
  89
  90     _constraints = [
  91         (_check_xml, 'Invalid XML for View Architecture!', ['arch'])
  92     ]
  93
  94     def _auto_init(self, cr, context=None):
  95         super(view, self)._auto_init(cr, context)
  96         cr.execute('SELECT indexname FROM pg_indexes WHERE indexname = \'ir_ui_view_model_type_inherit_id\'')
  97         if not cr.fetchone():
  98             cr.execute('CREATE INDEX ir_ui_view_model_type_inherit_id ON ir_ui_view (model, type, inherit_id)')
  99
 100     def get_inheriting_views_arch(self, cr, uid, view_id, model, context=None):
 101         """Retrieves the architecture of views that inherit from the given view.
 102
 103            :param int view_id: id of the view whose inheriting views should be retrieved
 104            :param str model: model identifier of the view's related model (for double-checking)
 105            :rtype: list of tuples
 106            :return: [(view_arch,view_id), ...]
 107         """
 108         cr.execute("""SELECT arch, id FROM ir_ui_view WHERE inherit_id=%s AND model=%s
 109                       ORDER BY priority""",
 110                       (view_id, model))
 111         return cr.fetchall()
 112
 113     def write(self, cr, uid, ids, vals, context=None):
 114         if not isinstance(ids, (list, tuple)):
 115             ids = [ids]
 116         result = super(view, self).write(cr, uid, ids, vals, context)
 117
 118         # drop the corresponding view customizations (used for dashboards for example), otherwise
 119         # not all users would see the updated views
 120         custom_view_ids = self.pool.get('ir.ui.view.custom').search(cr, uid, [('ref_id','in',ids)])
 121         if custom_view_ids:
 122             self.pool.get('ir.ui.view.custom').unlink(cr, uid, custom_view_ids)
 123
 124         return result
 125
 126     def graph_get(self, cr, uid, id, model, node_obj, conn_obj, src_node, des_node, label, scale, context=None):
 127         if not label:
 128             label = []
 129         nodes=[]
 130         nodes_name=[]
 131         transitions=[]
 132         start=[]
 133         tres={}
 134         labels={}
 135         no_ancester=[]
 136         blank_nodes = []
 137
 138         _Model_Obj=self.pool.get(model)
 139         _Node_Obj=self.pool.get(node_obj)
 140         _Arrow_Obj=self.pool.get(conn_obj)
 141
 142         for model_key,model_value in _Model_Obj._columns.items():
 143                 if model_value._type=='one2many':
 144                     if model_value._obj==node_obj:
 145                         _Node_Field=model_key
 146                         _Model_Field=model_value._fields_id
 147                     flag=False
 148                     for node_key,node_value in _Node_Obj._columns.items():
 149                         if node_value._type=='one2many':
 150                              if node_value._obj==conn_obj:
 151                                  if src_node in _Arrow_Obj._columns and flag:
 152                                     _Source_Field=node_key
 153                                  if des_node in _Arrow_Obj._columns and not flag:
 154                                     _Destination_Field=node_key
 155                                     flag = True
 156
 157         datas = _Model_Obj.read(cr, uid, id, [],context)
 158         for a in _Node_Obj.read(cr,uid,datas[_Node_Field],[]):
 159             if a[_Source_Field] or a[_Destination_Field]:
 160                 nodes_name.append((a['id'],a['name']))
 161                 nodes.append(a['id'])
 162             else:
 163                 blank_nodes.append({'id': a['id'],'name':a['name']})
 164
 165             if a.has_key('flow_start') and a['flow_start']:
 166                 start.append(a['id'])
 167             else:
 168                 if not a[_Source_Field]:
 169                     no_ancester.append(a['id'])
 170             for t in _Arrow_Obj.read(cr,uid, a[_Destination_Field],[]):
 171                 transitions.append((a['id'], t[des_node][0]))
 172                 tres[str(t['id'])] = (a['id'],t[des_node][0])
 173                 label_string = ""
 174                 if label:
 175                     for lbl in eval(label):
 176                         if t.has_key(tools.ustr(lbl)) and tools.ustr(t[lbl])=='False':
 177                             label_string = label_string + ' '
 178                         else:
 179                             label_string = label_string + " " + tools.ustr(t[lbl])
 180                 labels[str(t['id'])] = (a['id'],label_string)
 181         g  = graph(nodes, transitions, no_ancester)
 182         g.process(start)
 183         g.scale(*scale)
 184         result = g.result_get()
 185         results = {}
 186         for node in nodes_name:
 187             results[str(node[0])] = result[node[0]]
 188             results[str(node[0])]['name'] = node[1]
 189         return {'nodes': results,
 190                 'transitions': tres,
 191                 'label' : labels,
 192                 'blank_nodes': blank_nodes,
 193                 'node_parent_field': _Model_Field,}
 194 view()
 195
 196 class view_sc(osv.osv):
 197     _name = 'ir.ui.view_sc'
 198     _columns = {
 199         'name': fields.char('Shortcut Name', size=64), # Kept for backwards compatibility only - resource name used instead (translatable)
 200         'res_id': fields.integer('Resource Ref.', help="Reference of the target resource, whose model/table depends on the 'Resource Name' field."),
 201         'sequence': fields.integer('Sequence'),
 202         'user_id': fields.many2one('res.users', 'User Ref.', required=True, ondelete='cascade', select=True),
 203         'resource': fields.char('Resource Name', size=64, required=True, select=True)
 204     }
 205
 206     def _auto_init(self, cr, context=None):
 207         super(view_sc, self)._auto_init(cr, context)
 208         cr.execute('SELECT indexname FROM pg_indexes WHERE indexname = \'ir_ui_view_sc_user_id_resource\'')
 209         if not cr.fetchone():
 210             cr.execute('CREATE INDEX ir_ui_view_sc_user_id_resource ON ir_ui_view_sc (user_id, resource)')
 211
 212     def get_sc(self, cr, uid, user_id, model='ir.ui.menu', context=None):
 213         ids = self.search(cr, uid, [('user_id','=',user_id),('resource','=',model)], context=context)
 214         results = self.read(cr, uid, ids, ['res_id'], context=context)
 215         name_map = dict(self.pool.get(model).name_get(cr, uid, [x['res_id'] for x in results], context=context))
 216         # Make sure to return only shortcuts pointing to exisintg menu items.
 217         filtered_results = filter(lambda result: result['res_id'] in name_map, results)
 218         for result in filtered_results:
 219             result.update(name=name_map[result['res_id']])
 220         return filtered_results
 221
 222     _order = 'sequence,name'
 223     _defaults = {
 224         'resource': lambda *a: 'ir.ui.menu',
 225         'user_id': lambda obj, cr, uid, context: uid,
 226     }
 227     _sql_constraints = [
 228         ('shortcut_unique', 'unique(res_id, resource, user_id)', 'Shortcut for this menu already exists!'),
 229     ]
 230
 231 view_sc()
 232
 233 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 234