addons/web_diagram/controllers/main.py

   1 try:
   2     # embedded
   3     import openerp.addons.web.common.http as openerpweb
   4     from openerp.addons.web.controllers.main import View
   5 except ImportError:
   6     # standalone
   7     import web.common.http as openerpweb
   8     from web.controllers.main import View
   9
  10 class DiagramView(View):
  11     _cp_path = "/web_diagram/diagram"
  12
  13     @openerpweb.jsonrequest
  14     def load(self, req, model, view_id):
  15         fields_view = self.fields_view_get(req, model, view_id, 'diagram')
  16         return {'fields_view': fields_view}
  17
  18     @openerpweb.jsonrequest
  19     def get_diagram_info(self, req, id, model, node, connector,
  20                          src_node, des_node, label, **kw):
  21
  22         visible_node_fields = kw.get('visible_node_fields',[])
  23         invisible_node_fields = kw.get('invisible_node_fields',[])
  24         node_fields_string = kw.get('node_fields_string',[])
  25         connector_fields = kw.get('connector_fields',[])
  26         connector_fields_string = kw.get('connector_fields_string',[])
  27
  28         bgcolors = {}
  29         shapes = {}
  30         bgcolor = kw.get('bgcolor','')
  31         shape = kw.get('shape','')
  32
  33         if bgcolor:
  34             for color_spec in bgcolor.split(';'):
  35                 if color_spec:
  36                     colour, color_state = color_spec.split(':')
  37                     bgcolors[colour] = color_state
  38
  39         if shape:
  40             for shape_spec in shape.split(';'):
  41                 if shape_spec:
  42                     shape_colour, shape_color_state = shape_spec.split(':')
  43                     shapes[shape_colour] = shape_color_state
  44
  45         ir_view = req.session.model('ir.ui.view')
  46         graphs = ir_view.graph_get(
  47             int(id), model, node, connector, src_node, des_node, label,
  48             (140, 180), req.session.context)
  49         nodes = graphs['nodes']
  50         transitions = graphs['transitions']
  51         isolate_nodes = {}
  52         for blnk_node in graphs['blank_nodes']:
  53             isolate_nodes[blnk_node['id']] = blnk_node
  54         else:
  55             y = map(lambda t: t['y'],filter(lambda x: x['y'] if x['x']==20 else None, nodes.values()))
  56             y_max = (y and max(y)) or 120
  57
  58         connectors = {}
  59         list_tr = []
  60
  61         for tr in transitions:
  62             list_tr.append(tr)
  63             connectors.setdefault(tr, {
  64                 'id': tr,
  65                 's_id': transitions[tr][0],
  66                 'd_id': transitions[tr][1]
  67             })
  68         connector_tr = req.session.model(connector)
  69         connector_ids = connector_tr.search([('id', 'in', list_tr)], 0, 0, 0, req.session.context)
  70
  71         data_connectors =connector_tr.read(connector_ids, connector_fields, req.session.context)
  72
  73         for tr in data_connectors:
  74             transition_id = str(tr['id'])
  75             _sourceid, label = graphs['label'][transition_id]
  76             t = connectors[transition_id]
  77             t.update(
  78                 source=tr[src_node][1],
  79                 destination=tr[des_node][1],
  80                 options={},
  81                 signal=label
  82             )
  83
  84             for i, fld in enumerate(connector_fields):
  85                 t['options'][connector_fields_string[i]] = tr[fld]
  86
  87         fields = req.session.model('ir.model.fields')
  88         field_ids = fields.search([('model', '=', model), ('relation', '=', node)], 0, 0, 0, req.session.context)
  89         field_data = fields.read(field_ids, ['relation_field'], req.session.context)
  90         node_act = req.session.model(node)
  91         search_acts = node_act.search([(field_data[0]['relation_field'], '=', id)], 0, 0, 0, req.session.context)
  92         data_acts = node_act.read(search_acts, invisible_node_fields + visible_node_fields, req.session.context)
  93
  94         for act in data_acts:
  95             n = nodes.get(str(act['id']))
  96             if not n:
  97                 n = isolate_nodes.get(act['id'], {})
  98                 y_max += 140
  99                 n.update(x=20, y=y_max)
 100                 nodes[act['id']] = n
 101
 102             n.update(
 103                 id=act['id'],
 104                 color='white',
 105                 options={}
 106             )
 107             for color, expr in bgcolors.items():
 108                 if eval(expr, act):
 109                     n['color'] = color
 110
 111             for shape, expr in shapes.items():
 112                 if eval(expr, act):
 113                     n['shape'] = shape
 114
 115             for i, fld in enumerate(visible_node_fields):
 116                 n['options'][node_fields_string[i]] = act[fld]
 117
 118         id_model = req.session.model(model).read([id],['name'], req.session.context)[0]['name']
 119         return dict(nodes=nodes, conn=connectors, id_model=id_model)