addons/base_module_record/base_module_record.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2008 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
  29 installed = False
  30
  31 def fnct_call(fnct):
  32     def execute(*args, **argv):
  33         if len(args) >= 6 and isinstance(args[5], dict):
  34             _old_args = args[5].copy()
  35         else:
  36             _old_args = None
  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)
  44                 if args[5]:
  45                     mod.recording_data.append(('query', args, argv,res))
  46         return res
  47     return execute
  48
  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))
  56         return res
  57     return exec_workflow
  58
  59 class base_module_record(osv.osv):
  60     _name = "ir.module.record"
  61     _columns = {
  62
  63     }
  64     def __init__(self, pool, cr=None):
  65         global installed
  66         if super(base_module_record, self).__init__.func_code.co_argcount ==3:
  67             super(base_module_record, self).__init__(pool,cr)
  68         else:
  69             super(base_module_record, self).__init__(pool)
  70         self.recording = 0
  71         self.recording_data = []
  72         self.depends = {}
  73         if not installed:
  74             obj  = netsvc._service['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)
  79             installed = True
  80
  81     # To Be Improved
  82     def _create_id(self, cr, uid, model, data):
  83         i = 0
  84         while True:
  85             try:
  86                 name = filter(lambda x: x in string.letters, (data.get('name','') or '').lower())
  87             except:
  88                 name=''
  89             val = model.replace('.','_')+'_'+name+ str(i)
  90             i+=1
  91             if val not in self.ids.values():
  92                 break
  93         return val
  94
  95     def _get_id(self, cr, uid, model, id):
  96         if type(id)==type(()):
  97             id=id[0]
  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)])
 102         if not dtids:
 103             return None, None
 104         obj = dt.browse(cr, uid, dtids[0])
 105         self.depends[obj.module] = True
 106         return obj.module+'.'+obj.name, obj.noupdate
 107
 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'])
 115         if res:
 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')):
 120                 continue
 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'')):
 135                     id = val
 136                 else:
 137                     id,update = self._get_id(cr, uid, fields[key]['relation'], val)
 138                     noupdate = noupdate or update
 139                 if not id:
 140                     field.setAttribute("model", fields[key]['relation'])
 141                     name = self.pool.get(fields[key]['relation']).browse(cr, uid, val).name
 142                     if isinstance(name, basestring):
 143                         name = name.decode('utf8')
 144                     field.setAttribute("search", "[('name','=','"+name+"')]")
 145                 else:
 146                     field.setAttribute("ref", id)
 147                 record.appendChild(field)
 148             elif fields[key]['type'] in ('one2many',):
 149                 for valitem in (val or []):
 150                     if valitem[0]==(0,1):
 151                         if key in self.pool.get(model)._columns:
 152                             fname = self.pool.get(model)._columns[key]._fields_id
 153                         else:
 154                             fname = self.pool.get(model)._inherit_fields[key][2]._fields_id
 155                         valitem[2][fname] = record_id
 156                         newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
 157                         childrecord, update = self._create_record(cr, uid, doc, fields[key]['relation'],valitem[2], newid)
 158                         noupdate = noupdate or update
 159                         record_list += childrecord
 160                         self.ids[(fields[key]['relation'],newid)] = newid
 161                     else:
 162                         pass
 163             elif fields[key]['type'] in ('many2many',):
 164                 res = []
 165                 for valitem in (val or []):
 166                     if valitem[0]==6:
 167                         for id2 in valitem[2]:
 168                             id,update = self._get_id(cr, uid, fields[key]['relation'], id2)
 169                             self.ids[(fields[key]['relation'],id)] = id
 170                             noupdate = noupdate or update
 171                             res.append(id)
 172                         field = doc.createElement('field')
 173                         field.setAttribute("name", key)
 174                         field.setAttribute("eval", "[(6,0,["+','.join(map(lambda x: "ref('%s')" % (x,), res))+'])]')
 175                         record.appendChild(field)
 176             else:
 177                 field = doc.createElement('field')
 178                 field.setAttribute("name", key)
 179
 180                 if not isinstance(val, basestring):
 181                     val = str(val)
 182
 183                 val = val and ('"""%s"""' % val.replace('\\', '\\\\').replace('"', '\"')) or 'False'
 184                 if isinstance(val, basestring):
 185                     val = val.decode('utf8')
 186                 field.setAttribute(u"eval",  val)
 187                 record.appendChild(field)
 188         return record_list, noupdate
 189
 190     def get_copy_data(self, cr, uid,model,id,result):
 191         res = []
 192         obj=self.pool.get(model)
 193         data=obj.read(cr, uid,[id])
 194         if type(data)==type([]):
 195             del data[0]['id']
 196             data=data[0]
 197         else:
 198             del data['id']
 199         mod_fields = obj.fields_get(cr, uid)
 200
 201         for f in filter(lambda a: isinstance(obj._columns[a], fields.function)\
 202                     and (not obj._columns[a].store),obj._columns):
 203             del data[f]
 204
 205         for key,val in data.items():
 206             if result.has_key(key):
 207                 continue
 208             if mod_fields[key]['type'] == 'many2one':
 209                 if type(data[key])==type(True) or type(data[key])==type(1):
 210                     result[key]=data[key]
 211                 else:
 212                     result[key]=data[key][0]
 213
 214             elif mod_fields[key]['type'] in ('one2many',):
 215                 rel = mod_fields[key]['relation']
 216                 if len(data[key]):
 217                     res1=[]
 218                     for rel_id in data[key]:
 219                         res=[0,0]
 220                         res.append(self.get_copy_data(cr, uid,rel,rel_id,{}))
 221                         res1.append(res)
 222                     result[key]=res1
 223                 else:
 224                     result[key]=data[key]
 225
 226             elif fields[key]['type'] == 'many2many':
 227                 result[key]=[(6,0,data[key])]
 228
 229             else:
 230                 result[key]=data[key]
 231         for k,v in self.pool.get(model)._inherits.items():
 232             del result[v]
 233         return result
 234
 235     def _generate_object_xml(self, cr, uid, rec, recv, doc, result=None):
 236         record_list = []
 237         noupdate = False
 238         if rec[4]=='write':
 239             for id in rec[5]:
 240                 id,update = self._get_id(cr, uid, rec[3], id)
 241                 noupdate = noupdate or update
 242                 if not id:
 243                     continue
 244                 record,update = self._create_record(cr, uid, doc, rec[3], rec[6], id)
 245                 noupdate = noupdate or update
 246                 record_list += record
 247
 248         elif rec[4]=='create':
 249             id = self._create_id(cr, uid, rec[3],rec[5])
 250             record,noupdate = self._create_record(cr, uid, doc, rec[3], rec[5], id)
 251             self.ids[(rec[3],result)] = id
 252             record_list += record
 253
 254         elif rec[4]=='copy':
 255             data=self.get_copy_data(cr,uid,rec[3],rec[5],rec[6])
 256             copy_rec=(rec[0],rec[1],rec[2],rec[3],rec[4],rec[5],data,rec[7])
 257             rec=copy_rec
 258             rec_data=[(self.recording_data[0][0],rec,self.recording_data[0][2],self.recording_data[0][3])]
 259             self.recording_data=rec_data
 260             id = self._create_id(cr, uid, rec[3],rec[6])
 261             record,noupdate = self._create_record(cr, uid, doc, rec[3], rec[6], id)
 262             self.ids[(rec[3],result)] = id
 263             record_list += record
 264
 265         return record_list,noupdate
 266
 267     def _generate_assert_xml(self, rec, doc):
 268         pass
 269     def generate_xml(self, cr, uid):
 270         # Create the minidom document
 271         if len(self.recording_data):
 272             self.ids = {}
 273             doc = minidom.Document()
 274             terp = doc.createElement("openerp")
 275             doc.appendChild(terp)
 276             for rec in self.recording_data:
 277                 if rec[0]=='workflow':
 278                     rec_id,noupdate = self._get_id(cr, uid, rec[1][3], rec[1][5])
 279                     if not rec_id:
 280                         continue
 281                     data = doc.createElement("data")
 282         self.ids = {}
 283         doc = minidom.Document()
 284         terp = doc.createElement("openerp")
 285         doc.appendChild(terp)
 286         for rec in self.recording_data:
 287             if rec[0]=='workflow':
 288                 rec_id,noupdate = self._get_id(cr, uid, rec[1][3], rec[1][5])
 289                 if not rec_id:
 290                     continue
 291                 data = doc.createElement("data")
 292                 terp.appendChild(data)
 293                 wkf = doc.createElement('workflow')
 294                 data.appendChild(wkf)
 295                 wkf.setAttribute("model", rec[1][3])
 296                 wkf.setAttribute("action", rec[1][4])
 297                 if noupdate:
 298                     data.setAttribute("noupdate", "1")
 299                 wkf.setAttribute("ref", rec_id)
 300             if rec[0]=='query':
 301                 res_list,noupdate = self._generate_object_xml(cr, uid, rec[1], rec[2], doc, rec[3])
 302                 data = doc.createElement("data")
 303                 if noupdate:
 304                     data.setAttribute("noupdate", "1")
 305                 if res_list:
 306                     terp.appendChild(data)
 307                     wkf = doc.createElement('workflow')
 308                     data.appendChild(wkf)
 309                     wkf.setAttribute("model", rec[1][3])
 310                     wkf.setAttribute("action", rec[1][4])
 311                     if noupdate:
 312                         data.setAttribute("noupdate", "1")
 313                     wkf.setAttribute("ref", rec_id)
 314                 if rec[0]=='query':
 315                     res_list,noupdate = self._generate_object_xml(cr, uid, rec[1], rec[2], doc, rec[3])
 316                     data = doc.createElement("data")
 317                     if noupdate:
 318                         data.setAttribute("noupdate", "1")
 319                     if res_list:
 320                         terp.appendChild(data)
 321                     for res in res_list:
 322                         data.appendChild(res)
 323                 elif rec[0]=='assert':
 324                     pass
 325             res = doc.toprettyxml(indent="\t")
 326             return  doc.toprettyxml(indent="\t").encode('utf8')
 327 base_module_record()
 328 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 329