addons/stock/wizard/stock_fill_inventory.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
  25 class stock_fill_inventory(osv.osv_memory):
  26     _name = "stock.fill.inventory"
  27     _description = "Import Inventory"
  28     _columns = {
  29         'location_id': fields.many2one('stock.location', 'Location', required=True),
  30         'recursive': fields.boolean("Include children",help="If checked, products contained in child locations of selected location will be included as well."),
  31         'set_stock_zero': fields.boolean("Set to zero",help="If checked, all product quantities will be set to zero to help ensure a real physical inventory is done"),
  32     }
  33     def view_init(self, cr, uid, fields_list, context=None):
  34         """
  35          Creates view dynamically and adding fields at runtime.
  36          @param self: The object pointer.
  37          @param cr: A database cursor
  38          @param uid: ID of the user currently logged in
  39          @param context: A standard dictionary
  40          @return: New arch of view with new columns.
  41         """
  42         if context is None:
  43             context = {}
  44         super(stock_fill_inventory, self).view_init(cr, uid, fields_list, context=context)
  45
  46         if len(context.get('active_ids',[])) > 1:
  47             raise osv.except_osv(_('Error!'), _('You cannot perform this operation on more than one Stock Inventories.'))
  48
  49         if context.get('active_id', False):
  50             stock = self.pool.get('stock.inventory').browse(cr, uid, context.get('active_id', False))
  51
  52             if stock.state == 'done':
  53                 raise osv.except_osv(_('Warning!'), _('Stock Inventory is already Validated.'))
  54         return True
  55
  56     def fill_inventory(self, cr, uid, ids, context=None):
  57         """ To Import stock inventory according to products available in the selected locations.
  58         @param self: The object pointer.
  59         @param cr: A database cursor
  60         @param uid: ID of the user currently logged in
  61         @param ids: the ID or list of IDs if we want more than one
  62         @param context: A standard dictionary
  63         @return:
  64         """
  65         if context is None:
  66             context = {}
  67
  68         inventory_line_obj = self.pool.get('stock.inventory.line')
  69         location_obj = self.pool.get('stock.location')
  70         product_obj = self.pool.get('product.product')
  71         stock_location_obj = self.pool.get('stock.location')
  72         move_obj = self.pool.get('stock.move')
  73         uom_obj = self.pool.get('product.uom')
  74         if ids and len(ids):
  75             ids = ids[0]
  76         else:
  77              return {'type': 'ir.actions.act_window_close'}
  78         fill_inventory = self.browse(cr, uid, ids, context=context)
  79         res = {}
  80         res_location = {}
  81
  82         if fill_inventory.recursive:
  83             location_ids = location_obj.search(cr, uid, [('location_id',
  84                              'child_of', [fill_inventory.location_id.id])], order="id",
  85                              context=context)
  86         else:
  87             location_ids = [fill_inventory.location_id.id]
  88
  89         res = {}
  90         flag = False
  91
  92         for location in location_ids:
  93             datas = {}
  94             res[location] = {}
  95             move_ids = move_obj.search(cr, uid, ['|',('location_dest_id','=',location),('location_id','=',location),('state','=','done')], context=context)
  96
  97             for move in move_obj.browse(cr, uid, move_ids, context=context):
  98                 lot_id = move.prodlot_id.id
  99                 prod_id = move.product_id.id
 100                 if move.location_dest_id.id == location:
 101                     qty = uom_obj._compute_qty(cr, uid, move.product_uom.id,move.product_qty, move.product_id.uom_id.id)
 102                 else:
 103                     qty = -uom_obj._compute_qty(cr, uid, move.product_uom.id,move.product_qty, move.product_id.uom_id.id)
 104
 105
 106                 if datas.get((prod_id, lot_id)):
 107                     qty += datas[(prod_id, lot_id)]['product_qty']
 108
 109                 datas[(prod_id, lot_id)] = {'product_id': prod_id, 'location_id': location, 'product_qty': qty, 'product_uom': move.product_id.uom_id.id, 'prod_lot_id': lot_id}
 110
 111             if datas:
 112                 flag = True
 113                 res[location] = datas
 114
 115         if not flag:
 116             raise osv.except_osv(_('Warning !'), _('No product in this location.'))
 117
 118         for stock_move in res.values():
 119             for stock_move_details in stock_move.values():
 120                 stock_move_details.update({'inventory_id': context['active_ids'][0]})
 121                 domain = []
 122
 123                 if fill_inventory.set_stock_zero:
 124                     stock_move_details.update({'product_qty': 0})
 125
 126                 for field, value in stock_move_details.items():
 127                     domain.append((field, '=', value))
 128
 129                 line_ids = inventory_line_obj.search(cr, uid, domain, context=context)
 130
 131                 if not line_ids:
 132                     inventory_line_obj.create(cr, uid, stock_move_details, context=context)
 133
 134         return {'type': 'ir.actions.act_window_close'}
 135
 136 stock_fill_inventory()
 137
 138 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: