1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
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.
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.
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/>.
20 ##############################################################################
22 from xml.dom import minidom
23 from osv.osv import osv_pool
24 from osv import fields, osv
30 objects_proxy = netsvc.ExportService.getService('object').__class__
31 class recording_objects_proxy(objects_proxy):
32 def execute(self, *args, **argv):
33 if len(args) >= 6 and isinstance(args[5], dict):
34 _old_args = args[5].copy()
37 res = super(recording_objects_proxy, self).execute(*args, **argv)
38 pool = pooler.get_pool(args[0])
39 mod = pool.get('ir.module.record')
40 if mod and mod.recording:
41 if args[4] in ('copy', 'write', 'unlink', 'create'):
42 if _old_args is not None:
43 args[5].update(_old_args)
45 mod.recording_data.append(('query', args, argv, res))
48 def exec_workflow(self, *args, **argv):
49 res = super(recording_objects_proxy, self).exec_workflow(*args, **argv)
50 pool = pooler.get_pool(args[0])
51 mod = pool.get('ir.module.record')
52 if mod and mod.recording:
53 mod.recording_data.append(('workflow', args, argv))
56 recording_objects_proxy()
58 class xElement(minidom.Element):
59 """dom.Element with compact print
60 The Element in minidom has a problem: if printed, adds whitespace
61 around the text nodes. The standard will not ignore that whitespace.
62 This class simply prints the contained nodes in their compact form, w/o
65 def writexml(self, writer, indent="", addindent="", newl=""):
67 minidom.Element.writexml(self, writer, indent='', addindent='', newl='')
70 def doc_createXElement(xdoc, tagName):
72 e.ownerDocument = xdoc
75 class base_module_record(osv.osv):
76 _name = "ir.module.record"
80 def __init__(self, *args, **kwargs):
82 self.recording_data = []
84 super(base_module_record, self).__init__(*args, **kwargs)
87 def _create_id(self, cr, uid, model, data):
91 name = filter(lambda x: x in string.letters, (data.get('name', '') or '').lower())
94 val = model.replace('.', '_') + '_' + name + str(i)
96 if val not in self.ids.values():
100 def _get_id(self, cr, uid, model, id):
101 if type(id) == type(()):
103 if (model, id) in self.ids:
104 return self.ids[(model, id)], False
105 dt = self.pool.get('ir.model.data')
106 dtids = dt.search(cr, uid, [('model', '=', model), ('res_id', '=', id)])
109 obj = dt.browse(cr, uid, dtids[0])
110 self.depends[obj.module] = True
111 return obj.module + '.' + obj.name, obj.noupdate
113 def _create_record(self, cr, uid, doc, model, data, record_id, noupdate=False):
114 record = doc.createElement('record')
115 record.setAttribute("id", record_id)
116 record.setAttribute("model", model)
117 record_list = [record]
118 lids = self.pool.get('ir.model.data').search(cr, uid, [('model', '=', model)])
119 res = self.pool.get('ir.model.data').read(cr, uid, lids[:1], ['module'])
121 self.depends[res[0]['module']] = True
122 fields = self.pool.get(model).fields_get(cr, uid)
123 for key, val in data.items():
124 if not (val or (fields[key]['type'] == 'boolean')):
126 if fields[key]['type'] in ('integer', 'float'):
127 field = doc.createElement('field')
128 field.setAttribute("name", key)
129 field.setAttribute("eval", val and str(val) or 'False')
130 record.appendChild(field)
131 elif fields[key]['type'] in ('boolean',):
132 field = doc.createElement('field')
133 field.setAttribute("name", key)
134 field.setAttribute("eval", val and '1' or '0')
135 record.appendChild(field)
136 elif fields[key]['type'] in ('many2one',):
137 field = doc.createElement('field')
138 field.setAttribute("name", key)
139 if type(val) in (type(''), type(u'')):
142 id, update = self._get_id(cr, uid, fields[key]['relation'], val)
143 noupdate = noupdate or update
145 field.setAttribute("model", fields[key]['relation'])
146 fld_nm = self.pool.get(fields[key]['relation'])._rec_name
147 name = self.pool.get(fields[key]['relation']).read(cr, uid, val, [fld_nm])[fld_nm] or False
148 field.setAttribute("search", str([(str(fld_nm) , '=', name)]))
150 field.setAttribute("ref", id)
151 record.appendChild(field)
152 elif fields[key]['type'] in ('one2many',):
153 for valitem in (val or []):
154 if valitem[0] in (0, 1):
155 if key in self.pool.get(model)._columns:
156 fname = self.pool.get(model)._columns[key]._fields_id
158 fname = self.pool.get(model)._inherit_fields[key][2]._fields_id
159 valitem[2][fname] = record_id
160 newid, update = self._get_id(cr, uid, fields[key]['relation'], valitem[1])
162 newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
163 self.ids[(fields[key]['relation'], valitem[1])] = newid
165 childrecord, update = self._create_record(cr, uid, doc, fields[key]['relation'], valitem[2], newid)
166 noupdate = noupdate or update
167 record_list += childrecord
170 elif fields[key]['type'] in ('many2many',):
172 for valitem in (val or []):
174 for id2 in valitem[2]:
175 id, update = self._get_id(cr, uid, fields[key]['relation'], id2)
176 self.ids[(fields[key]['relation'], id2)] = id
177 noupdate = noupdate or update
179 field = doc.createElement('field')
180 field.setAttribute("name", key)
181 field.setAttribute("eval", "[(6,0,[" + ','.join(map(lambda x: "ref('%s')" % (x,), res)) + '])]')
182 record.appendChild(field)
184 field = doc_createXElement(doc, 'field')
185 field.setAttribute("name", key)
186 field.appendChild(doc.createTextNode(val))
187 record.appendChild(field)
189 return record_list, noupdate
191 def get_copy_data(self, cr, uid, model, id, result):
193 obj = self.pool.get(model)
194 data = obj.read(cr, uid, [id])
195 if type(data) == type([]):
201 mod_fields = obj.fields_get(cr, uid)
202 for f in filter(lambda a: isinstance(obj._columns[a], fields.function)\
203 and (not obj._columns[a].store), obj._columns):
206 for key, val in data.items():
207 if result.has_key(key):
209 if mod_fields[key]['type'] == 'many2one':
210 if type(data[key]) == type(True) or type(data[key]) == type(1):
211 result[key] = data[key]
215 result[key] = data[key][0]
217 elif mod_fields[key]['type'] in ('one2many',):
218 continue # due to this start stop recording will not record one2many field
219 rel = mod_fields[key]['relation']
222 for rel_id in data[key]:
224 res.append(self.get_copy_data(cr, uid, rel, rel_id, {}))
228 result[key] = data[key]
230 elif mod_fields[key]['type'] == 'many2many':
231 result[key] = [(6, 0, data[key])]
234 result[key] = data[key]
235 for k, v in obj._inherits.items():
239 def _create_function(self, cr, uid, doc, model, name, record_id):
240 record = doc.createElement('function')
241 record.setAttribute("name", name)
242 record.setAttribute("model", model)
243 record_list = [record]
245 value = doc.createElement('value')
246 value.setAttribute('eval', '[ref(\'%s\')]' % (record_id,))
247 value.setAttribute('model', model)
249 record.appendChild(value)
250 return record_list, False
252 def _generate_object_xml(self, cr, uid, rec, recv, doc, result=None):
255 if rec[4] == 'write':
257 id, update = self._get_id(cr, uid, rec[3], id)
258 noupdate = noupdate or update
261 record, update = self._create_record(cr, uid, doc, rec[3], rec[6], id)
262 noupdate = noupdate or update
263 record_list += record
265 elif rec[4] in ('menu_create',):
267 id, update = self._get_id(cr, uid, rec[3], id)
268 noupdate = noupdate or update
271 record, update = self._create_function(cr, uid, doc, rec[3], rec[4], id)
272 noupdate = noupdate or update
273 record_list += record
275 elif rec[4] == 'create':
276 id = self._create_id(cr, uid, rec[3], rec[5])
277 record, noupdate = self._create_record(cr, uid, doc, rec[3], rec[5], id)
278 self.ids[(rec[3], result)] = id
279 record_list += record
281 elif rec[4] == 'copy':
282 data = self.get_copy_data(cr, uid, rec[3], rec[5], rec[6])
283 copy_rec = (rec[0], rec[1], rec[2], rec[3], rec[4], rec[5], data, rec[7])
285 rec_data = [(self.recording_data[0][0], rec, self.recording_data[0][2], self.recording_data[0][3])]
286 self.recording_data = rec_data
287 id = self._create_id(cr, uid, rec[3], rec[6])
288 record, noupdate = self._create_record(cr, uid, doc, rec[3], rec[6], id)
289 self.ids[(rec[3], result)] = id
290 record_list += record
292 return record_list, noupdate
294 def _generate_assert_xml(self, rec, doc):
297 def generate_xml(self, cr, uid):
298 # Create the minidom document
299 if len(self.recording_data):
301 doc = minidom.Document()
302 terp = doc.createElement("openerp")
303 doc.appendChild(terp)
304 for rec in self.recording_data:
305 if rec[0] == 'workflow':
306 rec_id, noupdate = self._get_id(cr, uid, rec[1][3], rec[1][5])
309 data = doc.createElement("data")
310 terp.appendChild(data)
311 wkf = doc.createElement('workflow')
312 data.appendChild(wkf)
313 wkf.setAttribute("model", rec[1][3])
314 wkf.setAttribute("action", rec[1][4])
316 data.setAttribute("noupdate", "1")
317 wkf.setAttribute("ref", rec_id)
318 if rec[0] == 'query':
319 res_list, noupdate = self._generate_object_xml(cr, uid, rec[1], rec[2], doc, rec[3])
320 data = doc.createElement("data")
322 data.setAttribute("noupdate", "1")
324 terp.appendChild(data)
326 data.appendChild(res)
327 elif rec[0] == 'assert':
329 return doc.toprettyxml(indent="\t").encode('utf-8')
331 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: