addons/web_diagram/controllers/main.py

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