[MERGE] merge with main addons
[odoo/odoo.git] / addons / base_module_record / base_module_record.py
index a48bf68..af63a0c 100644 (file)
 ##############################################################################
 
 from xml.dom import minidom
-from osv.osv import osv_pool
 from osv import fields,osv
-import netsvc
 import pooler
 import string
 import tools
 
-#objects_proxy = netsvc.ExportService.getService('object').__class__
-class recording_objects_proxy(osv_pool):
-    def execute(self, *args, **argv):
-        if args[3] == 'create':
-            _old_args = args[4].copy()
-        elif args[3] == 'write':
-            _old_args = args[5].copy()
-        elif len(args) >= 5 and isinstance(args[4], dict):
-            _old_args = args[4].copy()
-        elif len(args) > 5 and args[3] != 'write' and isinstance(args[5], dict):
-            _old_args = args[5].copy()
-        else:
-            _old_args = None
-        res = super(recording_objects_proxy, self).execute(*args, **argv)
-        pool = pooler.get_pool(args[0])
-        mod = pool.get('ir.module.record')
-
-        if mod and mod.recording:
-            if args[3] not in ('default_get','read','fields_view_get','fields_get','search','search_count','name_search','name_get','get','request_get', 'get_sc', 'unlink'):
-                if _old_args is not None:
-                    if args[3] != 'write' and args[3] != 'create' and len(args) > 5 and isinstance(args[5], dict):
-                       args=list(args)
-                       args[5]=_old_args
-                       args=tuple(args)
-                       mod.recording_data.append(('osv_memory_action', args, argv ,None))
-                    else:
-                       if args[3] == 'create':
-                           args[4].update(_old_args)
-                       elif args[3] == 'write':
-                           args[5].update(_old_args)
-                       mod.recording_data.append(('query', args, argv,res))
-        return res
-
-    def exec_workflow(self, *args, **argv):
-        res = super(recording_objects_proxy, self).exec_workflow(*args, **argv)
-        pool = pooler.get_pool(args[0])
-        mod = pool.get('ir.module.record')
-        if mod and mod.recording:
-            mod.recording_data.append(('workflow', args, argv))
-        return res
-
-recording_objects_proxy()
 
 class xElement(minidom.Element):
     """dom.Element with compact print
@@ -88,47 +44,8 @@ def doc_createXElement(xdoc, tagName):
         return e
 
 import yaml
-
-class record(yaml.YAMLObject):
-    yaml_tag = u'!record'
-    def __init__(self, model, id=None, attrs={}):
-        self.model = model
-        self.id = id
-        self.attrs=attrs
-    def __repr__(self):
-        return '!record {model: %s, id: %s}:' % (str(self.model,), str(self.id,))
-
-class workflow(yaml.YAMLObject):
-    yaml_tag = u'!workflow'
-    def __init__(self, model, action, ref=None):
-        self.model = model
-        self.ref = ref
-        self.action=action
-    def __repr__(self):
-        return '!workflow {model: %s, action: %s, ref: %s}' % (str(self.model,), str(self.action,), str(self.ref,))
-
-class ref(yaml.YAMLObject):
-    yaml_tag = u'!ref'
-    def __init__(self, expr="False"):
-        self.expr = expr
-    def __repr__(self):
-        return 'ref(%s)' % (str(self.expr,))
-
-class eval(yaml.YAMLObject):
-    yaml_tag = u'!eval'
-    def __init__(self, expr="False"):
-        self.expr = expr
-    def __repr__(self):
-        return 'eval(%s)' % (str(self.expr,))
-
-class function(yaml.YAMLObject):
-    yaml_tag = u'!python'
-    def __init__(self, model, action=None,attrs={}):
-        self.model = model
-        self.action = action
-        self.attrs=attrs
-    def __repr__(self):
-        return '!python {model: %s}: |' % (str(self.model), )
+from tools import yaml_tag # This import is not unused! Do not remove!
+# Please do not override yaml_tag here: modify it in server bin/tools/yaml_tag.py
 
 class base_module_record(osv.osv):
     _name = "ir.module.record"
@@ -169,16 +86,16 @@ class base_module_record(osv.osv):
         obj = dt.browse(cr, uid, dtids[0])
         self.depends[obj.module] = True
         return obj.module+'.'+obj.name, obj.noupdate
-
+    
     def _create_record(self, cr, uid, doc, model, data, record_id, noupdate=False):
         data_pool = self.pool.get('ir.model.data')
         model_pool = self.pool.get(model)
-
+        
         record = doc.createElement('record')
         record.setAttribute("id", record_id)
         record.setAttribute("model", model)
         record_list = [record]
-
+        
         lids  = data_pool.search(cr, uid, [('model','=',model)])
         res = data_pool.read(cr, uid, lids[:1], ['module'])
         if res:
@@ -187,7 +104,8 @@ class base_module_record(osv.osv):
         for key,val in data.items():
             if not (val or (fields[key]['type']=='boolean')):
                 continue
-            if fields[key]['type'] in ('integer','float'):
+            if (fields[key]['type'] in ('integer','float') or
+                fields[key]['type'] == 'selection' and isinstance(val, int)):
                 field = doc.createElement('field')
                 field.setAttribute("name", key)
                 field.setAttribute("eval", val and str(val) or 'False' )
@@ -207,7 +125,7 @@ class base_module_record(osv.osv):
                     noupdate = noupdate or update
                 if not id:
                     relation_pool = self.pool.get(fields[key]['relation'])
-
+                    
                     field.setAttribute("model", fields[key]['relation'])
                     fld_nm = relation_pool._rec_name
                     name = relation_pool.read(cr, uid, val,[fld_nm])[fld_nm] or False
@@ -219,9 +137,9 @@ class base_module_record(osv.osv):
                 for valitem in (val or []):
                     if valitem[0] in (0,1):
                         if key in model_pool._columns:
-                            fname = model_pool._columns[key]._fields_id
+                            model_pool._columns[key]._fields_id
                         else:
-                            fname = model_pool._inherit_fields[key][2]._fields_id
+                            model_pool._inherit_fields[key][2]._fields_id
                         if valitem[0] == 0:
                             newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
                             valitem[1]=newid
@@ -231,7 +149,7 @@ class base_module_record(osv.osv):
                                 newid = self._create_id(cr, uid, fields[key]['relation'], valitem[2])
                                 valitem[1]=newid
                         self.ids[(fields[key]['relation'], valitem[1])] = newid
-
+                        
                         childrecord, update = self._create_record(cr, uid, doc, fields[key]['relation'],valitem[2], newid)
                         noupdate = noupdate or update
                         record_list += childrecord
@@ -260,11 +178,11 @@ class base_module_record(osv.osv):
 
     def _create_yaml_record(self, cr, uid, model, data, record_id):
         record={'model': model, 'id': str(record_id)}
-
+        
         model_pool = self.pool.get(model)
         data_pool = self.pool.get('ir.model.data')
         lids  = data_pool.search(cr, uid, [('model','=',model)])
-
+        
         res = data_pool.read(cr, uid, lids[:1], ['module'])
         attrs={}
         if res:
@@ -275,7 +193,7 @@ class base_module_record(osv.osv):
             defaults[model] = model_pool.default_get(cr, uid, data)
         except:
             defaults[model]={}
-        for key,val in data.items():
+        for key,val in data.items():  
             if ((key in defaults[model]) and (val ==  defaults[model][key])) and not(fields[key].get('required',False)):
                 continue
             if fields[key]['type'] in ('integer','float'):
@@ -303,7 +221,7 @@ class base_module_record(osv.osv):
                         else:
                             fname = model_pool._inherit_fields[key][2]._fields_id
                         del valitem[2][fname] #delete parent_field from child's fields list
-
+                        
                         childrecord = self._create_yaml_record(cr, uid, fields[key]['relation'],valitem[2], None)
                         items[0].append(childrecord['attrs'])
                 attrs[key] = items
@@ -343,7 +261,7 @@ class base_module_record(osv.osv):
         for f in filter(lambda a: isinstance(obj._columns[a], fields.function)\
                     and (not obj._columns[a].store),obj._columns):
             del data[f]
-
+            
         for key,val in data.items():
             if result.has_key(key):
                 continue
@@ -378,7 +296,6 @@ class base_module_record(osv.osv):
         return result
 
     def _create_function(self, cr, uid, doc, model, name, record_id):
-        print "creareeeeeeeeeeeeeeeeee"
         record = doc.createElement('function')
         record.setAttribute("name", name)
         record.setAttribute("model", model)
@@ -403,7 +320,7 @@ class base_module_record(osv.osv):
                 record,update = self._create_record(cr, uid, doc, rec[2], rec[5], id)
                 noupdate = noupdate or update
                 record_list += record
-
+                
         elif rec[4] in ('menu_create',):
             for id in rec[5]:
                 id,update = self._get_id(cr, uid, rec[3], id)
@@ -461,7 +378,6 @@ class base_module_record(osv.osv):
         return record
 
     def _generate_function_yaml(self, cr, uid, args):
-        print "geberareeeeeeeeeeeeeeee",args
         db, uid, model, action, ids, context = args
         temp_context = context.copy()
         active_id = temp_context['active_id']
@@ -486,7 +402,7 @@ class base_module_record(osv.osv):
         attrs=str(attrs)+'})'
         function['attrs'] = attrs
         return function
-
+            
     def _generate_assert_xml(self, rec, doc):
         pass
 
@@ -528,7 +444,7 @@ class base_module_record(osv.osv):
         self.ids = {}
         if len(self.recording_data):
             yaml_file='''\n'''
-
+    
             for rec in self.recording_data:
                 if rec[1][3] == 'create':
                     self.mode="create"
@@ -543,45 +459,44 @@ class base_module_record(osv.osv):
                 else:
                     continue
                 if self.mode == "workflow":
-                    record= self._generate_object_yaml(cr, uid, rec[1],rec[0])
-                    yaml_file+="!comment Performing a workflow action %s on module %s"%(record['action'], record['model']) + '''\n'''
-                    object=yaml.load(unicode('''\n !workflow %s \n'''%record,'iso-8859-1'))
-                    attrs=attrs.replace("''", '"')
+                    record = self._generate_object_yaml(cr, uid, rec[1],rec[0])
+                    yaml_file += "!comment Performing a workflow action %s on module %s"%(record['action'], record['model']) + '''\n'''
+                    object = yaml.load(unicode('''\n !workflow %s \n'''%record,'iso-8859-1'))
                     yaml_file += str(object) + '''\n\n'''
                 elif self.mode == 'osv_memory_action':
-                    osv_action= self._generate_function_yaml(cr, uid, rec[1])
-                    yaml_file+="!comment Performing an osv_memory action %s on module %s"%(osv_action['action'], osv_action['model']) + '''\n'''
-                    osv_action=yaml.load(unicode('''\n !python %s \n'''%osv_action,'iso-8859-1'))
+                    osv_action = self._generate_function_yaml(cr, uid, rec[1])
+                    yaml_file += "!comment Performing an osv_memory action %s on module %s"%(osv_action['action'], osv_action['model']) + '''\n'''
+                    osv_action = yaml.load(unicode('''\n !python %s \n'''%osv_action,'iso-8859-1'))
                     yaml_file += str(osv_action) + '''\n'''
-                    attrs=yaml.dump(osv_action.attrs, default_flow_style=False)
-                    attrs=attrs.replace("''", '"')
-                    attrs=attrs.replace("'", '')
+                    attrs = yaml.dump(osv_action.attrs, default_flow_style=False)
+                    attrs = attrs.replace("''", '"')
+                    attrs = attrs.replace("'", '')
                     yaml_file += attrs + '''\n\n'''
                 else:
-                    record= self._generate_object_yaml(cr, uid, rec[1], rec[3])
+                    record = self._generate_object_yaml(cr, uid, rec[1], rec[3])
                     if self.mode == "create" or self.mode == "copy":
-                        yaml_file+="!comment Creating a %s record"%(record['model']) + '''\n'''
+                        yaml_file += "!comment Creating a %s record"%(record['model']) + '''\n'''
                     else:
-                        yaml_file+="!comment Modifying a %s record"%(record['model']) + '''\n'''
-                    object= yaml.load(unicode('''\n !record %s \n'''%record,'iso-8859-1'))
+                        yaml_file += "!comment Modifying a %s record"%(record['model']) + '''\n'''
+                    object = yaml.load(unicode('''\n !record %s \n'''%record,'iso-8859-1'))
                     yaml_file += str(object) + '''\n'''
-                    attrs=yaml.dump(object.attrs, default_flow_style=False)
+                    attrs = yaml.dump(object.attrs, default_flow_style=False)
                     yaml_file += attrs + '''\n\n'''
-
+                    
         yaml_result=''''''
         for line in yaml_file.split('\n'):
             line=line.replace("''","'")
             if (line.find('!record') == 0) or (line.find('!workflow') == 0) or (line.find('!python') == 0):
                 line = "- \n" + "  " + line
             elif line.find('!comment') == 0:
-                line=line.replace('!comment','- \n ')
+                line=line.replace('!comment','- \n ')   
             elif line.find('- -') != -1:
                 line=line.replace('- -','  -')
                 line = "    " + line
             else:
                 line = "    " + line
             yaml_result += line + '''\n'''
-
+            
         return yaml_result
 
 base_module_record()