1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2009 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 class recording_objects_proxy(osv_pool):
31 def execute(self, *args, **argv):
32 if len(args) >= 6 and isinstance(args[5], dict):
33 _old_args = args[5].copy()
36 res = super(recording_objects_proxy, self).execute(*args, **argv)
37 pool = pooler.get_pool(args[0])
38 mod = pool.get('ir.module.record')
39 if mod and mod.recording:
40 if args[4] in ('copy','write','unlink','create'):
41 if _old_args is not None:
42 args[5].update(_old_args)
44 mod.recording_data.append(('query', args, argv,res))
47 def exec_workflow(self, *args, **argv):
48 res = super(recording_objects_proxy, self).exec_workflow(*args, **argv)
49 pool = pooler.get_pool(args[0])
50 mod = pool.get('ir.module.record')
51 if mod and mod.recording:
52 mod.recording_data.append(('workflow', args, argv))
55 recording_objects_proxy()
58 class base_module_record(osv.osv):
59 _name = "ir.module.record"
63 def __init__(self, *args, **kwargs):
65 self.recording_data = []
67 super(base_module_record, self).__init__(*args, **kwargs)
70 def _create_id(self, cr, uid, model, data):
74 name = filter(lambda x: x in string.letters, (data.get('name','') or '').lower())
77 val = model.replace('.','_')+'_'+name+ str(i)
79 if val not in self.ids.values():
83 def _get_id(self, cr, uid, model, id):
84 if type(id)==type(()):
86 if (model,id) in self.ids:
87 return self.ids[(model,id)], False
88 dt = self.pool.get('ir.model.data')
89 dtids = dt.search(cr, uid, [('model','=',model), ('res_id','=',id)])
92 obj = dt.browse(cr, uid, dtids[0])
93 self.depends[obj.module] = True
94 return obj.module+'.'+obj.name, obj.noupdate
96 def _create_record(self, cr, uid, doc, model, data, record_id, noupdate=False):
97 record = doc.createElement('record')
98 record.setAttribute("id", record_id)
99 record.setAttribute("model", model)
100 record_list = [record]
101 lids = self.pool.get('ir.model.data').search(cr, uid, [('model','=',model)])
102 res = self.pool.get('ir.model.data').read(cr, uid, lids[:1], ['module'])
104 self.depends[res[0]['module']]=True
105 fields = self.pool.get(model).fields_get(cr, uid)
106 for key,val in data.items():
107 if not (val or (fields[key]['type']=='boolean')):
109 if fields[key]['type'] in ('integer','float'):
110 field = doc.createElement('field')
111 field.setAttribute("name", key)
112 field.setAttribute("eval", val and str(val) or 'False' )
113 record.appendChild(field)
114 elif fields[key]['type'] in ('boolean',):
115 field = doc.createElement('field')
116 field.setAttribute("name", key)
117 field.setAttribute("eval", val and '1' or '0' )
118 record.appendChild(field)
119 elif fields[key]['type'] in ('many2one',):
120 field = doc.createElement('field')
121 field.setAttribute("name", key)
122 if type(val) in (type(''),type(u'')):
125 id,update = self._get_id(cr, uid, fields[key]['relation'], val)
126 noupdate = noupdate or update
128 field.setAttribute("model", fields[key]['relation'])
129 fld_nm = self.pool.get(fields[key]['relation'])._rec_name
130 name = self.pool.get(fields[key]['relation']).read(cr, uid, val,[fld_nm])[fld_nm] or False
131 field.setAttribute("search", str([(str(fld_nm) ,'=', name)]))
133 field.setAttribute("ref", id)
134 record.appendChild(field)
135 elif fields[key]['type'] in ('one2many',):
136 for valitem in (val or []):
137 if valitem[0] in (0,1):
138 if key in self.pool.get(model)._columns:
139 fname = self.pool.get(model)._columns[key]._fields_id
141 fname = self.pool.get(model)._inherit_fields[key][2]._fields_id
142 valitem[2][fname] = record_id
143 newid,update = self._get_id(cr, uid, fields[key]['relation'], valitem[1])
145 newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
146 self.ids[(fields[key]['relation'], valitem[1])] = newid
148 childrecord, update = self._create_record(cr, uid, doc, fields[key]['relation'],valitem[2], newid)
149 noupdate = noupdate or update
150 record_list += childrecord
153 elif fields[key]['type'] in ('many2many',):
155 for valitem in (val or []):
157 for id2 in valitem[2]:
158 id,update = self._get_id(cr, uid, fields[key]['relation'], id2)
159 self.ids[(fields[key]['relation'],id2)] = id
160 noupdate = noupdate or update
162 field = doc.createElement('field')
163 field.setAttribute("name", key)
164 field.setAttribute("eval", "[(6,0,["+','.join(map(lambda x: "ref('%s')" % (x,), res))+'])]')
165 record.appendChild(field)
167 field = doc.createElement('field')
168 field.setAttribute("name", key)
170 if not isinstance(val, basestring):
173 val = val and ('"""%s"""' % val.replace('\\', '\\\\').replace('"', '\"')) or 'False'
174 field.setAttribute(u"eval", tools.ustr(val))
175 record.appendChild(field)
176 return record_list, noupdate
178 def get_copy_data(self, cr, uid, model, id, result):
180 obj=self.pool.get(model)
181 data=obj.read(cr, uid,[id])
182 if type(data)==type([]):
188 mod_fields = obj.fields_get(cr, uid)
189 for f in filter(lambda a: isinstance(obj._columns[a], fields.function)\
190 and (not obj._columns[a].store),obj._columns):
193 for key,val in data.items():
194 if result.has_key(key):
196 if mod_fields[key]['type'] == 'many2one':
197 if type(data[key])==type(True) or type(data[key])==type(1):
198 result[key]=data[key]
202 result[key]=data[key][0]
204 elif mod_fields[key]['type'] in ('one2many',):
205 continue # due to this start stop recording will not record one2many field
206 rel = mod_fields[key]['relation']
209 for rel_id in data[key]:
211 res.append(self.get_copy_data(cr, uid,rel,rel_id,{}))
215 result[key]=data[key]
217 elif mod_fields[key]['type'] == 'many2many':
218 result[key]=[(6,0,data[key])]
221 result[key]=data[key]
222 for k,v in obj._inherits.items():
226 def _create_function(self, cr, uid, doc, model, name, record_id):
227 record = doc.createElement('function')
228 record.setAttribute("name", name)
229 record.setAttribute("model", model)
230 record_list = [record]
232 value = doc.createElement('value')
233 value.setAttribute('eval', '[ref(\'%s\')]' % (record_id, ))
234 value.setAttribute('model', model)
236 record.appendChild(value)
237 return record_list, False
239 def _generate_object_xml(self, cr, uid, rec, recv, doc, result=None):
244 id,update = self._get_id(cr, uid, rec[3], id)
245 noupdate = noupdate or update
248 record,update = self._create_record(cr, uid, doc, rec[3], rec[6], id)
249 noupdate = noupdate or update
250 record_list += record
252 elif rec[4] in ('menu_create',):
254 id,update = self._get_id(cr, uid, rec[3], id)
255 noupdate = noupdate or update
258 record,update = self._create_function(cr, uid, doc, rec[3], rec[4], id)
259 noupdate = noupdate or update
260 record_list += record
262 elif rec[4]=='create':
263 id = self._create_id(cr, uid, rec[3],rec[5])
264 record,noupdate = self._create_record(cr, uid, doc, rec[3], rec[5], id)
265 self.ids[(rec[3], result)] = id
266 record_list += record
269 data=self.get_copy_data(cr,uid,rec[3],rec[5],rec[6])
270 copy_rec=(rec[0],rec[1],rec[2],rec[3],rec[4],rec[5],data,rec[7])
272 rec_data=[(self.recording_data[0][0],rec,self.recording_data[0][2],self.recording_data[0][3])]
273 self.recording_data=rec_data
274 id = self._create_id(cr, uid, rec[3],rec[6])
275 record,noupdate = self._create_record(cr, uid, doc, rec[3], rec[6], id)
276 self.ids[(rec[3], result)] = id
277 record_list += record
279 return record_list,noupdate
281 def _generate_assert_xml(self, rec, doc):
284 def generate_xml(self, cr, uid):
285 # Create the minidom document
286 if len(self.recording_data):
288 doc = minidom.Document()
289 terp = doc.createElement("openerp")
290 doc.appendChild(terp)
291 for rec in self.recording_data:
292 if rec[0]=='workflow':
293 rec_id,noupdate = self._get_id(cr, uid, rec[1][3], rec[1][5])
296 data = doc.createElement("data")
297 terp.appendChild(data)
298 wkf = doc.createElement('workflow')
299 data.appendChild(wkf)
300 wkf.setAttribute("model", rec[1][3])
301 wkf.setAttribute("action", rec[1][4])
303 data.setAttribute("noupdate", "1")
304 wkf.setAttribute("ref", rec_id)
306 res_list,noupdate = self._generate_object_xml(cr, uid, rec[1], rec[2], doc, rec[3])
307 data = doc.createElement("data")
309 data.setAttribute("noupdate", "1")
311 terp.appendChild(data)
313 data.appendChild(res)
314 elif rec[0]=='assert':
316 return doc.toprettyxml(indent="\t").encode('utf-8')
318 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: