[IMP] account_analytic_analysis, account_analytic_default, account_analytic_plans: code cleaning

bzr revid: psi@tinyerp.co.in-20101011112550-ygkqmzmuhu1j7dm3

1 files changed, 7 insertions, 5 deletions

diff --git a/addons/account_analytic_plans/wizard/analytic_plan_create_model.py b/addons/account_analytic_plans/wizard/analytic_plan_create_model.py
index dc108d1700a..777421798cd 100644
--- a/addons/account_analytic_plans/wizard/analytic_plan_create_model.py
+++ b/addons/account_analytic_plans/wizard/analytic_plan_create_model.py
@@ -18,24 +18,26 @@
 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 ##############################################################################
+
 from osv import osv
 from tools.translate import _
 
 class analytic_plan_create_model(osv.osv_memory):
-    _name = 'analytic.plan.create.model'
-    _description = 'analytic.plan.create.model'
+    _name = "analytic.plan.create.model"
+    _description = "analytic.plan.create.model"
 
     def activate(self, cr, uid, ids, context=None):
         plan_obj = self.pool.get('account.analytic.plan.instance')
         mod_obj = self.pool.get('ir.model.data')
+        anlytic_plan_obj = self.pool.get('account.analytic.plan')
         if 'active_id' in context and context['active_id']:
             plan = plan_obj.browse(cr, uid, context['active_id'], context=context)
             if (not plan.name) or (not plan.code):
                 raise osv.except_osv(_('Error'), _('Please put a name and a code before saving the model !'))
-            pids  =  self.pool.get('account.analytic.plan').search(cr, uid, [], context=context)
-            if (not pids):
+            pids = anlytic_plan_obj.search(cr, uid, [], context=context)
+            if not pids:
                 raise osv.except_osv(_('Error'), _('No analytic plan defined !'))
-            plan_obj.write(cr, uid, [context['active_id']], {'plan_id':pids[0]})
+            plan_obj.write(cr, uid, [context['active_id']], {'plan_id':pids[0]}, context=context)
 
             model_data_ids = mod_obj.search(cr, uid, [('model', '=', 'ir.ui.view'),('name', '=', 'view_analytic_plan_create_model')], context=context)
             resource_id = mod_obj.read(cr, uid, model_data_ids, fields=['res_id'], context=context)[0]['res_id']