addons/web_diagram/controllers/main.py

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