addons/base_module_record/base_module_record.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>). All Rights Reserved
   6 #    $Id$
   7 #
   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.
  12 #
  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.
  17 #
  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/>.
  20 #
  21 ##############################################################################
  22
  23 from xml.dom import minidom
  24 from osv import fields,osv
  25 import netsvc
  26 import pooler
  27 import string
  28 import tools
  29
  30 objects_proxy = netsvc.SERVICES['object'].__class__
  31
  32 class recording_objects_proxy(objects_proxy):
  33     def execute(self, *args, **argv):
  34         if len(args) >= 6 and isinstance(args[5], dict):
  35             _old_args = args[5].copy()
  36         else:
  37             _old_args = None
  38         res = super(recording_objects_proxy, self).execute(*args, **argv)
  39         pool = pooler.get_pool(args[0])
  40         mod = pool.get('ir.module.record')
  41         if mod and mod.recording:
  42             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'):
  43                 if _old_args is not None:
  44                     args[5].update(_old_args)
  45                     if args[5]:
  46                         mod.recording_data.append(('query', args, argv,res))
  47         return res
  48
  49     def exec_workflow(self, *args, **argv):
  50         res = super(recording_objects_proxy, self).exec_workflow(*args, **argv)
  51         pool = pooler.get_pool(args[0])
  52         mod = pool.get('ir.module.record')
  53         if mod and mod.recording:
  54             mod.recording_data.append(('workflow', args, argv))
  55         return res
  56
  57 recording_objects_proxy()
  58
  59
  60 class base_module_record(osv.osv):
  61     _name = "ir.module.record"
  62     _columns = {
  63
  64     }
  65     def __init__(self, *args, **kwargs):
  66         self.recording = 0
  67         self.recording_data = []
  68         self.depends = {}
  69         super(base_module_record, self).__init__(*args, **kwargs)
  70
  71     # To Be Improved
  72     def _create_id(self, cr, uid, model, data):
  73         i = 0
  74         while True:
  75             try:
  76                 name = filter(lambda x: x in string.letters, (data.get('name','') or '').lower())
  77             except:
  78                 name=''
  79             val = model.replace('.','_')+'_'+name+ str(i)
  80             i+=1
  81             if val not in self.ids.values():
  82                 break
  83         return val
  84
  85     def _get_id(self, cr, uid, model, id):
  86         if type(id)==type(()):
  87             id=id[0]
  88         if (model,id) in self.ids:
  89             return self.ids[(model,id)], False
  90         dt = self.pool.get('ir.model.data')
  91         dtids = dt.search(cr, uid, [('model','=',model), ('res_id','=',id)])
  92         if not dtids:
  93             return None, None
  94         obj = dt.browse(cr, uid, dtids[0])
  95         self.depends[obj.module] = True
  96         return obj.module+'.'+obj.name, obj.noupdate
  97
  98     def _create_record(self, cr, uid, doc, model, data, record_id, noupdate=False):
  99         record = doc.createElement('record')
 100         record.setAttribute("id", record_id)
 101         record.setAttribute("model", model)
 102         record_list = [record]
 103         lids  = self.pool.get('ir.model.data').search(cr, uid, [('model','=',model)])
 104         res = self.pool.get('ir.model.data').read(cr, uid, lids[:1], ['module'])
 105         if res:
 106             self.depends[res[0]['module']]=True
 107         fields = self.pool.get(model).fields_get(cr, uid)
 108         for key,val in data.items():
 109             if not (val or (fields[key]['type']=='boolean')):
 110                 continue
 111             if fields[key]['type'] in ('integer','float'):
 112                 field = doc.createElement('field')
 113                 field.setAttribute("name", key)
 114                 field.setAttribute("eval", val and str(val) or 'False' )
 115                 record.appendChild(field)
 116             elif fields[key]['type'] in ('boolean',):
 117                 field = doc.createElement('field')
 118                 field.setAttribute("name", key)
 119                 field.setAttribute("eval", val and '1' or '0' )
 120                 record.appendChild(field)
 121             elif fields[key]['type'] in ('many2one',):
 122                 field = doc.createElement('field')
 123                 field.setAttribute("name", key)
 124                 if type(val) in (type(''),type(u'')):
 125                     id = val
 126                 else:
 127                     id,update = self._get_id(cr, uid, fields[key]['relation'], val)
 128                     noupdate = noupdate or update
 129                 if not id:
 130                     field.setAttribute("model", fields[key]['relation'])
 131                     fld_nm = self.pool.get(fields[key]['relation'])._rec_name
 132                     name = self.pool.get(fields[key]['relation']).read(cr, uid, val,[fld_nm])[fld_nm] or False
 133                     field.setAttribute("search", str([(str(fld_nm) ,'=', name)]))
 134                 else:
 135                     field.setAttribute("ref", id)
 136                 record.appendChild(field)
 137             elif fields[key]['type'] in ('one2many',):
 138                 for valitem in (val or []):
 139                     if valitem[0] in (0,1):
 140                         if key in self.pool.get(model)._columns:
 141                             fname = self.pool.get(model)._columns[key]._fields_id
 142                         else:
 143                             fname = self.pool.get(model)._inherit_fields[key][2]._fields_id
 144                         valitem[2][fname] = record_id
 145                         newid,update = self._get_id(cr, uid, fields[key]['relation'], valitem[1])
 146                         if not newid:
 147                             newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
 148                         self.ids[(fields[key]['relation'], valitem[1])] = newid
 149
 150                         childrecord, update = self._create_record(cr, uid, doc, fields[key]['relation'],valitem[2], newid)
 151                         noupdate = noupdate or update
 152                         record_list += childrecord
 153                     else:
 154                         pass
 155             elif fields[key]['type'] in ('many2many',):
 156                 res = []
 157                 for valitem in (val or []):
 158                     if valitem[0]==6:
 159                         for id2 in valitem[2]:
 160                             id,update = self._get_id(cr, uid, fields[key]['relation'], id2)
 161                             self.ids[(fields[key]['relation'],id2)] = id
 162                             noupdate = noupdate or update
 163                             res.append(id)
 164                         field = doc.createElement('field')
 165                         field.setAttribute("name", key)
 166                         field.setAttribute("eval", "[(6,0,["+','.join(map(lambda x: "ref('%s')" % (x,), res))+'])]')
 167                         record.appendChild(field)
 168             else:
 169                 field = doc.createElement('field')
 170                 field.setAttribute("name", key)
 171
 172                 if not isinstance(val, basestring):
 173                     val = str(val)
 174
 175                 val = val and ('"""%s"""' % val.replace('\\', '\\\\').replace('"', '\"')) or 'False'
 176                 field.setAttribute(u"eval",  tools.ustr(val))
 177                 record.appendChild(field)
 178         return record_list, noupdate
 179
 180     def get_copy_data(self, cr, uid, model, id, result):
 181         res = []
 182         obj=self.pool.get(model)
 183         data=obj.read(cr, uid,[id])
 184         if type(data)==type([]):
 185             del data[0]['id']
 186             data=data[0]
 187         else:
 188             del data['id']
 189
 190         mod_fields = obj.fields_get(cr, uid)
 191         for f in filter(lambda a: isinstance(obj._columns[a], fields.function)\
 192                     and (not obj._columns[a].store),obj._columns):
 193             del data[f]
 194
 195         for key,val in data.items():
 196             if result.has_key(key):
 197                 continue
 198             if mod_fields[key]['type'] == 'many2one':
 199                 if type(data[key])==type(True) or type(data[key])==type(1):
 200                     result[key]=data[key]
 201                 else:
 202                     result[key]=data[key][0]
 203
 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']
 207                 if len(data[key]):
 208                     res1=[]
 209                     for rel_id in data[key]:
 210                         res=[0,0]
 211                         res.append(self.get_copy_data(cr, uid,rel,rel_id,{}))
 212                         res1.append(res)
 213                     result[key]=res1
 214                 else:
 215                     result[key]=data[key]
 216
 217             elif mod_fields[key]['type'] == 'many2many':
 218                 result[key]=[(6,0,data[key])]
 219
 220             else:
 221                 result[key]=data[key]
 222         for k,v in obj._inherits.items():
 223             del result[v]
 224         return result
 225
 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]
 231
 232         value = doc.createElement('value')
 233         value.setAttribute('eval', '[ref(\'%s\')]' % (record_id, ))
 234         value.setAttribute('model', model)
 235
 236         record.appendChild(value)
 237         return record_list, False
 238
 239     def _generate_object_xml(self, cr, uid, rec, recv, doc, result=None):
 240         record_list = []
 241         noupdate = False
 242         if rec[4]=='write':
 243             for id in rec[5]:
 244                 id,update = self._get_id(cr, uid, rec[3], id)
 245                 noupdate = noupdate or update
 246                 if not id:
 247                     continue
 248                 record,update = self._create_record(cr, uid, doc, rec[3], rec[6], id)
 249                 noupdate = noupdate or update
 250                 record_list += record
 251
 252         elif rec[4] in ('menu_create',):
 253             for id in rec[5]:
 254                 id,update = self._get_id(cr, uid, rec[3], id)
 255                 noupdate = noupdate or update
 256                 if not id:
 257                     continue
 258                 record,update = self._create_function(cr, uid, doc, rec[3], rec[4], id)
 259                 noupdate = noupdate or update
 260                 record_list += record
 261
 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
 267
 268         elif rec[4]=='copy':
 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])
 271             rec=copy_rec
 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
 278
 279         return record_list,noupdate
 280
 281     def _generate_assert_xml(self, rec, doc):
 282         pass
 283
 284     def generate_xml(self, cr, uid):
 285         # Create the minidom document
 286         if len(self.recording_data):
 287             self.ids = {}
 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])
 294                     if not rec_id:
 295                         continue
 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])
 302                     if noupdate:
 303                         data.setAttribute("noupdate", "1")
 304                     wkf.setAttribute("ref", rec_id)
 305                 if rec[0]=='query':
 306                     res_list,noupdate = self._generate_object_xml(cr, uid, rec[1], rec[2], doc, rec[3])
 307                     data = doc.createElement("data")
 308                     if noupdate:
 309                         data.setAttribute("noupdate", "1")
 310                     if res_list:
 311                         terp.appendChild(data)
 312                     for res in res_list:
 313                         data.appendChild(res)
 314                 elif rec[0]=='assert':
 315                         pass
 316             return doc.toprettyxml(indent="\t").encode('utf-8')
 317 base_module_record()
 318 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 319