1 # -*- encoding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
8 # This program is free software: you can redistribute it and/or modify
9 # it under the terms of the GNU General Public License as published by
10 # the Free Software Foundation, either version 3 of the License, or
11 # (at your option) any later version.
13 # This program is distributed in the hope that it will be useful,
14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 # GNU General Public License for more details.
18 # You should have received a copy of the GNU General Public License
19 # along with this program. If not, see <http://www.gnu.org/licenses/>.
21 ##############################################################################
23 from xml.dom import minidom
24 from osv import fields,osv
32 def execute(*args, **argv):
33 if len(args) >= 6 and isinstance(args[5], dict):
34 _old_args = args[5].copy()
37 res = fnct(*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] not in ('default_get','read','fields_view_get','fields_get','search','search_count','name_search','name_get','get','request_get', 'get_sc'):
42 if _old_args is not None:
43 args[5].update(_old_args)
45 mod.recording_data.append(('query', args, argv,res))
49 def fnct_call_workflow(fnct):
50 def exec_workflow(*args, **argv):
51 res = fnct(*args, **argv)
52 pool = pooler.get_pool(args[0])
53 mod = pool.get('ir.module.record')
54 if mod and mod.recording:
55 mod.recording_data.append(('workflow', args, argv))
59 class base_module_record(osv.osv):
60 _name = "ir.module.record"
64 def __init__(self, pool, cr=None):
66 if super(base_module_record, self).__init__.func_code.co_argcount ==3:
67 super(base_module_record, self).__init__(pool,cr)
69 super(base_module_record, self).__init__(pool)
71 self.recording_data = []
74 obj = netsvc.SERVICES['object']
75 obj.execute = fnct_call(obj.execute)
76 obj.exportMethod(obj.execute)
77 obj.exec_workflow = fnct_call_workflow(obj.exec_workflow)
78 obj.exportMethod(obj.exec_workflow)
82 def _create_id(self, cr, uid, model, data):
86 name = filter(lambda x: x in string.letters, (data.get('name','') or '').lower())
89 val = model.replace('.','_')+'_'+name+ str(i)
91 if val not in self.ids.values():
95 def _get_id(self, cr, uid, model, id):
96 if type(id)==type(()):
98 if (model,id) in self.ids:
99 return self.ids[(model,id)], False
100 dt = self.pool.get('ir.model.data')
101 dtids = dt.search(cr, uid, [('model','=',model), ('res_id','=',id)])
104 obj = dt.browse(cr, uid, dtids[0])
105 self.depends[obj.module] = True
106 return obj.module+'.'+obj.name, obj.noupdate
108 def _create_record(self, cr, uid, doc, model, data, record_id, noupdate=False):
109 record = doc.createElement('record')
110 record.setAttribute("id", record_id)
111 record.setAttribute("model", model)
112 record_list = [record]
113 lids = self.pool.get('ir.model.data').search(cr, uid, [('model','=',model)])
114 res = self.pool.get('ir.model.data').read(cr, uid, lids[:1], ['module'])
116 self.depends[res[0]['module']]=True
117 fields = self.pool.get(model).fields_get(cr, uid)
118 for key,val in data.items():
119 if not (val or (fields[key]['type']=='boolean')):
121 if fields[key]['type'] in ('integer','float'):
122 field = doc.createElement('field')
123 field.setAttribute("name", key)
124 field.setAttribute("eval", val and str(val) or 'False' )
125 record.appendChild(field)
126 elif fields[key]['type'] in ('boolean',):
127 field = doc.createElement('field')
128 field.setAttribute("name", key)
129 field.setAttribute("eval", val and '1' or '0' )
130 record.appendChild(field)
131 elif fields[key]['type'] in ('many2one',):
132 field = doc.createElement('field')
133 field.setAttribute("name", key)
134 if type(val) in (type(''),type(u'')):
137 id,update = self._get_id(cr, uid, fields[key]['relation'], val)
138 noupdate = noupdate or update
140 field.setAttribute("model", fields[key]['relation'])
141 name = self.pool.get(fields[key]['relation']).browse(cr, uid, val).name
142 field.setAttribute("search", "[('name','=','"+name+"')]")
144 field.setAttribute("ref", id)
145 record.appendChild(field)
146 elif fields[key]['type'] in ('one2many',):
147 for valitem in (val or []):
148 if valitem[0] in (0,1):
149 if key in self.pool.get(model)._columns:
150 fname = self.pool.get(model)._columns[key]._fields_id
152 fname = self.pool.get(model)._inherit_fields[key][2]._fields_id
153 valitem[2][fname] = record_id
154 newid,update = self._get_id(cr, uid, fields[key]['relation'], valitem[1])
156 newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
158 # newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
160 # newid,update = self._get_id(cr, uid, fields[key]['relation'], valitem[1])
161 childrecord, update = self._create_record(cr, uid, doc, fields[key]['relation'],valitem[2], newid)
162 noupdate = noupdate or update
163 record_list += childrecord
164 self.ids[(fields[key]['relation'],newid)] = newid
167 elif fields[key]['type'] in ('many2many',):
169 for valitem in (val or []):
171 for id2 in valitem[2]:
172 id,update = self._get_id(cr, uid, fields[key]['relation'], id2)
173 self.ids[(fields[key]['relation'],id)] = id
174 noupdate = noupdate or update
176 field = doc.createElement('field')
177 field.setAttribute("name", key)
178 field.setAttribute("eval", "[(6,0,["+','.join(map(lambda x: "ref('%s')" % (x,), res))+'])]')
179 record.appendChild(field)
181 field = doc.createElement('field')
182 field.setAttribute("name", key)
184 if not isinstance(val, basestring):
187 val = val and ('"""%s"""' % val.replace('\\', '\\\\').replace('"', '\"')) or 'False'
188 field.setAttribute(u"eval", val.decode('utf-8'))
189 record.appendChild(field)
190 return record_list, noupdate
192 def get_copy_data(self, cr, uid, model, id, result):
194 obj=self.pool.get(model)
195 data=obj.read(cr, uid,[id])
196 if type(data)==type([]):
202 mod_fields = obj.fields_get(cr, uid)
203 for f in filter(lambda a: isinstance(obj._columns[a], fields.function)\
204 and (not obj._columns[a].store),obj._columns):
207 for key,val in data.items():
208 if result.has_key(key):
210 if mod_fields[key]['type'] == 'many2one':
211 if type(data[key])==type(True) or type(data[key])==type(1):
212 result[key]=data[key]
214 result[key]=data[key][0]
216 elif mod_fields[key]['type'] in ('one2many',):
217 rel = mod_fields[key]['relation']
220 for rel_id in data[key]:
222 res.append(self.get_copy_data(cr, uid,rel,rel_id,{}))
226 result[key]=data[key]
228 elif mod_fields[key]['type'] == 'many2many':
229 result[key]=[(6,0,data[key])]
232 result[key]=data[key]
233 for k,v in obj._inherits.items():
237 def _create_function(self, cr, uid, doc, model, name, record_id):
238 record = doc.createElement('function')
239 record.setAttribute("name", name)
240 record.setAttribute("model", model)
241 record_list = [record]
243 value = doc.createElement('value')
244 value.setAttribute('eval', '[ref(\'%s\')]' % (record_id, ))
245 value.setAttribute('model', model)
247 record.appendChild(value)
248 return record_list, False
250 def _generate_object_xml(self, cr, uid, rec, recv, doc, result=None):
255 id,update = self._get_id(cr, uid, rec[3], id)
256 noupdate = noupdate or update
259 record,update = self._create_record(cr, uid, doc, rec[3], rec[6], id)
260 noupdate = noupdate or update
261 record_list += record
263 elif rec[4] in ('menu_create',):
265 id,update = self._get_id(cr, uid, rec[3], id)
266 noupdate = noupdate or update
269 record,update = self._create_function(cr, uid, doc, rec[3], rec[4], id)
270 noupdate = noupdate or update
271 record_list += record
273 elif rec[4]=='create':
274 id = self._create_id(cr, uid, rec[3],rec[5])
275 record,noupdate = self._create_record(cr, uid, doc, rec[3], rec[5], id)
276 self.ids[(rec[3],result)] = id
277 record_list += record
280 data=self.get_copy_data(cr,uid,rec[3],rec[5],rec[6])
281 copy_rec=(rec[0],rec[1],rec[2],rec[3],rec[4],rec[5],data,rec[7])
283 rec_data=[(self.recording_data[0][0],rec,self.recording_data[0][2],self.recording_data[0][3])]
284 self.recording_data=rec_data
285 id = self._create_id(cr, uid, rec[3],rec[6])
286 record,noupdate = self._create_record(cr, uid, doc, rec[3], rec[6], id)
287 self.ids[(rec[3],result)] = id
288 record_list += record
290 return record_list,noupdate
292 def _generate_assert_xml(self, rec, doc):
295 def generate_xml(self, cr, uid):
296 # Create the minidom document
297 if len(self.recording_data):
299 doc = minidom.Document()
300 terp = doc.createElement("openerp")
301 doc.appendChild(terp)
302 for rec in self.recording_data:
303 if rec[0]=='workflow':
304 rec_id,noupdate = self._get_id(cr, uid, rec[1][3], rec[1][5])
307 data = doc.createElement("data")
308 terp.appendChild(data)
309 wkf = doc.createElement('workflow')
310 data.appendChild(wkf)
311 wkf.setAttribute("model", rec[1][3])
312 wkf.setAttribute("action", rec[1][4])
314 data.setAttribute("noupdate", "1")
315 wkf.setAttribute("ref", rec_id)
317 res_list,noupdate = self._generate_object_xml(cr, uid, rec[1], rec[2], doc, rec[3])
318 data = doc.createElement("data")
320 data.setAttribute("noupdate", "1")
322 terp.appendChild(data)
324 data.appendChild(res)
325 elif rec[0]=='assert':
327 return doc.toprettyxml(indent="\t").encode('utf-8')
329 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: