addons/base_module_record/base_module_record.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 # Copyright (c) 2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
   5 #                    Fabien Pinckaers <fp@tiny.Be>
   6 #
   7 # WARNING: This program as such is intended to be used by professional
   8 # programmers who take the whole responsability of assessing all potential
   9 # consequences resulting from its eventual inadequacies and bugs
  10 # End users who are looking for a ready-to-use solution with commercial
  11 # garantees and support are strongly adviced to contract a Free Software
  12 # Service Company
  13 #
  14 # This program is Free Software; you can redistribute it and/or
  15 # modify it under the terms of the GNU General Public License
  16 # as published by the Free Software Foundation; either version 2
  17 # of the License, or (at your option) any later version.
  18 #
  19 # This program is distributed in the hope that it will be useful,
  20 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  21 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  22 # GNU General Public License for more details.
  23 #
  24 # You should have received a copy of the GNU General Public License
  25 # along with this program; if not, write to the Free Software
  26 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  27 #
  28 ##############################################################################
  29
  30 from xml.dom import minidom
  31 from osv import fields,osv
  32 import netsvc
  33 import pooler
  34 import string
  35
  36 class base_module_record(osv.osv):
  37     _name = "ir.module.record"
  38     _columns = {
  39
  40     }
  41     def __init__(self, pool, cr=None):
  42         if super(base_module_record, self).__init__.func_code.co_argcount ==3:
  43             super(base_module_record, self).__init__(pool,cr)
  44         else:
  45             super(base_module_record, self).__init__(pool)
  46         self.recording = 0
  47         self.recording_data = []
  48         self.depends = {}
  49
  50     # To Be Improved
  51     def _create_id(self, cr, uid, model, data):
  52         i = 0
  53         while True:
  54             name = filter(lambda x: x in string.letters, (data.get('name','') or '').lower())
  55             val = model.replace('.','_')+'_'+name+ str(i)
  56             i+=1
  57             if val not in self.ids.values():
  58                 break
  59         return val
  60
  61     def _get_id(self, cr, uid, model, id):
  62         if (model,id) in self.ids:
  63             return self.ids[(model,id)], False
  64         dt = self.pool.get('ir.model.data')
  65         dtids = dt.search(cr, uid, [('model','=',model), ('res_id','=',id)])
  66         if not dtids:
  67             return None, None
  68         obj = dt.browse(cr, uid, dtids[0])
  69         self.depends[obj.module] = True
  70         return obj.module+'.'+obj.name, obj.noupdate
  71
  72     def _create_record(self, cr, uid, doc, model, data, record_id, noupdate=False):
  73         record = doc.createElement('record')
  74         record.setAttribute("id", record_id)
  75         record.setAttribute("model", model)
  76         lids  = self.pool.get('ir.model.data').search(cr, uid, [('model','=',model)])
  77         res = self.pool.get('ir.model.data').read(cr, uid, lids[:1], ['module'])
  78         if res:
  79             self.depends[res[0]['module']]=True
  80         record_list = [record]
  81         fields = self.pool.get(model).fields_get(cr, uid)
  82 #        print "before>>>>",fields
  83 #        fields = self.pool.get(model).fields_view_get(cr, uid, context={})['fields']
  84 #        print "after>>>>",fields
  85         for key,val in data.items():
  86             print key,val
  87             if not (val or (fields[key]['type']=='boolean')):
  88                 continue
  89             if fields[key]['type'] in ('integer','float'):
  90                 field = doc.createElement('field')
  91                 field.setAttribute("name", key)
  92                 field.setAttribute("eval", val and str(val) or 'False' )
  93                 record.appendChild(field)
  94             elif fields[key]['type'] in ('boolean',):
  95                 field = doc.createElement('field')
  96                 field.setAttribute("name", key)
  97                 field.setAttribute("eval", val and '1' or '0' )
  98                 record.appendChild(field)
  99             elif fields[key]['type'] in ('many2one',):
 100                 field = doc.createElement('field')
 101                 field.setAttribute("name", key)
 102                 if type(val) in (type(''),type(u'')):
 103                     id = val
 104                 else:
 105                     id,update = self._get_id(cr, uid, fields[key]['relation'], val)
 106                     noupdate = noupdate or update
 107                 if not id:
 108                     field.setAttribute("model", fields[key]['relation'])
 109                     name = self.pool.get(fields[key]['relation']).browse(cr, uid, val).name
 110                     print name
 111                     field.setAttribute("search", "[('name','=','"+name+"')]")
 112                 else:
 113                     field.setAttribute("ref", id)
 114                 record.appendChild(field)
 115             elif fields[key]['type'] in ('one2many',):
 116                 for valitem in (val or []):
 117                     if valitem[0]==0:
 118                         if key in self.pool.get(model)._columns:
 119                             fname = self.pool.get(model)._columns[key]._fields_id
 120                         else:
 121                             fname = self.pool.get(model)._inherit_fields[key]._fields_id
 122                         valitem[2][fname] = record_id
 123                         newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
 124                         childrecord, update = self._create_record(cr, uid, doc, fields[key]['relation'],valitem[2], newid)
 125                         noupdate = noupdate or update
 126                         record_list += childrecord
 127                         self.ids[(fields[key]['relation'],newid)] = newid
 128                     else:
 129                         pass
 130             elif fields[key]['type'] in ('many2many',):
 131                 res = []
 132                 for valitem in (val or []):
 133                     if valitem[0]==6:
 134                         for id2 in valitem[2]:
 135                             id,update = self._get_id(cr, uid, fields[key]['relation'], id2)
 136                             self.ids[(fields[key]['relation'],id)] = id
 137                             noupdate = noupdate or update
 138                             res.append(id)
 139                         field = doc.createElement('field')
 140                         field.setAttribute("name", key)
 141                         field.setAttribute("eval", "[(6,0,["+','.join(map(lambda x: "ref('%s')" % (x,), res))+'])]')
 142                         record.appendChild(field)
 143             else:
 144                 field = doc.createElement('field')
 145                 field.setAttribute("name", key)
 146
 147                 if not isinstance(val, basestring):
 148                     val = str(val)
 149
 150                 val = val and ('"""%s"""' % val.replace('\\', '\\\\').replace('"', '\"')) or 'False'
 151                 if isinstance(val, basestring):
 152                     val = val.decode('utf8')
 153
 154                 field.setAttribute(u"eval",  val)
 155                 record.appendChild(field)
 156         return record_list, noupdate
 157
 158     def _generate_object_xml(self, cr, uid, rec, recv, doc, result=None):
 159         print "_generate_object_xml>>>",rec,recv,doc
 160         record_list = []
 161         noupdate = False
 162         if rec[4]=='write':
 163             for id in rec[5]:
 164                 id,update = self._get_id(cr, uid, rec[3], id)
 165                 noupdate = noupdate or update
 166                 if not id:
 167                     continue
 168                 record,update = self._create_record(cr, uid, doc, rec[3], rec[6], id)
 169                 noupdate = noupdate or update
 170                 record_list += record
 171         elif rec[4]=='create':
 172             id = self._create_id(cr, uid, rec[3],rec[5])
 173             record,noupdate = self._create_record(cr, uid, doc, rec[3], rec[5], id)
 174             self.ids[(rec[3],result)] = id
 175             record_list += record
 176         return record_list,noupdate
 177     def _generate_assert_xml(self, rec, doc):
 178         pass
 179     def generate_xml(self, cr, uid):
 180         # Create the minidom document
 181         self.ids = {}
 182         doc = minidom.Document()
 183         terp = doc.createElement("terp")
 184         doc.appendChild(terp)
 185         for rec in self.recording_data:
 186             if rec[0]=='workflow':
 187                 rec_id,noupdate = self._get_id(cr, uid, rec[1][3], rec[1][5])
 188                 if not rec_id:
 189                     continue
 190                 data = doc.createElement("data")
 191                 terp.appendChild(data)
 192                 wkf = doc.createElement('workflow')
 193                 data.appendChild(wkf)
 194                 wkf.setAttribute("model", rec[1][3])
 195                 wkf.setAttribute("action", rec[1][4])
 196                 if noupdate:
 197                     data.setAttribute("noupdate", "1")
 198                 wkf.setAttribute("ref", rec_id)
 199             if rec[0]=='query':
 200                 res_list,noupdate = self._generate_object_xml(cr, uid, rec[1], rec[2], doc, rec[3])
 201                 data = doc.createElement("data")
 202                 if noupdate:
 203                     data.setAttribute("noupdate", "1")
 204                 if res_list:
 205                     terp.appendChild(data)
 206                 for res in res_list:
 207                     data.appendChild(res)
 208             elif rec[0]=='assert':
 209                 pass
 210         res = doc.toprettyxml(indent="\t")
 211         return  doc.toprettyxml(indent="\t").encode('utf8')
 212 base_module_record()
 213
 214 def fnct_call(fnct):
 215     def execute(*args, **argv):
 216         if len(args) >= 6 and isinstance(args[5], dict):
 217             _old_args = args[5].copy()
 218         else:
 219             _old_args = None
 220         res = fnct(*args, **argv)
 221         pool = pooler.get_pool(args[0])
 222         mod = pool.get('ir.module.record')
 223         if mod and mod.recording:
 224             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'):
 225                 if _old_args is not None:
 226                     args[5].update(_old_args)
 227                 mod.recording_data.append(('query', args, argv,res))
 228         return res
 229     return execute
 230
 231 def fnct_call_workflow(fnct):
 232     def exec_workflow(*args, **argv):
 233         res = fnct(*args, **argv)
 234         pool = pooler.get_pool(args[0])
 235         mod = pool.get('ir.module.record')
 236         if mod and mod.recording:
 237             mod.recording_data.append(('workflow', args, argv))
 238         return res
 239     return exec_workflow
 240
 241 obj  = netsvc._service['object']
 242 obj.execute = fnct_call(obj.execute)
 243 obj.exportMethod(obj.execute)
 244 obj.exec_workflow = fnct_call_workflow(obj.exec_workflow)
 245 obj.exportMethod(obj.exec_workflow)
 246
 247 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 248