121 lines
5.5 KiB
Python
121 lines
5.5 KiB
Python
# -*- coding: 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/>.
|
|
#
|
|
##############################################################################
|
|
|
|
from osv import fields, osv
|
|
from tools.translate import _
|
|
|
|
class stock_fill_inventory(osv.osv_memory):
|
|
_name = "stock.fill.inventory"
|
|
_description = "Import Inventory"
|
|
_columns = {
|
|
'location_id': fields.many2one('stock.location', 'Location', required=True),
|
|
'recursive': fields.boolean("Include children",help="If checked, products contained in child locations of selected location will be included as well."),
|
|
'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"),
|
|
}
|
|
def view_init(self, cr, uid, fields_list, context=None):
|
|
"""
|
|
Creates view dynamically and adding fields at runtime.
|
|
@param self: The object pointer.
|
|
@param cr: A database cursor
|
|
@param uid: ID of the user currently logged in
|
|
@param context: A standard dictionary
|
|
@return: New arch of view with new columns.
|
|
"""
|
|
if context==None:
|
|
context={}
|
|
res = super(stock_fill_inventory, self).view_init(cr, uid, fields_list, context=context)
|
|
if context.get('active_id', False):
|
|
stock = self.pool.get('stock.inventory').browse(cr, uid, context.get('active_id', False))
|
|
if stock.state=='done':
|
|
raise osv.except_osv('Error!','Stock Inventory is done')
|
|
True
|
|
|
|
def fill_inventory(self, cr, uid, ids, context=None):
|
|
""" To Import stock inventory according to products available in the selected locations.
|
|
@param self: The object pointer.
|
|
@param cr: A database cursor
|
|
@param uid: ID of the user currently logged in
|
|
@param ids: the ID or list of IDs if we want more than one
|
|
@param context: A standard dictionary
|
|
@return:
|
|
"""
|
|
if context is None:
|
|
context = {}
|
|
inventory_line_obj = self.pool.get('stock.inventory.line')
|
|
location_obj = self.pool.get('stock.location')
|
|
product_obj = self.pool.get('product.product')
|
|
stock_location_obj = self.pool.get('stock.location')
|
|
if ids and len(ids):
|
|
ids = ids[0]
|
|
else:
|
|
return {}
|
|
fill_inventory = self.browse(cr, uid, ids, context=context)
|
|
res = {}
|
|
res_location = {}
|
|
if fill_inventory.recursive :
|
|
location_ids = location_obj.search(cr, uid, [('location_id',
|
|
'child_of', fill_inventory.location_id.id)])
|
|
for location in location_ids :
|
|
res = location_obj._product_get(cr, uid, location)
|
|
res_location[location] = res
|
|
else:
|
|
context.update({'compute_child': False})
|
|
res = location_obj._product_get(cr, uid,
|
|
fill_inventory.location_id.id, context=context)
|
|
res_location[fill_inventory.location_id.id] = res
|
|
|
|
product_ids = []
|
|
for location in res_location.keys():
|
|
res = res_location[location]
|
|
for product_id in res.keys():
|
|
prod = product_obj.browse(cr, uid, product_id, context=context)
|
|
uom = prod.uom_id.id
|
|
context.update(uom=uom, compute_child=False)
|
|
amount = stock_location_obj._product_get(cr, uid,
|
|
location, [product_id], context=context)[product_id]
|
|
if(amount):
|
|
if fill_inventory.set_stock_zero:
|
|
amount = 0
|
|
line_ids=inventory_line_obj.search(cr, uid,
|
|
[('inventory_id', '=', context['active_ids']),
|
|
('location_id', '=', location),
|
|
('product_id', '=', product_id),
|
|
('product_uom', '=', uom),
|
|
('product_qty', '=', amount)])
|
|
if not len(line_ids):
|
|
inventory_line = {
|
|
'inventory_id': context['active_ids'][0],
|
|
'location_id': location,
|
|
'product_id': product_id,
|
|
'product_uom': uom,
|
|
'product_qty': amount
|
|
}
|
|
inventory_line_obj.create(cr, uid, inventory_line)
|
|
product_ids.append(product_id)
|
|
|
|
if(len(product_ids) == 0):
|
|
raise osv.except_osv(_('Message !'), _('No product in this location.'))
|
|
return {}
|
|
|
|
stock_fill_inventory()
|
|
|
|
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|