[IMP] stock:inventory_merge wizard has been converted into osv memory
[odoo/odoo.git] / addons / stock / wizard / stock_inventory_merge.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 service import web_services
24 from tools.misc import UpdateableStr, UpdateableDict
25 from tools.translate import _
26 import netsvc
27 import pooler
28 import time
29 import wizard
30
31 class stock_inventory_merge(osv.osv_memory):
32     _name = "stock.inventory.merge"
33     _description = "Merge Inventory"
34     _columns = {
35            
36             }
37
38     def do_merge(self, cr, uid, ids, context):
39         """ 
40              To merge selected Inventories.
41             
42              @param self: The object pointer.
43              @param cr: A database cursor
44              @param uid: ID of the user currently logged in
45              @param ids: List of IDs selected 
46              @param context: A standard dictionary 
47              
48              @return: 
49         
50         """ 
51         invent_obj = self.pool.get('stock.inventory')
52         invent_line_obj = self.pool.get('stock.inventory.line')
53
54         invent_lines = {}
55
56         if len(context['active_ids']) < 2:
57             raise osv.except_osv(_('Warning'),
58             _('Please select at least two inventories.'))
59
60         for inventory in invent_obj.browse(cr, uid, context['active_ids'], context=context):
61             if inventory.state == "done":
62                 raise osv.except_osv(_('Warning'),
63                 _('Merging is only allowed on draft inventories.'))
64
65             for line in inventory.inventory_line_id:
66                 key = (line.location_id.id, line.product_id.id, line.product_uom.id)
67                 if key in invent_lines:
68                     invent_lines[key] += line.product_qty
69                 else:
70                     invent_lines[key] = line.product_qty
71
72
73         new_invent = invent_obj.create(cr, uid, {
74             'name': 'Merged inventory'
75             }, context=context)
76
77         for key, quantity in invent_lines.items():
78             invent_line_obj.create(cr, uid, {
79                 'inventory_id': new_invent,
80                 'location_id': key[0],
81                 'product_id': key[1],
82                 'product_uom': key[2],
83                 'product_qty': quantity,
84                 })
85
86         return {}
87
88 stock_inventory_merge()
89
90 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
91