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