addons/mrp/wizard/change_production_qty.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 from osv import fields, osv
  23 from tools.translate import _
  24 import decimal_precision as dp
  25
  26 class change_production_qty(osv.osv_memory):
  27     _name = 'change.production.qty'
  28     _description = 'Change Quantity of Products'
  29
  30     _columns = {
  31         'product_qty': fields.float('Product Qty', digits_compute=dp.get_precision('Product UoM'), required=True),
  32     }
  33
  34     def default_get(self, cr, uid, fields, context=None):
  35         """ To get default values for the object.
  36         @param self: The object pointer.
  37         @param cr: A database cursor
  38         @param uid: ID of the user currently logged in
  39         @param fields: List of fields for which we want default values
  40         @param context: A standard dictionary
  41         @return: A dictionary which of fields with values.
  42         """
  43         if context is None:
  44             context = {}
  45         res = super(change_production_qty, self).default_get(cr, uid, fields, context=context)
  46         prod_obj = self.pool.get('mrp.production')
  47         prod = prod_obj.browse(cr, uid, context.get('active_id'), context=context)
  48         if 'product_qty' in fields:
  49             res.update({'product_qty': prod.product_qty})
  50         return res
  51
  52     def change_prod_qty(self, cr, uid, ids, context=None):
  53         """
  54         Changes the Quantity of Product.
  55         @param self: The object pointer.
  56         @param cr: A database cursor
  57         @param uid: ID of the user currently logged in
  58         @param ids: List of IDs selected
  59         @param context: A standard dictionary
  60         @return:
  61         """
  62         record_id = context and context.get('active_id',False)
  63         assert record_id, _('Active Id is not found')
  64         prod_obj = self.pool.get('mrp.production')
  65         bom_obj = self.pool.get('mrp.bom')
  66         for wiz_qty in self.browse(cr, uid, ids, context=context):
  67             prod = prod_obj.browse(cr, uid, record_id, context=context)
  68             prod_obj.write(cr, uid,prod.id, {'product_qty': wiz_qty.product_qty})
  69             prod_obj.action_compute(cr, uid, [prod.id])
  70
  71             move_lines_obj = self.pool.get('stock.move')
  72             for move in prod.move_lines:
  73                 bom_point = prod.bom_id
  74                 bom_id = prod.bom_id.id
  75                 if not bom_point:
  76                     bom_id = bom_obj._bom_find(cr, uid, prod.product_id.id, prod.product_uom.id)
  77                     if not bom_id:
  78                         raise osv.except_osv(_('Error'), _("Couldn't find bill of material for product"))
  79                     prod_obj.write(cr, uid, [prod.id], {'bom_id': bom_id})
  80                     bom_point = bom_obj.browse(cr, uid, [bom_id])[0]
  81
  82                 if not bom_id:
  83                     raise osv.except_osv(_('Error'), _("Couldn't find bill of material for product"))
  84
  85                 factor = prod.product_qty * prod.product_uom.factor / bom_point.product_uom.factor
  86                 res = bom_obj._bom_explode(cr, uid, bom_point, factor / bom_point.product_qty, [])
  87                 for r in res[0]:
  88                     if r['product_id'] == move.product_id.id:
  89                         move_lines_obj.write(cr, uid, [move.id], {'product_qty' :  r['product_qty']})
  90             for m in prod.move_created_ids:
  91                 move_lines_obj.write(cr, uid, [m.id], {'product_qty': wiz_qty.product_qty})
  92
  93         return {}
  94
  95 change_production_qty()
  96
  97 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: