[FIX] account_asset: change duration wizard. Fixed context using and historization...

bzr revid: qdp-launchpad@openerp.com-20110701080512-211krjr6tjbtw28x

diff --cc addons/account_asset/wizard/account_asset_change_duration.py
index 0000000,6a4da64..3497855
mode 000000,100755..100755
--- /dev/null
--- 2/addons/account_asset/wizard/account_asset_change_duration.py
+++ b/addons/account_asset/wizard/account_asset_change_duration.py
@@@ -1,0 -1,127 +1,130 @@@
+ # -*- encoding: utf-8 -*-
+ ##############################################################################
+ #    
+ #    OpenERP, Open Source Management Solution
+ #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
+ #
+ #    This program is free software: you can redistribute it and/or modify
+ #    it under the terms of the GNU Affero General Public License as
+ #    published by the Free Software Foundation, either version 3 of the
+ #    License, or (at your option) any later version.
+ #
+ #    This program is distributed in the hope that it will be useful,
+ #    but WITHOUT ANY WARRANTY; without even the implied warranty of
+ #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ #    GNU Affero General Public License for more details.
+ #
+ #    You should have received a copy of the GNU Affero General Public License
+ #    along with this program.  If not, see <http://www.gnu.org/licenses/>.     
+ #
+ ##############################################################################
+ import time
+ from lxml import etree
+ 
+ from osv import osv, fields
+ 
+ class asset_modify(osv.osv_memory):
+     _name = 'asset.modify'
+     _description = 'Modify Asset'
+ 
+     _columns = {
+         'name': fields.char('Reason', size=64, required=True),
 -        'method_delay': fields.integer('Number of Depreciation'),
++        'method_delay': fields.integer('Number of Depreciations'),
+         'method_period': fields.integer('Period Length'),
+         'method_end': fields.date('Ending date'),
+         'note': fields.text('Notes'),
+     }
+ 
+     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
+         """ Returns views and fields for current model.
+         @param cr: A database cursor
+         @param user: ID of the user currently logged in
+         @param view_id: list of fields, which required to read signatures
+         @param view_type: defines a view type. it can be one of (form, tree, graph, calender, gantt, search, mdx)
+         @param context: context arguments, like lang, time zone
+         @param toolbar: contains a list of reports, wizards, and links related to current model
+ 
+         @return: Returns a dictionary that contains definition for fields, views, and toolbars
+         """
 -        asset_obj = self.pool.get('account.asset.asset')
 -        result = super(asset_modify, self).fields_view_get(cr, uid, view_id, view_type, context=context, toolbar=toolbar, submenu=submenu)
+         if not context:
+             context = {}
++        asset_obj = self.pool.get('account.asset.asset')
++        result = super(asset_modify, self).fields_view_get(cr, uid, view_id, view_type, context=context, toolbar=toolbar, submenu=submenu)
+         asset_id = context.get('active_id', False)
+         active_model = context.get('active_model', '')
+         if active_model == 'account.asset.asset' and asset_id:
+             asset = asset_obj.browse(cr, uid, asset_id, context=context)
+             doc = etree.XML(result['arch'])
+             if asset.method_time == 'delay':
+                 node = doc.xpath("//field[@name='method_end']")[0]
+                 node.set('invisible', '1')
+             elif asset.method_time == 'end':
+                 node = doc.xpath("//field[@name='method_delay']")[0]
+                 node.set('invisible', '1')
+             result['arch'] = etree.tostring(doc)
+         return result
+ 
+     def default_get(self, cr, uid, fields, context=None):
+         """ To get default values for the object.
+         @param self: The object pointer.
+         @param cr: A database cursor
+         @param uid: ID of the user currently logged in
+         @param fields: List of fields for which we want default values 
+         @param context: A standard dictionary 
+         @return: A dictionary which of fields with values. 
+         """ 
++        if not context:
++            context = {}
+         asset_obj = self.pool.get('account.asset.asset')
+         res = super(asset_modify, self).default_get(cr, uid, fields, context=context)
+         asset_id = context.get('active_id', False)
+         asset = asset_obj.browse(cr, uid, asset_id, context=context)
+         if 'name' in fields:
+             res.update({'name': asset.name})
+         if 'method_delay' in fields and asset.method_time == 'delay':
+             res.update({'method_delay': asset.method_delay})
+         if 'method_period' in fields:
+             res.update({'method_period': asset.method_period})
+         if 'method_end' in fields and asset.method_time == 'end':
+             res.update({'method_end': asset.method_end})
+         return res
+     
+     def modify(self, cr, uid, ids, context=None):
+         """ Modifies the duration of asset for calculating depreciation
+         and maintains the history of old values.
+         @param self: The object pointer.
+         @param cr: A database cursor
+         @param uid: ID of the user currently logged in
+         @param ids: List of Ids 
+         @param context: A standard dictionary 
+         @return: Close the wizard. 
+         """ 
++        if not context:
++            context = {}
+         asset_obj = self.pool.get('account.asset.asset')
+         history_obj = self.pool.get('account.asset.history')
+         asset_id = context.get('active_id', False)
+         asset = asset_obj.browse(cr, uid, asset_id, context=context)
+         data = self.browse(cr, uid, ids[0], context=context)
+         history_vals = {
+             'asset_id': asset_id,
 -            'name': asset.name,
++            'name': data.name,
+             'method_delay': asset.method_delay,
+             'method_period': asset.method_period,
+             'method_end': asset.method_end,
+             'user_id': uid,
+             'date': time.strftime('%Y-%m-%d'),
+             'note': data.note,
+         }
+         history_obj.create(cr, uid, history_vals, context=context)
+         asset_vals = {
 -            'name': data.name,
+             'method_delay': data.method_delay,
+             'method_period': data.method_period,
+             'method_end': data.method_end,
+         }
+         asset_obj.write(cr, uid, [asset_id], asset_vals, context=context)
+         return {'type': 'ir.actions.act_window_close'}
+ 
+ asset_modify()
+ 
+ # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: