[MERGE] merged the branch with the refactoring of picking partial processing and...
[odoo/odoo.git] / addons / stock / stock.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 datetime import date, datetime
23 from dateutil import relativedelta
24
25 import time
26
27 from openerp.osv import fields, osv
28 from openerp.tools.translate import _
29 from openerp import tools
30 from openerp.tools import DEFAULT_SERVER_DATETIME_FORMAT
31 from openerp import SUPERUSER_ID
32 import openerp.addons.decimal_precision as dp
33 import logging
34 _logger = logging.getLogger(__name__)
35
36
37 #----------------------------------------------------------
38 # Incoterms
39 #----------------------------------------------------------
40 class stock_incoterms(osv.osv):
41     _name = "stock.incoterms"
42     _description = "Incoterms"
43     _columns = {
44         'name': fields.char('Name', size=64, required=True, help="Incoterms are series of sales terms. They are used to divide transaction costs and responsibilities between buyer and seller and reflect state-of-the-art transportation practices."),
45         'code': fields.char('Code', size=3, required=True, help="Incoterm Standard Code"),
46         'active': fields.boolean('Active', help="By unchecking the active field, you may hide an INCOTERM you will not use."),
47     }
48     _defaults = {
49         'active': True,
50     }
51
52 #----------------------------------------------------------
53 # Stock Location
54 #----------------------------------------------------------
55
56 class stock_location(osv.osv):
57     _name = "stock.location"
58     _description = "Inventory Locations"
59     _parent_name = "location_id"
60     _parent_store = True
61     _parent_order = 'name'
62     _order = 'parent_left'
63     _rec_name = 'complete_name'
64
65     def _complete_name(self, cr, uid, ids, name, args, context=None):
66         """ Forms complete name of location from parent location to child location.
67         @return: Dictionary of values
68         """
69         res = {}
70         for m in self.browse(cr, uid, ids, context=context):
71             res[m.id] = m.name
72             parent = m.location_id
73             while parent:
74                 res[m.id] = parent.name + ' / ' + res[m.id]
75                 parent = parent.location_id
76         return res
77
78     def _get_sublocations(self, cr, uid, ids, context=None):
79         """ return all sublocations of the given stock locations (included) """
80         if context is None:
81             context = {}
82         context_with_inactive = context.copy()
83         context_with_inactive['active_test'] = False
84         return self.search(cr, uid, [('id', 'child_of', ids)], context=context_with_inactive)
85
86     _columns = {
87         'name': fields.char('Location Name', size=64, required=True, translate=True),
88         'active': fields.boolean('Active', help="By unchecking the active field, you may hide a location without deleting it."),
89         'usage': fields.selection([('supplier', 'Supplier Location'), ('view', 'View'), ('internal', 'Internal Location'), ('customer', 'Customer Location'), ('inventory', 'Inventory'), ('procurement', 'Procurement'), ('production', 'Production'), ('transit', 'Transit Location for Inter-Companies Transfers')], 'Location Type', required=True,
90                  help="""* Supplier Location: Virtual location representing the source location for products coming from your suppliers
91                        \n* View: Virtual location used to create a hierarchical structures for your warehouse, aggregating its child locations ; can't directly contain products
92                        \n* Internal Location: Physical locations inside your own warehouses,
93                        \n* Customer Location: Virtual location representing the destination location for products sent to your customers
94                        \n* Inventory: Virtual location serving as counterpart for inventory operations used to correct stock levels (Physical inventories)
95                        \n* Procurement: Virtual location serving as temporary counterpart for procurement operations when the source (supplier or production) is not known yet. This location should be empty when the procurement scheduler has finished running.
96                        \n* Production: Virtual counterpart location for production operations: this location consumes the raw material and produces finished products
97                       """, select=True),
98
99         'complete_name': fields.function(_complete_name, type='char', string="Location Name",
100                             store={'stock.location': (_get_sublocations, ['name', 'location_id', 'active'], 10)}),
101         'location_id': fields.many2one('stock.location', 'Parent Location', select=True, ondelete='cascade'),
102         'child_ids': fields.one2many('stock.location', 'location_id', 'Contains'),
103
104         'partner_id': fields.many2one('res.partner', 'Owner', help="Owner of the location if not internal"),
105
106         'comment': fields.text('Additional Information'),
107         'posx': fields.integer('Corridor (X)', help="Optional localization details, for information purpose only"),
108         'posy': fields.integer('Shelves (Y)', help="Optional localization details, for information purpose only"),
109         'posz': fields.integer('Height (Z)', help="Optional localization details, for information purpose only"),
110
111         'parent_left': fields.integer('Left Parent', select=1),
112         'parent_right': fields.integer('Right Parent', select=1),
113
114         'company_id': fields.many2one('res.company', 'Company', select=1, help='Let this field empty if this location is shared between all companies'),
115         'scrap_location': fields.boolean('Scrap Location', help='Check this box to allow using this location to put scrapped/damaged goods.'),
116         'removal_strategy_ids': fields.one2many('product.removal', 'location_id', 'Removal Strategies'),
117         'putaway_strategy_ids': fields.one2many('product.putaway', 'location_id', 'Put Away Strategies'),
118     }
119     _defaults = {
120         'active': True,
121         'usage': 'internal',
122         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.location', context=c),
123         'posx': 0,
124         'posy': 0,
125         'posz': 0,
126         'scrap_location': False,
127     }
128
129     def get_putaway_strategy(self, cr, uid, location, product, context=None):
130         pa = self.pool.get('product.putaway')
131         categ = product.categ_id
132         categs = [categ.id, False]
133         while categ.parent_id:
134             categ = categ.parent_id
135             categs.append(categ.id)
136
137         result = pa.search(cr, uid, [('location_id', '=', location.id), ('product_categ_id', 'in', categs)], context=context)
138         if result:
139             return pa.browse(cr, uid, result[0], context=context)
140
141     def get_removal_strategy(self, cr, uid, location, product, context=None):
142         pr = self.pool.get('product.removal')
143         categ = product.categ_id
144         categs = [categ.id, False]
145         while categ.parent_id:
146             categ = categ.parent_id
147             categs.append(categ.id)
148
149         result = pr.search(cr, uid, [('location_id', '=', location.id), ('product_categ_id', 'in', categs)], context=context)
150         if result:
151             return pr.browse(cr, uid, result[0], context=context).method
152
153
154 #----------------------------------------------------------
155 # Routes
156 #----------------------------------------------------------
157
158 class stock_location_route(osv.osv):
159     _name = 'stock.location.route'
160     _description = "Inventory Routes"
161     _order = 'sequence'
162
163     _columns = {
164         'name': fields.char('Route Name', required=True),
165         'sequence': fields.integer('Sequence'),
166         'pull_ids': fields.one2many('procurement.rule', 'route_id', 'Pull Rules'),
167         'active': fields.boolean('Active', help="If the active field is set to False, it will allow you to hide the route without removing it."),
168         'push_ids': fields.one2many('stock.location.path', 'route_id', 'Push Rules'),
169         'product_selectable': fields.boolean('Applicable on Product'),
170         'product_categ_selectable': fields.boolean('Applicable on Product Category'),
171         'warehouse_selectable': fields.boolean('Applicable on Warehouse'),
172         'supplied_wh_id': fields.many2one('stock.warehouse', 'Supplied Warehouse'),
173         'supplier_wh_id': fields.many2one('stock.warehouse', 'Supplier Warehouse'),
174     }
175
176     _defaults = {
177         'sequence': lambda self, cr, uid, ctx: 0,
178         'active': True,
179         'product_selectable': True,
180     }
181
182
183 #----------------------------------------------------------
184 # Quants
185 #----------------------------------------------------------
186
187 class stock_quant(osv.osv):
188     """
189     Quants are the smallest unit of stock physical instances
190     """
191     _name = "stock.quant"
192     _description = "Quants"
193
194     def _get_quant_name(self, cr, uid, ids, name, args, context=None):
195         """ Forms complete name of location from parent location to child location.
196         @return: Dictionary of values
197         """
198         res = {}
199         for q in self.browse(cr, uid, ids, context=context):
200
201             res[q.id] = q.product_id.code or ''
202             if q.lot_id:
203                 res[q.id] = q.lot_id.name
204             res[q.id] += ': ' + str(q.qty) + q.product_id.uom_id.name
205         return res
206
207     _columns = {
208         'name': fields.function(_get_quant_name, type='char', string='Identifier'),
209         'product_id': fields.many2one('product.product', 'Product', required=True),
210         'location_id': fields.many2one('stock.location', 'Location', required=True),
211         'qty': fields.float('Quantity', required=True, help="Quantity of products in this quant, in the default unit of measure of the product"),
212         'package_id': fields.many2one('stock.quant.package', string='Package', help="The package containing this quant"),
213         'packaging_type_id': fields.related('package_id', 'packaging_id', type='many2one', relation='product.packaging', string='Type of packaging', store=True),
214         'reservation_id': fields.many2one('stock.move', 'Reserved for Move', help="The move the quant is reserved for"),
215         'link_move_operation_id': fields.many2one('stock.move.operation.link', 'Reserved for Link between Move and Pack Operation', help="Technical field decpicting for with tuple (move, operation) this quant is reserved for"),
216         'lot_id': fields.many2one('stock.production.lot', 'Lot'),
217         'cost': fields.float('Unit Cost'),
218         'owner_id': fields.many2one('res.partner', 'Owner', help="This is the owner of the quant"),
219
220         'create_date': fields.datetime('Creation Date'),
221         'in_date': fields.datetime('Incoming Date'),
222
223         'history_ids': fields.many2many('stock.move', 'stock_quant_move_rel', 'quant_id', 'move_id', 'Moves', help='Moves that operate(d) on this quant'),
224         'company_id': fields.many2one('res.company', 'Company', help="The company to which the quants belong", required=True),
225
226         # Used for negative quants to reconcile after compensated by a new positive one
227         'propagated_from_id': fields.many2one('stock.quant', 'Linked Quant', help='The negative quant this is coming from'),
228         'negative_dest_location_id': fields.many2one('stock.location', 'Destination Location', help='Technical field used to record the destination location of a move that created a negative quant'),
229     }
230
231     _defaults = {
232         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.quant', context=c),
233     }
234
235     def action_view_quant_history(self, cr, uid, ids, context=None):
236         '''
237         This function returns an action that display the history of the quant, which
238         mean all the stock moves that lead to this quant creation with this quant quantity.
239         '''
240         mod_obj = self.pool.get('ir.model.data')
241         act_obj = self.pool.get('ir.actions.act_window')
242
243         result = mod_obj.get_object_reference(cr, uid, 'stock', 'action_move_form2')
244         id = result and result[1] or False
245         result = act_obj.read(cr, uid, [id], context={})[0]
246
247         move_ids = []
248         for quant in self.browse(cr, uid, ids, context=context):
249             move_ids += [move.id for move in quant.history_ids]
250
251         result['domain'] = "[('id','in',[" + ','.join(map(str, move_ids)) + "])]"
252         return result
253
254     def quants_reserve(self, cr, uid, quants, move, link=False, context=None):
255         '''This function reserves quants for the given move (and optionally given link). If the total of quantity reserved is enough, the move's state
256         is also set to 'assigned'
257
258         :param quants: list of tuple(quant browse record or None, qty to reserve). If None is given as first tuple element, the item will be ignored
259         :param move: browse record
260         :param link: browse record (stock.move.operation.link)
261         '''
262         toreserve = []
263         #split quants if needed
264         for quant, qty in quants:
265             if not quant:
266                 continue
267             self._quant_split(cr, uid, quant, qty, context=context)
268             toreserve.append(quant.id)
269         #reserve quants
270         if toreserve:
271             self.write(cr, SUPERUSER_ID, toreserve, {'reservation_id': move.id, 'link_move_operation_id': link and link.id or False}, context=context)
272         #check if move'state needs to be set as 'assigned'
273         move.refresh()
274         if sum([q.qty for q in move.reserved_quant_ids]) == move.product_qty and move.state == 'confirmed':
275             self.pool.get('stock.move').write(cr, uid, [move.id], {'state': 'assigned'}, context=context)
276
277     def quants_move(self, cr, uid, quants, move, lot_id=False, owner_id=False, src_package_id=False, dest_package_id=False, context=None):
278         """Moves all given stock.quant in the destination location of the given move.
279
280         :param quants: list of tuple(browse record(stock.quant) or None, quantity to move)
281         :param move: browse record (stock.move)
282         :param lot_id: ID of the lot that mus be set on the quants to move
283         :param owner_id: ID of the partner that must own the quants to move
284         :param src_package_id: ID of the package that contains the quants to move
285         :param dest_package_id: ID of the package that must be set on the moved quant
286         """
287         for quant, qty in quants:
288             if not quant:
289                 #If quant is None, we will create a quant to move (and potentially a negative counterpart too)
290                 quant = self._quant_create(cr, uid, qty, move, lot_id=lot_id, owner_id=owner_id, src_package_id=src_package_id, dest_package_id=dest_package_id, context=context)
291             self.move_single_quant_tuple(cr, uid, quant, qty, move, context=context)
292
293     def check_preferred_location(self, cr, uid, move, qty, context=None):
294         '''Checks the preferred location on the move, if any returned by a putaway strategy, and returns a list of
295         tuple(location, qty) where the quant have to be moved
296
297         :param move: browse record (stock.move)
298         :param qty: float
299         :returns: list of tuple build as [(browe record (stock.move), float)]
300         '''
301         if move.putaway_ids:
302             res = []
303             for record in move.putaway_ids:
304                 res.append((record.location_id, record.quantity))
305             return res
306         return [(move.location_dest_id, qty)]
307
308     def move_single_quant(self, cr, uid, quant, location_to, qty, move, context=None):
309         '''Moves the given 'quant' in 'location_to' for the given 'qty', and logs the stock.move that triggered this move in the quant history.
310         If needed, the quant may be split if it's not totally moved.
311
312         :param quant: browse record (stock.quant)
313         :param location_to: browse record (stock.location)
314         :param qty: float
315         :param move: browse record (stock.move)
316         '''
317         new_quant = self._quant_split(cr, uid, quant, qty, context=context)
318         vals = {
319             'location_id': location_to.id,
320             'history_ids': [(4, move.id)],
321         }
322         #if the quant we are moving had been split and was inside a package, it means we unpacked it
323         if new_quant and new_quant.package_id:
324             vals['package_id'] = False
325         self.write(cr, SUPERUSER_ID, [quant.id], vals, context=context)
326         quant.refresh()
327         return new_quant
328
329     def move_single_quant_tuple(self, cr, uid, quant, qty, move, context=None):
330         '''Effectively process the move of a tuple (quant record, qty to move). This may result in several quants moved
331         if the preferred locations on the move say so but by default it will only move the quant record given as argument
332         :param quant: browse record (stock.quant)
333         :param qty: float
334         :param move: browse record (stock.move)
335         '''
336         for location_to, qty in self.check_preferred_location(cr, uid, move, qty, context=context):
337             if not quant:
338                 break
339             new_quant = self.move_single_quant(cr, uid, quant, location_to, qty, move, context=context)
340             self._quant_reconcile_negative(cr, uid, quant, context=context)
341             quant = new_quant
342
343     def quants_get_prefered_domain(self, cr, uid, location, product, qty, domain=None, prefered_domain=False, fallback_domain=False, restrict_lot_id=False, restrict_partner_id=False, context=None):
344         ''' This function tries to find quants in the given location for the given domain, by trying to first limit
345             the choice on the quants that match the prefered_domain as well. But if the qty requested is not reached
346             it tries to find the remaining quantity by using the fallback_domain.
347         '''
348         if prefered_domain and fallback_domain:
349             if domain is None:
350                 domain = []
351             quants = self.quants_get(cr, uid, location, product, qty, domain=domain + prefered_domain, restrict_lot_id=restrict_lot_id, restrict_partner_id=restrict_partner_id, context=context)
352             res_qty = qty
353             quant_ids = []
354             for quant in quants:
355                 if quant[0]:
356                     quant_ids.append(quant[0].id)
357                     res_qty -= quant[1]
358             if res_qty > 0:
359                 #try to replace the last tuple (None, res_qty) with something that wasn't chosen at first because of the prefered order
360                 quants.pop()
361                 #make sure the quants aren't found twice (if the prefered_domain and the fallback_domain aren't orthogonal
362                 domain += [('id', 'not in', quant_ids)]
363                 unprefered_quants = self.quants_get(cr, uid, location, product, res_qty, domain=domain + fallback_domain, restrict_lot_id=restrict_lot_id, restrict_partner_id=restrict_partner_id, context=context)
364                 for quant in unprefered_quants:
365                     quants.append(quant)
366             return quants
367         return self.quants_get(cr, uid, location, product, qty, domain=domain, restrict_lot_id=restrict_lot_id, restrict_partner_id=restrict_partner_id, context=context)
368
369     def quants_get(self, cr, uid, location, product, qty, domain=None, restrict_lot_id=False, restrict_partner_id=False, context=None):
370         """
371         Use the removal strategies of product to search for the correct quants
372         If you inherit, put the super at the end of your method.
373
374         :location: browse record of the parent location where the quants have to be found
375         :product: browse record of the product to find
376         :qty in UoM of product
377         """
378         result = []
379         domain = domain or [('qty', '>', 0.0)]
380         if restrict_partner_id:
381             domain += [('owner_id', '=', restrict_partner_id)]
382         if restrict_lot_id:
383             domain += [('lot_id', '=', restrict_lot_id)]
384         if location:
385             removal_strategy = self.pool.get('stock.location').get_removal_strategy(cr, uid, location, product, context=context) or 'fifo'
386             if removal_strategy == 'fifo':
387                 result += self._quants_get_fifo(cr, uid, location, product, qty, domain, context=context)
388             elif removal_strategy == 'lifo':
389                 result += self._quants_get_lifo(cr, uid, location, product, qty, domain, context=context)
390             else:
391                 raise osv.except_osv(_('Error!'), _('Removal strategy %s not implemented.' % (removal_strategy,)))
392         return result
393
394     def _quant_create(self, cr, uid, qty, move, lot_id=False, owner_id=False, src_package_id=False, dest_package_id=False, force_location=False, context=None):
395         '''Create a quant in the destination location and create a negative quant in the source location if it's an internal location.
396         '''
397         if context is None:
398             context = {}
399         price_unit = self.pool.get('stock.move').get_price_unit(cr, uid, move, context=context)
400         location = force_location or move.location_dest_id
401         vals = {
402             'product_id': move.product_id.id,
403             'location_id': location.id,
404             'qty': qty,
405             'cost': price_unit,
406             'history_ids': [(4, move.id)],
407             'in_date': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
408             'company_id': move.company_id.id,
409             'lot_id': lot_id,
410             'owner_id': owner_id,
411             'package_id': dest_package_id,
412         }
413
414         if move.location_id.usage == 'internal':
415             #if we were trying to move something from an internal location and reach here (quant creation),
416             #it means that a negative quant has to be created as well.
417             negative_vals = vals.copy()
418             negative_vals['location_id'] = move.location_id.id
419             negative_vals['qty'] = -qty
420             negative_vals['cost'] = price_unit
421             negative_vals['negative_dest_location_id'] = move.location_dest_id.id
422             negative_vals['package_id'] = src_package_id
423             negative_quant_id = self.create(cr, SUPERUSER_ID, negative_vals, context=context)
424             vals.update({'propagated_from_id': negative_quant_id})
425
426         #create the quant as superuser, because we want to restrict the creation of quant manually: they should always use this method to create quants
427         quant_id = self.create(cr, SUPERUSER_ID, vals, context=context)
428         return self.browse(cr, uid, quant_id, context=context)
429
430     def _quant_split(self, cr, uid, quant, qty, context=None):
431         context = context or {}
432         if (quant.qty > 0 and quant.qty <= qty) or (quant.qty <= 0 and quant.qty >= qty):
433             return False
434         new_quant = self.copy(cr, SUPERUSER_ID, quant.id, default={'qty': quant.qty - qty}, context=context)
435         self.write(cr, SUPERUSER_ID, quant.id, {'qty': qty}, context=context)
436         quant.refresh()
437         return self.browse(cr, uid, new_quant, context=context)
438
439     def _get_latest_move(self, cr, uid, quant, context=None):
440         move = False
441         for m in quant.history_ids:
442             if not move or m.date > move.date:
443                 move = m
444         return move
445
446     def _quants_merge(self, cr, uid, solved_quant_ids, solving_quant, context=None):
447         path = []
448         for move in solving_quant.history_ids:
449             path.append((4, move.id))
450         self.write(cr, SUPERUSER_ID, solved_quant_ids, {'history_ids': path}, context=context)
451
452     def _quant_reconcile_negative(self, cr, uid, quant, context=None):
453         """
454             When new quant arrive in a location, try to reconcile it with
455             negative quants. If it's possible, apply the cost of the new
456             quant to the conter-part of the negative quant.
457         """
458         if quant.location_id.usage != 'internal':
459             return False
460         solving_quant = quant
461         dom = [('qty', '<', 0)]
462         dom += [('lot_id', '=', quant.lot_id and quant.lot_id.id or False)]
463         dom += [('owner_id', '=', quant.owner_id and quant.owner_id.id or False)]
464         dom += [('package_id', '=', quant.package_id and quant.package_id.id or False)]
465         quants = self.quants_get(cr, uid, quant.location_id, quant.product_id, quant.qty, [('qty', '<', '0')], context=context)
466         for quant_neg, qty in quants:
467             if not quant_neg:
468                 continue
469             to_solve_quant_ids = self.search(cr, uid, [('propagated_from_id', '=', quant_neg.id)], context=context)
470             if not to_solve_quant_ids:
471                 continue
472             solving_qty = qty
473             solved_quant_ids = []
474             for to_solve_quant in self.browse(cr, uid, to_solve_quant_ids, context=context):
475                 if solving_qty <= 0:
476                     continue
477                 solved_quant_ids.append(to_solve_quant.id)
478                 self._quant_split(cr, uid, to_solve_quant, min(solving_qty, to_solve_quant.qty), context=context)
479                 solving_qty -= min(solving_qty, to_solve_quant.qty)
480             remaining_solving_quant = self._quant_split(cr, uid, solving_quant, qty, context=context)
481             remaining_neg_quant = self._quant_split(cr, uid, quant_neg, -qty, context=context)
482             #if the reconciliation was not complete, we need to link together the remaining parts
483             if remaining_neg_quant:
484                 remaining_to_solve_quant_ids = self.search(cr, uid, [('propagated_from_id', '=', quant_neg.id), ('id', 'not in', solved_quant_ids)], context=context)
485                 if remaining_to_solve_quant_ids:
486                     self.write(cr, SUPERUSER_ID, remaining_to_solve_quant_ids, {'propagated_from_id': remaining_neg_quant.id}, context=context)
487             #delete the reconciled quants, as it is replaced by the solved quants
488             self.unlink(cr, SUPERUSER_ID, [quant_neg.id], context=context)
489             #price update + accounting entries adjustments
490             self._price_update(cr, uid, solved_quant_ids, solving_quant.cost, context=context)
491             #merge history (and cost?)
492             self._quants_merge(cr, uid, solved_quant_ids, solving_quant, context=context)
493             self.unlink(cr, SUPERUSER_ID, [solving_quant.id], context=context)
494             solving_quant = remaining_solving_quant
495
496     def _price_update(self, cr, uid, ids, newprice, context=None):
497         self.write(cr, SUPERUSER_ID, ids, {'cost': newprice}, context=context)
498
499     def write(self, cr, uid, ids, vals, context=None):
500         #We want to trigger the move with nothing on reserved_quant_ids for the store of the remaining quantity
501         if 'reservation_id' in vals:
502             reservation_ids = self.browse(cr, uid, ids, context=context)
503             moves_to_warn = set()
504             for reser in reservation_ids:
505                 if reser.reservation_id:
506                     moves_to_warn.add(reser.reservation_id.id)
507             self.pool.get('stock.move').write(cr, uid, list(moves_to_warn), {'reserved_quant_ids': []}, context=context)
508         return super(stock_quant, self).write(cr, SUPERUSER_ID, ids, vals, context=context)
509
510     def quants_unreserve(self, cr, uid, move, context=None):
511         related_quants = [x.id for x in move.reserved_quant_ids]
512         return self.write(cr, SUPERUSER_ID, related_quants, {'reservation_id': False, 'link_move_operation_id': False}, context=context)
513
514     def _quants_get_order(self, cr, uid, location, product, quantity, domain=[], orderby='in_date', context=None):
515         ''' Implementation of removal strategies
516             If it can not reserve, it will return a tuple (None, qty)
517         '''
518         domain += location and [('location_id', 'child_of', location.id)] or []
519         domain += [('product_id', '=', product.id)] + domain
520         res = []
521         offset = 0
522         while quantity > 0:
523             quants = self.search(cr, uid, domain, order=orderby, limit=10, offset=offset, context=context)
524             if not quants:
525                 res.append((None, quantity))
526                 break
527             for quant in self.browse(cr, uid, quants, context=context):
528                 if quantity >= abs(quant.qty):
529                     res += [(quant, abs(quant.qty))]
530                     quantity -= abs(quant.qty)
531                 elif quantity != 0:
532                     res += [(quant, quantity)]
533                     quantity = 0
534                     break
535             offset += 10
536         return res
537
538     def _quants_get_fifo(self, cr, uid, location, product, quantity, domain=[], context=None):
539         order = 'in_date'
540         return self._quants_get_order(cr, uid, location, product, quantity, domain, order, context=context)
541
542     def _quants_get_lifo(self, cr, uid, location, product, quantity, domain=[], context=None):
543         order = 'in_date desc'
544         return self._quants_get_order(cr, uid, location, product, quantity, domain, order, context=context)
545
546     def _location_owner(self, cr, uid, quant, location, context=None):
547         ''' Return the company owning the location if any '''
548         return location and (location.usage == 'internal') and location.company_id or False
549
550     def _check_location(self, cr, uid, ids, context=None):
551         for record in self.browse(cr, uid, ids, context=context):
552             if record.location_id.usage == 'view':
553                 raise osv.except_osv(_('Error'), _('You cannot move product %s to a location of type view %s.') % (record.product_id.name, record.location_id.name))
554         return True
555
556     # FP Note: rehab this, with the auto creation algo
557     # def _check_tracking(self, cr, uid, ids, context=None):
558     #     """ Checks if serial number is assigned to stock move or not.
559     #     @return: True or False
560     #     """
561     #     for move in self.browse(cr, uid, ids, context=context):
562     #         if not move.lot_id and \
563     #            (move.state == 'done' and \
564     #            ( \
565     #                (move.product_id.track_production and move.location_id.usage == 'production') or \
566     #                (move.product_id.track_production and move.location_dest_id.usage == 'production') or \
567     #                (move.product_id.track_incoming and move.location_id.usage == 'supplier') or \
568     #                (move.product_id.track_outgoing and move.location_dest_id.usage == 'customer') or \
569     #                (move.product_id.track_incoming and move.location_id.usage == 'inventory') \
570     #            )):
571     #             return False
572     #     return True
573
574     _constraints = [
575         (_check_location, 'You cannot move products to a location of the type view.', ['location_id'])
576         #(_check_tracking, 'You must assign a serial number for this product.', ['prodlot_id']),
577     ]
578
579
580 #----------------------------------------------------------
581 # Stock Picking
582 #----------------------------------------------------------
583
584 class stock_picking(osv.osv):
585     _name = "stock.picking"
586     _inherit = ['mail.thread']
587     _description = "Picking List"
588     _order = "priority desc, date desc, id desc"
589
590     def _set_min_date(self, cr, uid, id, field, value, arg, context=None):
591         move_obj = self.pool.get("stock.move")
592         if value:
593             move_ids = [move.id for move in self.browse(cr, uid, id, context=context).move_lines]
594             move_obj.write(cr, uid, move_ids, {'date_expected': value}, context=context)
595
596     def get_min_max_date(self, cr, uid, ids, field_name, arg, context=None):
597         """ Finds minimum and maximum dates for picking.
598         @return: Dictionary of values
599         """
600         res = {}
601         for id in ids:
602             res[id] = {'min_date': False, 'max_date': False}
603         if not ids:
604             return res
605         cr.execute("""select
606                 picking_id,
607                 min(date_expected),
608                 max(date_expected)
609             from
610                 stock_move
611             where
612                 picking_id IN %s
613             group by
614                 picking_id""", (tuple(ids),))
615         for pick, dt1, dt2 in cr.fetchall():
616             res[pick]['min_date'] = dt1
617             res[pick]['max_date'] = dt2
618         return res
619
620     def create(self, cr, user, vals, context=None):
621         context = context or {}
622         if ('name' not in vals) or (vals.get('name') in ('/', False)):
623             ptype_id = vals.get('picking_type_id', context.get('default_picking_type_id', False))
624             sequence_id = self.pool.get('stock.picking.type').browse(cr, user, ptype_id, context=context).sequence_id.id
625             vals['name'] = self.pool.get('ir.sequence').get_id(cr, user, sequence_id, 'id', context=context)
626         return super(stock_picking, self).create(cr, user, vals, context)
627
628     def _state_get(self, cr, uid, ids, field_name, arg, context=None):
629         '''The state of a picking depends on the state of its related stock.move
630             draft: the picking has no line or any one of the lines is draft
631             done, draft, cancel: all lines are done / draft / cancel
632             confirmed, auto, assigned depends on move_type (all at once or direct)
633         '''
634         res = {}
635         for pick in self.browse(cr, uid, ids, context=context):
636             if (not pick.move_lines) or any([x.state == 'draft' for x in pick.move_lines]):
637                 res[pick.id] = 'draft'
638                 continue
639             if all([x.state == 'cancel' for x in pick.move_lines]):
640                 res[pick.id] = 'cancel'
641                 continue
642             if all([x.state in ('cancel', 'done') for x in pick.move_lines]):
643                 res[pick.id] = 'done'
644                 continue
645
646             order = {'confirmed': 0, 'waiting': 1, 'assigned': 2}
647             order_inv = dict(zip(order.values(), order.keys()))
648             lst = [order[x.state] for x in pick.move_lines if x.state not in ('cancel', 'done')]
649             if pick.move_lines == 'one':
650                 res[pick.id] = order_inv[min(lst)]
651             else:
652                 res[pick.id] = order_inv[max(lst)]
653         return res
654
655     def _get_pickings(self, cr, uid, ids, context=None):
656         res = set()
657         for move in self.browse(cr, uid, ids, context=context):
658             if move.picking_id:
659                 res.add(move.picking_id.id)
660         return list(res)
661
662     def _get_pack_operation_exist(self, cr, uid, ids, field_name, arg, context=None):
663         res = {}
664         for pick in self.browse(cr, uid, ids, context=context):
665             res[pick.id] = False
666             if pick.pack_operation_ids:
667                 res[pick.id] = True
668         return res
669
670     def _get_quant_reserved_exist(self, cr, uid, ids, field_name, arg, context=None):
671         res = {}
672         for pick in self.browse(cr, uid, ids, context=context):
673             res[pick.id] = False
674             for move in pick.move_lines:
675                 if move.reserved_quant_ids:
676                     res[pick.id] = True
677                     continue
678         return res
679
680     def action_assign_owner(self, cr, uid, ids, context=None):
681         for picking in self.browse(cr, uid, ids, context=context):
682             packop_ids = [op.id for op in picking.pack_operation_ids]
683             self.pool.get('stock.pack.operation').write(cr, uid, packop_ids, {'owner_id': picking.owner_id.id}, context=context)
684
685     _columns = {
686         'name': fields.char('Reference', size=64, select=True, states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
687         'origin': fields.char('Source Document', size=64, states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}, help="Reference of the document", select=True),
688         'backorder_id': fields.many2one('stock.picking', 'Back Order of', states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}, help="If this shipment was split, then this field links to the shipment which contains the already processed part.", select=True),
689         'note': fields.text('Notes', states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
690         'move_type': fields.selection([('direct', 'Partial'), ('one', 'All at once')], 'Delivery Method', required=True, states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}, help="It specifies goods to be deliver partially or all at once"),
691         'state': fields.function(_state_get, type="selection", store={
692             'stock.picking': (lambda self, cr, uid, ids, ctx: ids, ['move_type', 'move_lines'], 20),
693             'stock.move': (_get_pickings, ['state', 'picking_id'], 20)}, selection=[
694                 ('draft', 'Draft'),
695                 ('cancel', 'Cancelled'),
696                 ('waiting', 'Waiting Another Operation'),
697                 ('confirmed', 'Waiting Availability'),
698                 ('assigned', 'Ready to Transfer'),
699                 ('done', 'Transferred'),
700                 ], string='Status', readonly=True, select=True, track_visibility='onchange', help="""
701                 * Draft: not confirmed yet and will not be scheduled until confirmed\n
702                 * Waiting Another Operation: waiting for another move to proceed before it becomes automatically available (e.g. in Make-To-Order flows)\n
703                 * Waiting Availability: still waiting for the availability of products\n
704                 * Ready to Transfer: products reserved, simply waiting for confirmation.\n
705                 * Transferred: has been processed, can't be modified or cancelled anymore\n
706                 * Cancelled: has been cancelled, can't be confirmed anymore"""
707         ),
708         'priority': fields.selection([('0', 'Low'), ('1', 'Normal'), ('2', 'High')], string='Priority', required=True),
709         'min_date': fields.function(get_min_max_date, multi="min_max_date", fnct_inv=_set_min_date,
710                  store={'stock.move': (_get_pickings, ['state', 'date_expected'], 20)}, type='datetime', string='Scheduled Date', select=1, help="Scheduled time for the first part of the shipment to be processed"),
711         'max_date': fields.function(get_min_max_date, multi="min_max_date",
712                  store={'stock.move': (_get_pickings, ['state', 'date_expected'], 20)}, type='datetime', string='Max. Expected Date', select=2, help="Scheduled time for the last part of the shipment to be processed"),
713         'date': fields.datetime('Commitment Date', help="Date promised for the completion of the transfer order, usually set the time of the order and revised later on.", select=True, states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
714         'date_done': fields.datetime('Date of Transfer', help="Date of Completion", states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
715         'move_lines': fields.one2many('stock.move', 'picking_id', 'Internal Moves', states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
716         'quant_reserved_exist': fields.function(_get_quant_reserved_exist, type='boolean', string='Quant already reserved ?', help='technical field used to know if there is already at least one quant reserved on moves of a given picking'),
717         'partner_id': fields.many2one('res.partner', 'Partner', states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
718         'company_id': fields.many2one('res.company', 'Company', required=True, select=True, states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
719         'pack_operation_ids': fields.one2many('stock.pack.operation', 'picking_id', string='Related Packing Operations'),
720         'pack_operation_exist': fields.function(_get_pack_operation_exist, type='boolean', string='Pack Operation Exists?', help='technical field for attrs in view'),
721         'picking_type_id': fields.many2one('stock.picking.type', 'Picking Type', required=True),
722
723         'owner_id': fields.many2one('res.partner', 'Owner', help="Default Owner"),
724         # Used to search on pickings
725         'product_id': fields.related('move_lines', 'product_id', type='many2one', relation='product.product', string='Product'),
726         'location_id': fields.related('move_lines', 'location_id', type='many2one', relation='stock.location', string='Location', readonly=True),
727         'location_dest_id': fields.related('move_lines', 'location_dest_id', type='many2one', relation='stock.location', string='Destination Location', readonly=True),
728         'group_id': fields.related('move_lines', 'group_id', type='many2one', relation='procurement.group', string='Procurement Group', readonly=True,
729               store={
730                   'stock.picking': (lambda self, cr, uid, ids, ctx: ids, ['move_lines'], 10),
731                   'stock.move': (_get_pickings, ['group_id', 'picking_id'], 10),
732               }),
733     }
734
735     _defaults = {
736         'name': lambda self, cr, uid, context: '/',
737         'state': 'draft',
738         'move_type': 'one',
739         'priority': '1',  # normal
740         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
741         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.picking', context=c)
742     }
743     _sql_constraints = [
744         ('name_uniq', 'unique(name, company_id)', 'Reference must be unique per company!'),
745     ]
746
747     def copy(self, cr, uid, id, default=None, context=None):
748         if default is None:
749             default = {}
750         default = default.copy()
751         picking_obj = self.browse(cr, uid, id, context=context)
752         if ('name' not in default) or (picking_obj.name == '/'):
753             default['name'] = '/'
754         if not default.get('backorder_id'):
755             default['backorder_id'] = False
756         default['pack_operation_ids'] = []
757         return super(stock_picking, self).copy(cr, uid, id, default, context)
758
759     def action_confirm(self, cr, uid, ids, context=None):
760         todo = []
761         todo_force_assign = []
762         for picking in self.browse(cr, uid, ids, context=context):
763             if picking.picking_type_id.auto_force_assign:
764                 todo_force_assign.append(picking.id)
765             for r in picking.move_lines:
766                 if r.state == 'draft':
767                     todo.append(r.id)
768         if len(todo):
769             self.pool.get('stock.move').action_confirm(cr, uid, todo, context=context)
770
771         if todo_force_assign:
772             self.force_assign(cr, uid, todo_force_assign, context=context)
773         return True
774
775     def action_assign(self, cr, uid, ids, context=None):
776         """ Check availability of picking moves.
777         This has the effect of changing the state and reserve quants on available moves, and may
778         also impact the state of the picking as it is computed based on move's states.
779         @return: True
780         """
781         for pick in self.browse(cr, uid, ids, context=context):
782             if pick.state == 'draft':
783                 self.action_confirm(cr, uid, [pick.id], context=context)
784             #skip the moves that don't need to be checked
785             move_ids = [x.id for x in pick.move_lines if x.state not in ('draft', 'cancel', 'done')]
786             if not move_ids:
787                 raise osv.except_osv(_('Warning!'), _('Nothing to check the availability for.'))
788             self.pool.get('stock.move').action_assign(cr, uid, move_ids, context=context)
789         return True
790
791     def force_assign(self, cr, uid, ids, context=None):
792         """ Changes state of picking to available if moves are confirmed or waiting.
793         @return: True
794         """
795         for pick in self.browse(cr, uid, ids, context=context):
796             move_ids = [x.id for x in pick.move_lines if x.state in ['confirmed', 'waiting']]
797             self.pool.get('stock.move').force_assign(cr, uid, move_ids, context=context)
798         return True
799
800     def cancel_assign(self, cr, uid, ids, *args):
801         """ Cancels picking and moves.
802         @return: True
803         """
804         for pick in self.browse(cr, uid, ids):
805             move_ids = [x.id for x in pick.move_lines]
806             self.pool.get('stock.move').cancel_assign(cr, uid, move_ids)
807         return True
808
809     def action_cancel(self, cr, uid, ids, context=None):
810         for pick in self.browse(cr, uid, ids, context=context):
811             ids2 = [move.id for move in pick.move_lines]
812             self.pool.get('stock.move').action_cancel(cr, uid, ids2, context)
813         return True
814
815     def action_done(self, cr, uid, ids, context=None):
816         """Changes picking state to done by processing the Stock Moves of the Picking
817
818         Normally that happens when the button "Done" is pressed on a Picking view.
819         @return: True
820         """
821         for pick in self.browse(cr, uid, ids, context=context):
822             todo = []
823             for move in pick.move_lines:
824                 if move.state == 'draft':
825                     self.pool.get('stock.move').action_confirm(cr, uid, [move.id],
826                         context=context)
827                     todo.append(move.id)
828                 elif move.state in ('assigned', 'confirmed'):
829                     todo.append(move.id)
830             if len(todo):
831                 self.pool.get('stock.move').action_done(cr, uid, todo, context=context)
832         return True
833
834     def unlink(self, cr, uid, ids, context=None):
835         move_obj = self.pool.get('stock.move')
836         context = context or {}
837         for pick in self.browse(cr, uid, ids, context=context):
838             ids2 = [move.id for move in pick.move_lines]
839             move_obj.action_cancel(cr, uid, ids2, context=context)
840             move_obj.unlink(cr, uid, ids2, context=context)
841         return super(stock_picking, self).unlink(cr, uid, ids, context=context)
842
843     def write(self, cr, uid, ids, vals, context=None):
844         res = super(stock_picking, self).write(cr, uid, ids, vals, context=context)
845         if 'move_lines' in vals or 'pack_operation_ids' in vals:
846             self.do_recompute_remaining_quantities(cr, uid, ids, context=context)
847         return res
848
849     def _create_backorder(self, cr, uid, picking, backorder_moves=[], context=None):
850         """
851             Move all non-done lines into a new backorder picking. If the key 'do_only_split' is given in the context, then move all lines not in context.get('split', []) instead of all non-done lines.
852         """
853         if not backorder_moves:
854             backorder_moves = picking.move_lines
855         backorder_move_ids = [x.id for x in backorder_moves if x.state not in ('done', 'cancel')]
856         if 'do_only_split' in context and context['do_only_split']:
857             backorder_move_ids = [x.id for x in backorder_moves if x.id not in context.get('split', [])]
858
859         if backorder_move_ids:
860             backorder_id = self.copy(cr, uid, picking.id, {
861                 'name': '/',
862                 'move_lines': [],
863                 'pack_operation_ids': [],
864                 'backorder_id': picking.id,
865             })
866             back_order_name = self.browse(cr, uid, backorder_id, context=context).name
867             self.message_post(cr, uid, picking.id, body=_("Back order <em>%s</em> <b>created</b>.") % (back_order_name), context=context)
868             move_obj = self.pool.get("stock.move")
869             move_obj.write(cr, uid, backorder_move_ids, {'picking_id': backorder_id}, context=context)
870
871             self.pool.get("stock.picking").action_confirm(cr, uid, [picking.id], context=context)
872             self.action_confirm(cr, uid, [backorder_id], context=context)
873             return backorder_id
874         return False
875
876     def do_prepare_partial(self, cr, uid, picking_ids, context=None):
877         context = context or {}
878         pack_operation_obj = self.pool.get('stock.pack.operation')
879         pack_obj = self.pool.get("stock.quant.package")
880         quant_obj = self.pool.get("stock.quant")
881         for picking in self.browse(cr, uid, picking_ids, context=context):
882             for move in picking.move_lines:
883                 if move.state != 'assigned':
884                     continue
885                 #Check which of the reserved quants are entirely in packages (can be in separate method)
886                 packages = list(set([x.package_id for x in move.reserved_quant_ids if x.package_id]))
887                 done_packages = []
888                 for pack in packages:
889                     cont = True
890                     good_pack = False
891                     test_pack = pack
892                     while cont:
893                         quants = pack_obj.get_content(cr, uid, [test_pack.id], context=context)
894                         if all([x.reservation_id.id == move.id for x in quant_obj.browse(cr, uid, quants, context=context) if x.reservation_id]):
895                             good_pack = test_pack.id
896                             if test_pack.parent_id:
897                                 test_pack = test_pack.parent_id
898                             else:
899                                 cont = False
900                         else:
901                             cont = False
902                     if good_pack:
903                         done_packages.append(good_pack)
904                 done_packages = list(set(done_packages))
905
906                 #Create package operations
907                 reserved = set([x.id for x in move.reserved_quant_ids])
908                 remaining_qty = move.product_qty
909                 for pack in pack_obj.browse(cr, uid, done_packages, context=context):
910                     quantl = pack_obj.get_content(cr, uid, [pack.id], context=context)
911                     for quant in quant_obj.browse(cr, uid, quantl, context=context):
912                         remaining_qty -= quant.qty
913                     quants = set(pack_obj.get_content(cr, uid, [pack.id], context=context))
914                     reserved -= quants
915                     pack_operation_obj.create(cr, uid, {
916                         'picking_id': picking.id,
917                         'package_id': pack.id,
918                         'product_qty': 1.0,
919                     }, context=context)
920
921                 yet_to_reserve = list(reserved)
922                 #Create operations based on quants
923                 for quant in quant_obj.browse(cr, uid, yet_to_reserve, context=context):
924                     qty = min(quant.qty, move.product_qty)
925                     remaining_qty -= qty
926                     pack_operation_obj.create(cr, uid, {
927                         'picking_id': picking.id,
928                         'product_qty': qty,
929                         'quant_id': quant.id,
930                         'product_id': quant.product_id.id,
931                         'lot_id': quant.lot_id and quant.lot_id.id or False,
932                         'product_uom_id': quant.product_id.uom_id.id,
933                         'owner_id': quant.owner_id and quant.owner_id.id or False,
934                         'cost': quant.cost,
935                         'package_id': quant.package_id and quant.package_id.id or False,
936                     }, context=context)
937                 if remaining_qty > 0:
938                     pack_operation_obj.create(cr, uid, {
939                         'picking_id': picking.id,
940                         'product_qty': remaining_qty,
941                         'product_id': move.product_id.id,
942                         'product_uom_id': move.product_id.uom_id.id,
943                         'cost': move.product_id.standard_price,
944                     }, context=context)
945
946     def do_unreserve(self, cr, uid, picking_ids, context=None):
947         """
948           Will remove all quants for picking in picking_ids
949         """
950         moves_to_unreserve = []
951         for picking in self.browse(cr, uid, picking_ids, context=context):
952             moves_to_unreserve += [m.id for m in picking.move_lines]
953         if moves_to_unreserve:
954             self.pool.get('stock.move').do_unreserve(cr, uid, moves_to_unreserve, context=context)
955
956     def do_recompute_remaining_quantities(self, cr, uid, picking_ids, context=None):
957         def _create_link_for_product(product_id, qty):
958             qty_to_assign = qty
959             for move in picking.move_lines:
960                 if move.product_id.id == product_id:
961                     qty_on_link = min(move.remaining_qty, qty_to_assign)
962                     link_obj.create(cr, uid, {'move_id': move.id, 'operation_id': op.id, 'qty': qty_on_link}, context=context)
963                     qty_to_assign -= qty_on_link
964                     if qty_to_assign <= 0:
965                         break
966
967         link_obj = self.pool.get('stock.move.operation.link')
968         uom_obj = self.pool.get('product.uom')
969         package_obj = self.pool.get('stock.quant.package')
970         for picking in self.browse(cr, uid, picking_ids, context=context):
971             for op in picking.pack_operation_ids:
972                 to_unlink_ids = [x.id for x in op.linked_move_operation_ids]
973                 if to_unlink_ids:
974                     link_obj.unlink(cr, uid, to_unlink_ids, context=context)
975                 if op.product_id:
976                     normalized_qty = uom_obj._compute_qty(cr, uid, op.product_uom_id.id, op.product_qty, op.product_id.uom_id.id)
977                     _create_link_for_product(op.product_id.id, normalized_qty)
978                 elif op.package_id:
979                     for product_id, qty in package_obj._get_all_products_quantities(cr, uid, op.package_id.id, context=context).items():
980                         _create_link_for_product(product_id, qty)
981
982     def _create_extra_moves(self, cr, uid, picking, context=None):
983         '''This function creates move lines on a picking, at the time of do_transfer, based on unexpected product transfers (or exceeding quantities) found in the pack operations
984         '''
985         move_obj = self.pool.get('stock.move')
986         operation_obj = self.pool.get('stock.pack.operation')
987         for op in picking.pack_operation_ids:
988             for product_id, remaining_qty in operation_obj._get_remaining_prod_quantities(cr, uid, op, context=context).items():
989                 if remaining_qty > 0:
990                     product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
991                     vals = {
992                         'picking_id': picking.id,
993                         'location_id': picking.location_id.id,
994                         'location_dest_id': picking.location_dest_id.id,
995                         'product_id': product_id,
996                         'product_uom': product.uom_id.id,
997                         'product_uom_qty': remaining_qty,
998                         'name': _('Extra Move: ') + product.name,
999                         'state': 'confirmed',
1000                     }
1001                     move_obj.create(cr, uid, vals, context=context)
1002         self.do_recompute_remaining_quantities(cr, uid, [picking.id], context=context)
1003
1004     def rereserve_quants(self, cr, uid, picking, move_ids=[], context=None):
1005         stock_move_obj = self.pool.get('stock.move')
1006         if not move_ids:
1007             self.do_unreserve(cr, uid, [picking.id], context=context)
1008             self.action_assign(cr, uid, [picking.id], context=context)
1009         else:
1010             stock_move_obj.do_unreserve(cr, uid, move_ids, context=context)
1011             stock_move_obj.action_assign(cr, uid, move_ids, context=context)
1012
1013     def do_transfer(self, cr, uid, picking_ids, context=None):
1014         """
1015             If no pack operation, we do simple action_done of the picking
1016             Otherwise, do the pack operations
1017         """
1018         if not context:
1019             context = {}
1020         stock_move_obj = self.pool.get('stock.move')
1021         for picking in self.browse(cr, uid, picking_ids, context=context):
1022             if not picking.pack_operation_ids:
1023                 self.action_done(cr, uid, [picking.id], context=context)
1024                 continue
1025             else:
1026                 self.do_recompute_remaining_quantities(cr, uid, [picking.id], context=context)
1027                 #create extra moves in the picking (unexpected product moves coming from pack operations)
1028                 self._create_extra_moves(cr, uid, picking, context=context)
1029                 picking.refresh()
1030                 #split move lines eventually
1031                 todo_move_ids = []
1032                 toassign_move_ids = []
1033                 for move in picking.move_lines:
1034                     if move.state == 'draft':
1035                         toassign_move_ids.append(move.id)
1036                     if move.remaining_qty == 0:
1037                         if move.state in ('draft', 'assigned', 'confirmed'):
1038                             todo_move_ids.append(move.id)
1039                     elif move.remaining_qty > 0:
1040                         new_move = stock_move_obj.split(cr, uid, move, move.remaining_qty, context=context)
1041                         todo_move_ids.append(move.id)
1042                         #Assign move as it was assigned before
1043                         toassign_move_ids.append(new_move)
1044                     else:
1045                         #this should never happens
1046                         raise
1047                 self.rereserve_quants(cr, uid, picking, move_ids=todo_move_ids, context=context)
1048                 if todo_move_ids and not context.get('do_only_split'):
1049                     self.pool.get('stock.move').action_done(cr, uid, todo_move_ids, context=context)
1050                 elif context.get('do_only_split'):
1051                     context.update({'split': todo_move_ids})
1052             picking.refresh()
1053             self._create_backorder(cr, uid, picking, context=context)
1054             if toassign_move_ids:
1055                 stock_move_obj.action_assign(cr, uid, toassign_move_ids, context=context)
1056         return True
1057
1058     def do_split(self, cr, uid, picking_ids, context=None):
1059         """
1060             just split the picking without making it 'done'
1061         """
1062         if context is None:
1063             context = {}
1064         ctx = context.copy()
1065         ctx['do_only_split'] = True
1066         self.do_transfer(cr, uid, picking_ids, context=ctx)
1067         return True
1068
1069     # Methods for the barcode UI
1070     def get_picking_for_packing_ui(self, cr, uid, context=None):
1071         return self.search(cr, uid, [('state', 'in', ('confirmed', 'assigned')), ('picking_type_id', '=', context.get('default_picking_type_id'))], context=context)
1072
1073     def action_done_from_packing_ui(self, cr, uid, picking_id, only_split_lines=False, context=None):
1074         self.do_transfer(cr, uid, picking_id, only_split_lines, context=context)
1075         #return id of next picking to work on
1076         return self.get_picking_for_packing_ui(cr, uid, context=context)
1077
1078     def action_pack(self, cr, uid, picking_ids, context=None):
1079         stock_operation_obj = self.pool.get('stock.pack.operation')
1080         package_obj = self.pool.get('stock.quant.package')
1081         for picking_id in picking_ids:
1082             operation_ids = stock_operation_obj.search(cr, uid, [('picking_id', '=', picking_id), ('result_package_id', '=', False)], context=context)
1083             if operation_ids:
1084                 package_id = package_obj.create(cr, uid, {}, context=context)
1085                 stock_operation_obj.write(cr, uid, operation_ids, {'result_package_id': package_id}, context=context)
1086         return True
1087
1088     def _deal_with_quants(self, cr, uid, picking_id, quant_ids, context=None):
1089         stock_operation_obj = self.pool.get('stock.pack.operation')
1090         todo_on_moves = []
1091         todo_on_operations = []
1092         for quant in self.pool.get('stock.quant').browse(cr, uid, quant_ids, context=context):
1093             tmp_moves, tmp_operations = stock_operation_obj._search_and_increment(cr, uid, picking_id, ('quant_id', '=', quant.id), context=context)
1094             todo_on_moves += tmp_moves
1095             todo_on_operations += tmp_operations
1096         return todo_on_moves, todo_on_operations
1097
1098     def get_barcode_and_return_todo_stuff(self, cr, uid, picking_id, barcode_str, context=None):
1099         '''This function is called each time there barcode scanner reads an input'''
1100         #TODO: better error messages handling => why not real raised errors
1101         quant_obj = self.pool.get('stock.quant')
1102         package_obj = self.pool.get('stock.quant.package')
1103         product_obj = self.pool.get('product.product')
1104         stock_operation_obj = self.pool.get('stock.pack.operation')
1105         error_msg = ''
1106         todo_on_moves = []
1107         todo_on_operations = []
1108         #check if the barcode correspond to a product
1109         matching_product_ids = product_obj.search(cr, uid, [('ean13', '=', barcode_str)], context=context)
1110         if matching_product_ids:
1111             todo_on_moves, todo_on_operations = stock_operation_obj._search_and_increment(cr, uid, picking_id, ('product_id', '=', matching_product_ids[0]), context=context)
1112
1113         #check if the barcode correspond to a quant
1114         matching_quant_ids = quant_obj.search(cr, uid, [('name', '=', barcode_str)], context=context)  # TODO need the location clause
1115         if matching_quant_ids:
1116             todo_on_moves, todo_on_operations = self._deal_with_quants(cr, uid, picking_id, [matching_quant_ids[0]], context=context)
1117
1118         #check if the barcode correspond to a package
1119         matching_package_ids = package_obj.search(cr, uid, [('name', '=', barcode_str)], context=context)
1120         if matching_package_ids:
1121             included_package_ids = package_obj.search(cr, uid, [('parent_id', 'child_of', matching_package_ids[0])], context=context)
1122             included_quant_ids = quant_obj.search(cr, uid, [('package_id', 'in', included_package_ids)], context=context)
1123             todo_on_moves, todo_on_operations = self._deal_with_quants(cr, uid, picking_id, included_quant_ids, context=context)
1124         #write remaining qty on stock.move, to ease the treatment server side
1125         for todo in todo_on_moves:
1126             if todo[0] == 1:
1127                 self.pool.get('stock.move').write(cr, uid, todo[1], todo[2], context=context)
1128             elif todo[0] == 0:
1129                 self.pool.get('stock.move').create(cr, uid, todo[2], context=context)
1130         return {'warnings': error_msg, 'moves_to_update': todo_on_moves, 'operations_to_update': todo_on_operations}
1131
1132
1133 class stock_production_lot(osv.osv):
1134     _name = 'stock.production.lot'
1135     _inherit = ['mail.thread']
1136     _description = 'Lot/Serial'
1137     _columns = {
1138         'name': fields.char('Serial Number', size=64, required=True, help="Unique Serial Number"),
1139         'ref': fields.char('Internal Reference', size=256, help="Internal reference number in case it differs from the manufacturer's serial number"),
1140         'product_id': fields.many2one('product.product', 'Product', required=True, domain=[('type', '<>', 'service')]),
1141         'quant_ids': fields.one2many('stock.quant', 'lot_id', 'Quants'),
1142         'create_date': fields.datetime('Creation Date'),
1143     }
1144     _defaults = {
1145         'name': lambda x, y, z, c: x.pool.get('ir.sequence').get(y, z, 'stock.lot.serial'),
1146         'product_id': lambda x, y, z, c: c.get('product_id', False),
1147     }
1148     _sql_constraints = [
1149         ('name_ref_uniq', 'unique (name, ref)', 'The combination of Serial Number and internal reference must be unique !'),
1150     ]
1151
1152
1153 # ----------------------------------------------------
1154 # Move
1155 # ----------------------------------------------------
1156
1157 class stock_move(osv.osv):
1158     _name = "stock.move"
1159     _description = "Stock Move"
1160     _order = 'date_expected desc, id'
1161     _log_create = False
1162
1163     def get_price_unit(self, cr, uid, move, context=None):
1164         """ Returns the unit price to store on the quant """
1165         return move.price_unit or move.product_id.standard_price
1166
1167     def name_get(self, cr, uid, ids, context=None):
1168         res = []
1169         for line in self.browse(cr, uid, ids, context=context):
1170             name = line.location_id.name + ' > ' + line.location_dest_id.name
1171             if line.product_id.code:
1172                 name = line.product_id.code + ': ' + name
1173             if line.picking_id.origin:
1174                 name = line.picking_id.origin + '/ ' + name
1175             res.append((line.id, name))
1176         return res
1177
1178     def _quantity_normalize(self, cr, uid, ids, name, args, context=None):
1179         uom_obj = self.pool.get('product.uom')
1180         res = {}
1181         for m in self.browse(cr, uid, ids, context=context):
1182             res[m.id] = uom_obj._compute_qty_obj(cr, uid, m.product_uom, m.product_uom_qty, m.product_id.uom_id, round=False)
1183         return res
1184
1185     def _get_remaining_qty(self, cr, uid, ids, field_name, args, context=None):
1186         uom_obj = self.pool.get('product.uom')
1187         res = {}
1188         for move in self.browse(cr, uid, ids, context=context):
1189             qty = move.product_qty
1190             for record in move.linked_move_operation_ids:
1191                 qty -= record.qty
1192             #converting the remaining quantity in the move UoM
1193             res[move.id] = uom_obj._compute_qty(cr, uid, move.product_id.uom_id.id, qty, move.product_uom.id)
1194         return res
1195
1196     def _get_lot_ids(self, cr, uid, ids, field_name, args, context=None):
1197         res = dict.fromkeys(ids, False)
1198         for move in self.browse(cr, uid, ids, context=context):
1199             if move.state == 'done':
1200                 res[move.id] = [q.id for q in move.quant_ids]
1201             else:
1202                 res[move.id] = [q.id for q in move.reserved_quant_ids]
1203         return res
1204
1205     def _get_product_availability(self, cr, uid, ids, field_name, args, context=None):
1206         quant_obj = self.pool.get('stock.quant')
1207         res = dict.fromkeys(ids, False)
1208         for move in self.browse(cr, uid, ids, context=context):
1209             if move.state == 'done':
1210                 res[move.id] = move.product_qty
1211             else:
1212                 sublocation_ids = self.pool.get('stock.location').search(cr, uid, [('id', 'child_of', [move.location_id.id])], context=context)
1213                 quant_ids = quant_obj.search(cr, uid, [('location_id', 'in', sublocation_ids), ('product_id', '=', move.product_id.id), ('reservation_id', '=', False)], context=context)
1214                 availability = 0
1215                 for quant in quant_obj.browse(cr, uid, quant_ids, context=context):
1216                     availability += quant.qty
1217                 res[move.id] = min(move.product_qty, availability)
1218         return res
1219
1220     def _get_move(self, cr, uid, ids, context=None):
1221         res = set()
1222         for quant in self.browse(cr, uid, ids, context=context):
1223             if quant.reservation_id:
1224                 res.add(quant.reservation_id.id)
1225         return list(res)
1226
1227     _columns = {
1228         'name': fields.char('Description', required=True, select=True),
1229         'priority': fields.selection([('0', 'Not urgent'), ('1', 'Urgent')], 'Priority'),
1230         'create_date': fields.datetime('Creation Date', readonly=True, select=True),
1231         'date': fields.datetime('Date', required=True, select=True, help="Move date: scheduled date until move is done, then date of actual move processing", states={'done': [('readonly', True)]}),
1232         'date_expected': fields.datetime('Scheduled Date', states={'done': [('readonly', True)]}, required=True, select=True, help="Scheduled date for the processing of this move"),
1233         'product_id': fields.many2one('product.product', 'Product', required=True, select=True, domain=[('type', '<>', 'service')], states={'done': [('readonly', True)]}),
1234         # TODO: improve store to add dependency on product UoM
1235         'product_qty': fields.function(_quantity_normalize, type='float', store=True, string='Quantity',
1236             digits_compute=dp.get_precision('Product Unit of Measure'),
1237             help='Quantity in the default UoM of the product'),
1238         'product_uom_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product Unit of Measure'),
1239             required=True, states={'done': [('readonly', True)]},
1240             help="This is the quantity of products from an inventory "
1241                 "point of view. For moves in the state 'done', this is the "
1242                 "quantity of products that were actually moved. For other "
1243                 "moves, this is the quantity of product that is planned to "
1244                 "be moved. Lowering this quantity does not generate a "
1245                 "backorder. Changing this quantity on assigned moves affects "
1246                 "the product reservation, and should be done with care."
1247         ),
1248         'product_uom': fields.many2one('product.uom', 'Unit of Measure', required=True, states={'done': [('readonly', True)]}),
1249         'product_uos_qty': fields.float('Quantity (UOS)', digits_compute=dp.get_precision('Product Unit of Measure'), states={'done': [('readonly', True)]}),
1250         'product_uos': fields.many2one('product.uom', 'Product UOS', states={'done': [('readonly', True)]}),
1251
1252         'product_packaging': fields.many2one('product.packaging', 'Prefered Packaging', help="It specifies attributes of packaging like type, quantity of packaging,etc."),
1253
1254         'location_id': fields.many2one('stock.location', 'Source Location', required=True, select=True, states={'done': [('readonly', True)]}, help="Sets a location if you produce at a fixed location. This can be a partner location if you subcontract the manufacturing operations."),
1255         'location_dest_id': fields.many2one('stock.location', 'Destination Location', required=True, states={'done': [('readonly', True)]}, select=True, help="Location where the system will stock the finished products."),
1256
1257         # FP Note: should we remove this?
1258         'partner_id': fields.many2one('res.partner', 'Destination Address ', states={'done': [('readonly', True)]}, help="Optional address where goods are to be delivered, specifically used for allotment"),
1259
1260
1261         'move_dest_id': fields.many2one('stock.move', 'Destination Move', help="Optional: next stock move when chaining them", select=True),
1262         'move_orig_ids': fields.one2many('stock.move', 'move_dest_id', 'Original Move', help="Optional: previous stock move when chaining them", select=True),
1263
1264         'picking_id': fields.many2one('stock.picking', 'Reference', select=True, states={'done': [('readonly', True)]}),
1265         'picking_priority': fields.related('picking_id', 'priority', type='selection', selection=[('0', 'Low'), ('1', 'Normal'), ('2', 'High')], string='Picking Priority'),
1266         'note': fields.text('Notes'),
1267         'state': fields.selection([('draft', 'New'),
1268                                    ('cancel', 'Cancelled'),
1269                                    ('waiting', 'Waiting Another Move'),
1270                                    ('confirmed', 'Waiting Availability'),
1271                                    ('assigned', 'Available'),
1272                                    ('done', 'Done'),
1273                                    ], 'Status', readonly=True, select=True,
1274                  help= "* New: When the stock move is created and not yet confirmed.\n"\
1275                        "* Waiting Another Move: This state can be seen when a move is waiting for another one, for example in a chained flow.\n"\
1276                        "* Waiting Availability: This state is reached when the procurement resolution is not straight forward. It may need the scheduler to run, a component to me manufactured...\n"\
1277                        "* Available: When products are reserved, it is set to \'Available\'.\n"\
1278                        "* Done: When the shipment is processed, the state is \'Done\'."),
1279
1280         'price_unit': fields.float('Unit Price', help="Technical field used to record the product cost set by the user during a picking confirmation (when average price costing method is used). Value given in company currency and in product uom."),  # as it's a technical field, we intentionally don't provide the digits attribute
1281
1282         'company_id': fields.many2one('res.company', 'Company', required=True, select=True),
1283         'backorder_id': fields.related('picking_id', 'backorder_id', type='many2one', relation="stock.picking", string="Back Order of", select=True),
1284         'origin': fields.char("Source"),
1285         'procure_method': fields.selection([('make_to_stock', 'Make to Stock'), ('make_to_order', 'Make to Order')], 'Procurement Method', required=True, help="Make to Stock: When needed, the product is taken from the stock or we wait for replenishment. \nMake to Order: When needed, the product is purchased or produced."),
1286
1287         # used for colors in tree views:
1288         'scrapped': fields.related('location_dest_id', 'scrap_location', type='boolean', relation='stock.location', string='Scrapped', readonly=True),
1289
1290         'quant_ids': fields.many2many('stock.quant', 'stock_quant_move_rel', 'move_id', 'quant_id', 'Moved Quants'),
1291         'reserved_quant_ids': fields.one2many('stock.quant', 'reservation_id', 'Reserved quants'),
1292         'linked_move_operation_ids': fields.one2many('stock.move.operation.link', 'move_id', string='Linked Operations', readonly=True, help='Operations that impact this move for the computation of the remaining quantities'),
1293         'remaining_qty': fields.function(_get_remaining_qty, type='float', string='Remaining Quantity',
1294                                          digits_compute=dp.get_precision('Product Unit of Measure'), states={'done': [('readonly', True)]},),
1295         'procurement_id': fields.many2one('procurement.order', 'Procurement'),
1296         'group_id': fields.many2one('procurement.group', 'Procurement Group'),
1297         'rule_id': fields.many2one('procurement.rule', 'Procurement Rule', help='The pull rule that created this stock move'),
1298         'propagate': fields.boolean('Propagate cancel and split', help='If checked, when this move is cancelled, cancel the linked move too'),
1299         'picking_type_id': fields.many2one('stock.picking.type', 'Picking Type'),
1300         'inventory_id': fields.many2one('stock.inventory', 'Inventory'),
1301         'lot_ids': fields.function(_get_lot_ids, type='many2many', relation='stock.quant', string='Lots'),
1302         'origin_returned_move_id': fields.many2one('stock.move', 'Origin return move', help='move that created the return move'),
1303         'returned_move_ids': fields.one2many('stock.move', 'origin_returned_move_id', 'All returned moves', help='Optional: all returned moves created from this move'),
1304         'availability': fields.function(_get_product_availability, type='float', string='Availability'),
1305         'restrict_lot_id': fields.many2one('stock.production.lot', 'Lot', help="Technical field used to depict a restriction on the lot of quants to consider when marking this move as 'done'"),
1306         'restrict_partner_id': fields.many2one('res.partner', 'Owner ', help="Technical field used to depict a restriction on the ownership of quants to consider when marking this move as 'done'"),
1307         'putaway_ids': fields.one2many('stock.move.putaway', 'move_id', 'Put Away Suggestions'),
1308         'route_ids': fields.many2many('stock.location.route', 'stock_location_route_move', 'move_id', 'route_id', 'Destination route', help="Preferred route to be followed by the procurement order"),
1309         'warehouse_id': fields.many2one('stock.warehouse', 'Warehouse', help="Technical field depicting the warehouse to consider for the route selection on the next procurement (if any)."),
1310     }
1311
1312     def _default_location_destination(self, cr, uid, context=None):
1313         context = context or {}
1314         if context.get('default_picking_type_id', False):
1315             pick_type = self.pool.get('stock.picking.type').browse(cr, uid, context['default_picking_type_id'], context=context)
1316             return pick_type.default_location_dest_id and pick_type.default_location_dest_id.id or False
1317         return False
1318
1319     def _default_location_source(self, cr, uid, context=None):
1320         context = context or {}
1321         if context.get('default_picking_type_id', False):
1322             pick_type = self.pool.get('stock.picking.type').browse(cr, uid, context['default_picking_type_id'], context=context)
1323             return pick_type.default_location_src_id and pick_type.default_location_src_id.id or False
1324         return False
1325
1326     def _default_destination_address(self, cr, uid, context=None):
1327         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
1328         return user.company_id.partner_id.id
1329
1330     _defaults = {
1331         'location_id': _default_location_source,
1332         'location_dest_id': _default_location_destination,
1333         'partner_id': _default_destination_address,
1334         'state': 'draft',
1335         'priority': '1',
1336         'product_qty': 1.0,
1337         'product_uom_qty': 1.0,
1338         'scrapped': False,
1339         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
1340         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.move', context=c),
1341         'date_expected': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
1342         'procure_method': 'make_to_stock',
1343         'propagate': True,
1344     }
1345
1346     def _check_uom(self, cr, uid, ids, context=None):
1347         for move in self.browse(cr, uid, ids, context=context):
1348             if move.product_id.uom_id.category_id.id != move.product_uom.category_id.id:
1349                 return False
1350         return True
1351
1352     _constraints = [
1353         (_check_uom,
1354             'You try to move a product using a UoM that is not compatible with the UoM of the product moved. Please use an UoM in the same UoM category.',
1355             ['product_uom'])]
1356
1357     def copy_data(self, cr, uid, id, default=None, context=None):
1358         if default is None:
1359             default = {}
1360         default = default.copy()
1361         default['move_orig_ids'] = []
1362         default['quant_ids'] = []
1363         default['reserved_quant_ids'] = []
1364         default['returned_move_ids'] = []
1365         default['linked_move_operation_ids'] = []
1366         default['origin_returned_move_id'] = False
1367         default['state'] = 'draft'
1368         return super(stock_move, self).copy_data(cr, uid, id, default, context)
1369
1370     def do_unreserve(self, cr, uid, move_ids, context=None):
1371         ids_to_free = []
1372         quant_obj = self.pool.get("stock.quant")
1373         for move in self.browse(cr, uid, move_ids, context=context):
1374             quant_obj.quants_unreserve(cr, uid, move, context=context)
1375
1376     def _prepare_procurement_from_move(self, cr, uid, move, context=None):
1377         origin = (move.group_id and (move.group_id.name + ":") or "") + (move.rule_id and move.rule_id.name or "/")
1378         group_id = move.group_id and move.group_id.id or False
1379         if move.rule_id:
1380             if move.rule_id.group_propagation_option == 'fixed' and move.rule_id.group_id:
1381                 group_id = move.rule_id.group_id.id
1382             elif move.rule_id.group_propagation_option == 'none':
1383                 group_id = False
1384         return {
1385             'name': move.rule_id and move.rule_id.name or "/",
1386             'origin': origin,
1387             'company_id': move.company_id and move.company_id.id or False,
1388             'date_planned': move.date,
1389             'product_id': move.product_id.id,
1390             'product_qty': move.product_qty,
1391             'product_uom': move.product_uom.id,
1392             'product_uos_qty': (move.product_uos and move.product_uos_qty) or move.product_qty,
1393             'product_uos': (move.product_uos and move.product_uos.id) or move.product_uom.id,
1394             'location_id': move.location_id.id,
1395             'move_dest_id': move.id,
1396             'group_id': group_id,
1397             'route_ids': [(4, x.id) for x in move.route_ids],
1398             'warehouse_id': move.warehouse_id and move.warehouse_id.id or False,
1399         }
1400
1401     def _push_apply(self, cr, uid, moves, context):
1402         push_obj = self.pool.get("stock.location.path")
1403         for move in moves:
1404             if not move.move_dest_id:
1405                 routes = [x.id for x in move.product_id.route_ids + move.product_id.categ_id.total_route_ids]
1406                 routes = routes or [x.id for x in move.route_ids]
1407                 if routes:
1408                     domain = [('route_id', 'in', routes), ('location_from_id', '=', move.location_dest_id.id)]
1409                     if move.warehouse_id:
1410                         domain += [('warehouse_id', '=', move.warehouse_id.id)]
1411                     rules = push_obj.search(cr, uid, domain, context=context)
1412                     if rules:
1413                         rule = push_obj.browse(cr, uid, rules[0], context=context)
1414                         push_obj._apply(cr, uid, rule, move, context=context)
1415         return True
1416
1417     # Create the stock.move.putaway records
1418     def _putaway_apply(self, cr, uid, ids, context=None):
1419         moveputaway_obj = self.pool.get('stock.move.putaway')
1420         for move in self.browse(cr, uid, ids, context=context):
1421             putaway = self.pool.get('stock.location').get_putaway_strategy(cr, uid, move.location_dest_id, move.product_id, context=context)
1422             if putaway:
1423                 # Should call different methods here in later versions
1424                 # TODO: take care of lots
1425                 if putaway.method == 'fixed' and putaway.location_spec_id:
1426                     moveputaway_obj.create(cr, SUPERUSER_ID, {'move_id': move.id,
1427                                                      'location_id': putaway.location_spec_id.id,
1428                                                      'quantity': move.product_qty}, context=context)
1429         return True
1430
1431     def _create_procurement(self, cr, uid, move, context=None):
1432         """ This will create a procurement order """
1433         return self.pool.get("procurement.order").create(cr, uid, self._prepare_procurement_from_move(cr, uid, move, context=context))
1434
1435     # Check that we do not modify a stock.move which is done
1436     def write(self, cr, uid, ids, vals, context=None):
1437         if isinstance(ids, (int, long)):
1438             ids = [ids]
1439         frozen_fields = set(['product_qty', 'product_uom', 'product_uos_qty', 'product_uos', 'location_id', 'location_dest_id', 'product_id'])
1440         for move in self.browse(cr, uid, ids, context=context):
1441             if move.state == 'done':
1442                 if frozen_fields.intersection(vals):
1443                     raise osv.except_osv(_('Operation Forbidden!'),
1444                         _('Quantities, Units of Measure, Products and Locations cannot be modified on stock moves that have already been processed (except by the Administrator).'))
1445         return super(stock_move, self).write(cr, uid, ids, vals, context=context)
1446
1447     def onchange_quantity(self, cr, uid, ids, product_id, product_qty, product_uom, product_uos):
1448         """ On change of product quantity finds UoM and UoS quantities
1449         @param product_id: Product id
1450         @param product_qty: Changed Quantity of product
1451         @param product_uom: Unit of measure of product
1452         @param product_uos: Unit of sale of product
1453         @return: Dictionary of values
1454         """
1455         result = {
1456             'product_uos_qty': 0.00
1457         }
1458         warning = {}
1459
1460         if (not product_id) or (product_qty <= 0.0):
1461             result['product_qty'] = 0.0
1462             return {'value': result}
1463
1464         product_obj = self.pool.get('product.product')
1465         uos_coeff = product_obj.read(cr, uid, product_id, ['uos_coeff'])
1466
1467         # Warn if the quantity was decreased
1468         if ids:
1469             for move in self.read(cr, uid, ids, ['product_qty']):
1470                 if product_qty < move['product_qty']:
1471                     warning.update({
1472                         'title': _('Information'),
1473                         'message': _("By changing this quantity here, you accept the "
1474                                 "new quantity as complete: OpenERP will not "
1475                                 "automatically generate a back order.")})
1476                 break
1477
1478         if product_uos and product_uom and (product_uom != product_uos):
1479             result['product_uos_qty'] = product_qty * uos_coeff['uos_coeff']
1480         else:
1481             result['product_uos_qty'] = product_qty
1482
1483         return {'value': result, 'warning': warning}
1484
1485     def onchange_uos_quantity(self, cr, uid, ids, product_id, product_uos_qty,
1486                           product_uos, product_uom):
1487         """ On change of product quantity finds UoM and UoS quantities
1488         @param product_id: Product id
1489         @param product_uos_qty: Changed UoS Quantity of product
1490         @param product_uom: Unit of measure of product
1491         @param product_uos: Unit of sale of product
1492         @return: Dictionary of values
1493         """
1494         result = {
1495             'product_uom_qty': 0.00
1496         }
1497         warning = {}
1498
1499         if (not product_id) or (product_uos_qty <= 0.0):
1500             result['product_uos_qty'] = 0.0
1501             return {'value': result}
1502
1503         product_obj = self.pool.get('product.product')
1504         uos_coeff = product_obj.read(cr, uid, product_id, ['uos_coeff'])
1505
1506         # Warn if the quantity was decreased
1507         for move in self.read(cr, uid, ids, ['product_uos_qty']):
1508             if product_uos_qty < move['product_uos_qty']:
1509                 warning.update({
1510                     'title': _('Warning: No Back Order'),
1511                     'message': _("By changing the quantity here, you accept the "
1512                                 "new quantity as complete: OpenERP will not "
1513                                 "automatically generate a Back Order.")})
1514                 break
1515
1516         if product_uos and product_uom and (product_uom != product_uos):
1517             result['product_uom_qty'] = product_uos_qty / uos_coeff['uos_coeff']
1518         else:
1519             result['product_uom_qty'] = product_uos_qty
1520         return {'value': result, 'warning': warning}
1521
1522     def onchange_product_id(self, cr, uid, ids, prod_id=False, loc_id=False, loc_dest_id=False, partner_id=False):
1523         """ On change of product id, if finds UoM, UoS, quantity and UoS quantity.
1524         @param prod_id: Changed Product id
1525         @param loc_id: Source location id
1526         @param loc_dest_id: Destination location id
1527         @param partner_id: Address id of partner
1528         @return: Dictionary of values
1529         """
1530         if not prod_id:
1531             return {}
1532         user = self.pool.get('res.users').browse(cr, uid, uid)
1533         lang = user and user.lang or False
1534         if partner_id:
1535             addr_rec = self.pool.get('res.partner').browse(cr, uid, partner_id)
1536             if addr_rec:
1537                 lang = addr_rec and addr_rec.lang or False
1538         ctx = {'lang': lang}
1539
1540         product = self.pool.get('product.product').browse(cr, uid, [prod_id], context=ctx)[0]
1541         uos_id = product.uos_id and product.uos_id.id or False
1542         result = {
1543             'product_uom': product.uom_id.id,
1544             'product_uos': uos_id,
1545             'product_uom_qty': 1.00,
1546             'product_uos_qty': self.pool.get('stock.move').onchange_quantity(cr, uid, ids, prod_id, 1.00, product.uom_id.id, uos_id)['value']['product_uos_qty'],
1547         }
1548         if not ids:
1549             result['name'] = product.partner_ref
1550         if loc_id:
1551             result['location_id'] = loc_id
1552         if loc_dest_id:
1553             result['location_dest_id'] = loc_dest_id
1554         return {'value': result}
1555
1556     def _picking_assign(self, cr, uid, move, context=None):
1557         if move.picking_id or not move.picking_type_id:
1558             return False
1559         context = context or {}
1560         pick_obj = self.pool.get("stock.picking")
1561         picks = []
1562         group = move.group_id and move.group_id.id or False
1563         picks = pick_obj.search(cr, uid, [
1564                 ('group_id', '=', group),
1565                 ('location_id', '=', move.location_id.id),
1566                 ('location_dest_id', '=', move.location_dest_id.id),
1567                 ('state', 'in', ['draft', 'confirmed', 'waiting'])], context=context)
1568         if picks:
1569             pick = picks[0]
1570         else:
1571             values = {
1572                 'origin': move.origin,
1573                 'company_id': move.company_id and move.company_id.id or False,
1574                 'move_type': move.group_id and move.group_id.move_type or 'one',
1575                 'partner_id': move.group_id and move.group_id.partner_id and move.group_id.partner_id.id or False,
1576                 'date_done': move.date_expected,
1577                 'picking_type_id': move.picking_type_id and move.picking_type_id.id or False,
1578             }
1579             pick = pick_obj.create(cr, uid, values, context=context)
1580         move.write({'picking_id': pick})
1581         return True
1582
1583     def onchange_date(self, cr, uid, ids, date, date_expected, context=None):
1584         """ On change of Scheduled Date gives a Move date.
1585         @param date_expected: Scheduled Date
1586         @param date: Move Date
1587         @return: Move Date
1588         """
1589         if not date_expected:
1590             date_expected = time.strftime('%Y-%m-%d %H:%M:%S')
1591         return {'value': {'date': date_expected}}
1592
1593     def action_confirm(self, cr, uid, ids, context=None):
1594         """ Confirms stock move or put it in waiting if it's linked to another move.
1595         @return: List of ids.
1596         """
1597         states = {
1598             'confirmed': [],
1599             'waiting': []
1600         }
1601         for move in self.browse(cr, uid, ids, context=context):
1602             state = 'confirmed'
1603             for m in move.move_orig_ids:
1604                 if m.state not in ('done', 'cancel'):
1605                     state = 'waiting'
1606             states[state].append(move.id)
1607             self._picking_assign(cr, uid, move, context=context)
1608
1609         for state, write_ids in states.items():
1610             if len(write_ids):
1611                 self.write(cr, uid, write_ids, {'state': state})
1612                 if state == 'confirmed':
1613                     for move in self.browse(cr, uid, write_ids, context=context):
1614                         if move.procure_method == 'make_to_order':
1615                             self._create_procurement(cr, uid, move, context=context)
1616         moves = self.browse(cr, uid, ids, context=context)
1617         self._push_apply(cr, uid, moves, context=context)
1618         return True
1619
1620     def force_assign(self, cr, uid, ids, context=None):
1621         """ Changes the state to assigned.
1622         @return: True
1623         """
1624         self.action_assign(cr, uid, ids, context=context)
1625         self.write(cr, uid, ids, {'state': 'assigned'})
1626         return True
1627
1628     def cancel_assign(self, cr, uid, ids, context=None):
1629         """ Changes the state to confirmed.
1630         @return: True
1631         """
1632         return self.write(cr, uid, ids, {'state': 'confirmed'})
1633
1634     def action_assign(self, cr, uid, ids, context=None):
1635         """ Checks the product type and accordingly writes the state.
1636         @return: No. of moves done
1637         """
1638         context = context or {}
1639         quant_obj = self.pool.get("stock.quant")
1640         done = []
1641         for move in self.browse(cr, uid, ids, context=context):
1642             if move.state not in ('confirmed', 'waiting', 'assigned'):
1643                 continue
1644             if move.product_id.type == 'consu':
1645                 done.append(move.id)
1646                 continue
1647             else:
1648                 #build the prefered domain based on quants that moved in previous linked done move
1649                 prev_quant_ids = []
1650                 for m2 in move.move_orig_ids:
1651                     for q in m2.quant_ids:
1652                         prev_quant_ids.append(q.id)
1653                 prefered_domain = prev_quant_ids and [('id', 'in', prev_quant_ids)] or []
1654                 fallback_domain = prev_quant_ids and [('id', 'not in', prev_quant_ids)] or []
1655                 #we always keep the quants already assigned and try to find the remaining quantity on quants not assigned only
1656                 main_domain = [('reservation_id', '=', False), ('qty', '>', 0)]
1657                 #first try to find quants based on specific domains given by linked operations
1658                 for record in move.linked_move_operation_ids:
1659                     domain = main_domain + self.pool.get('stock.move.operation.link').get_specific_domain(cr, uid, record, context=context)
1660                     qty_already_assigned = sum([q.qty for q in record.reserved_quant_ids])
1661                     qty = record.qty - qty_already_assigned
1662                     quants = quant_obj.quants_get_prefered_domain(cr, uid, move.location_id, move.product_id, qty, domain=domain, prefered_domain=prefered_domain, fallback_domain=fallback_domain, restrict_lot_id=move.restrict_lot_id.id, restrict_partner_id=move.restrict_partner_id.id, context=context)
1663                     quant_obj.quants_reserve(cr, uid, quants, move, record, context=context)
1664                 #then if the move isn't totally assigned, try to find quants without any specific domain
1665                 if move.state != 'assigned':
1666                     qty_already_assigned = sum([q.qty for q in move.reserved_quant_ids])
1667                     qty = move.product_qty - qty_already_assigned
1668                     quants = quant_obj.quants_get_prefered_domain(cr, uid, move.location_id, move.product_id, qty, domain=main_domain, prefered_domain=prefered_domain, fallback_domain=fallback_domain, restrict_lot_id=move.restrict_lot_id.id, restrict_partner_id=move.restrict_partner_id.id, context=context)
1669                     quant_obj.quants_reserve(cr, uid, quants, move, context=context)
1670
1671         self._putaway_apply(cr, uid, ids, context=context)
1672
1673     def action_cancel(self, cr, uid, ids, context=None):
1674         """ Cancels the moves and if all moves are cancelled it cancels the picking.
1675         @return: True
1676         """
1677         procurement_obj = self.pool.get('procurement.order')
1678         context = context or {}
1679         for move in self.browse(cr, uid, ids, context=context):
1680             if move.state == 'done':
1681                 raise osv.except_osv(_('Operation Forbidden!'),
1682                         _('You cannot cancel a stock move that has been set to \'Done\'.'))
1683             if move.reserved_quant_ids:
1684                 self.pool.get("stock.quant").quants_unreserve(cr, uid, move, context=context)
1685             if context.get('cancel_procurement'):
1686                 if move.propagate:
1687                     procurement_ids = procurement_obj.search(cr, uid, [('move_dest_id', '=', move.id)], context=context)
1688                     procurement_obj.cancel(cr, uid, procurement_ids, context=context)
1689             elif move.move_dest_id:
1690                 #cancel chained moves
1691                 if move.propagate:
1692                     self.action_cancel(cr, uid, [move.move_dest_id.id], context=context)
1693                 elif move.move_dest_id.state == 'waiting':
1694                     self.write(cr, uid, [move.move_dest_id.id], {'state': 'confirmed'})
1695         return self.write(cr, uid, ids, {'state': 'cancel', 'move_dest_id': False})
1696
1697     def action_done(self, cr, uid, ids, context=None):
1698         """ Makes the move done and if all moves are done, it will finish the picking.
1699         It assumes that quants are already assigned to stock moves.
1700         Putaway strategies should be applied
1701         @return:
1702         """
1703         context = context or {}
1704         quant_obj = self.pool.get("stock.quant")
1705         pack_op_obj = self.pool.get("stock.pack.operation")
1706         todo = [move.id for move in self.browse(cr, uid, ids, context=context) if move.state == "draft"]
1707         if todo:
1708             self.action_confirm(cr, uid, todo, context=context)
1709
1710         pickings = set()
1711         procurement_ids = []
1712         for move in self.browse(cr, uid, ids, context=context):
1713             if move.picking_id:
1714                 pickings.add(move.picking_id.id)
1715             qty = move.product_qty
1716             main_domain = [('qty', '>', 0)]
1717             prefered_domain = [('reservation_id', '=', move.id)]
1718             fallback_domain = [('reservation_id', '=', False)]
1719             #first, process the move per linked operation first because it may imply some specific domains to consider
1720             for record in move.linked_move_operation_ids:
1721                 dom = main_domain + self.pool.get('stock.move.operation.link').get_specific_domain(cr, uid, record, context=context)
1722                 quants = quant_obj.quants_get_prefered_domain(cr, uid, move.location_id, move.product_id, record.qty, domain=dom, prefered_domain=prefered_domain, fallback_domain=fallback_domain, context=context)
1723                 package_id = False
1724                 if not record.operation_id.package_id:
1725                     #if a package and a result_package is given, we don't enter here because it will be processed by process_packaging() later
1726                     #but for operations having only result_package_id, we will create new quants in the final package directly
1727                     package_id = record.operation_id.result_package_id.id or False
1728                 quant_obj.quants_move(cr, uid, quants, move, lot_id=record.operation_id.lot_id.id, owner_id=record.operation_id.owner_id.id, src_package_id=record.operation_id.package_id.id, dest_package_id=package_id, context=context)
1729                 #packaging process
1730                 pack_op_obj.process_packaging(cr, uid, record.operation_id, quants, context=context)
1731                 qty -= record.qty
1732             #then if the total quantity processed this way isn't enough, process the remaining quantity without any specific domain
1733             if qty > 0:
1734                 quants = quant_obj.quants_get_prefered_domain(cr, uid, move.location_id, move.product_id, qty, domain=main_domain, prefered_domain=prefered_domain, fallback_domain=fallback_domain, context=context)
1735                 quant_obj.quants_move(cr, uid, quants, move, context=context)
1736             #unreserve the quants and make them available for other operations/moves
1737             quant_obj.quants_unreserve(cr, uid, move, context=context)
1738
1739             #Check moves that were pushed
1740             if move.move_dest_id.state in ('waiting', 'confirmed'):
1741                 other_upstream_move_ids = self.search(cr, uid, [('id', '!=', move.id), ('state', 'not in', ['done', 'cancel']),
1742                                             ('move_dest_id', '=', move.move_dest_id.id)], context=context)
1743                 #If no other moves for the move that got pushed:
1744                 if not other_upstream_move_ids and move.move_dest_id.state in ('waiting', 'confirmed'):
1745                     self.action_assign(cr, uid, [move.move_dest_id.id], context=context)
1746             if move.procurement_id:
1747                 procurement_ids.append(move.procurement_id.id)
1748         self.write(cr, uid, ids, {'state': 'done', 'date': time.strftime(DEFAULT_SERVER_DATETIME_FORMAT)}, context=context)
1749         self.pool.get('procurement.order').check(cr, uid, procurement_ids, context=context)
1750         return True
1751
1752     def unlink(self, cr, uid, ids, context=None):
1753         context = context or {}
1754         for move in self.browse(cr, uid, ids, context=context):
1755             if move.state not in ('draft', 'cancel'):
1756                 raise osv.except_osv(_('User Error!'), _('You can only delete draft moves.'))
1757         return super(stock_move, self).unlink(cr, uid, ids, context=context)
1758
1759     def action_scrap(self, cr, uid, ids, quantity, location_id, context=None):
1760         """ Move the scrap/damaged product into scrap location
1761         @param cr: the database cursor
1762         @param uid: the user id
1763         @param ids: ids of stock move object to be scrapped
1764         @param quantity : specify scrap qty
1765         @param location_id : specify scrap location
1766         @param context: context arguments
1767         @return: Scraped lines
1768         """
1769         #quantity should be given in MOVE UOM
1770         if quantity <= 0:
1771             raise osv.except_osv(_('Warning!'), _('Please provide a positive quantity to scrap.'))
1772         res = []
1773         for move in self.browse(cr, uid, ids, context=context):
1774             source_location = move.location_id
1775             if move.state == 'done':
1776                 source_location = move.location_dest_id
1777             #Previously used to prevent scraping from virtual location but not necessary anymore
1778             #if source_location.usage != 'internal':
1779                 #restrict to scrap from a virtual location because it's meaningless and it may introduce errors in stock ('creating' new products from nowhere)
1780                 #raise osv.except_osv(_('Error!'), _('Forbidden operation: it is not allowed to scrap products from a virtual location.'))
1781             move_qty = move.product_qty
1782             uos_qty = quantity / move_qty * move.product_uos_qty
1783             default_val = {
1784                 'location_id': source_location.id,
1785                 'product_uom_qty': quantity,
1786                 'product_uos_qty': uos_qty,
1787                 'state': move.state,
1788                 'scrapped': True,
1789                 'location_dest_id': location_id,
1790                 #TODO lot_id is now on quant and not on move, need to do something for this
1791                 #'lot_id': move.lot_id.id,
1792             }
1793             new_move = self.copy(cr, uid, move.id, default_val)
1794
1795             res += [new_move]
1796             product_obj = self.pool.get('product.product')
1797             for product in product_obj.browse(cr, uid, [move.product_id.id], context=context):
1798                 if move.picking_id:
1799                     uom = product.uom_id.name if product.uom_id else ''
1800                     message = _("%s %s %s has been <b>moved to</b> scrap.") % (quantity, uom, product.name)
1801                     move.picking_id.message_post(body=message)
1802
1803         self.action_done(cr, uid, res, context=context)
1804         return res
1805
1806     def action_consume(self, cr, uid, ids, quantity, location_id=False, context=None):
1807         """ Consumed product with specific quatity from specific source location
1808         @param cr: the database cursor
1809         @param uid: the user id
1810         @param ids: ids of stock move object to be consumed
1811         @param quantity : specify consume quantity
1812         @param location_id : specify source location
1813         @param context: context arguments
1814         @return: Consumed lines
1815         """
1816         #quantity should be given in MOVE UOM
1817         if context is None:
1818             context = {}
1819         if quantity <= 0:
1820             raise osv.except_osv(_('Warning!'), _('Please provide proper quantity.'))
1821         res = []
1822         for move in self.browse(cr, uid, ids, context=context):
1823             move_qty = move.product_qty
1824             if move_qty <= 0:
1825                 raise osv.except_osv(_('Error!'), _('Cannot consume a move with negative or zero quantity.'))
1826             quantity_rest = move.product_qty
1827             quantity_rest -= quantity
1828             uos_qty_rest = quantity_rest / move_qty * move.product_uos_qty
1829             if quantity_rest <= 0:
1830                 quantity_rest = 0
1831                 uos_qty_rest = 0
1832                 quantity = move.product_qty
1833
1834             uos_qty = quantity / move_qty * move.product_uos_qty
1835             if quantity_rest > 0:
1836                 default_val = {
1837                     'product_uom_qty': quantity,
1838                     'product_uos_qty': uos_qty,
1839                     'state': move.state,
1840                     'location_id': location_id or move.location_id.id,
1841                 }
1842                 current_move = self.copy(cr, uid, move.id, default_val)
1843                 res += [current_move]
1844                 update_val = {}
1845                 update_val['product_uom_qty'] = quantity_rest
1846                 update_val['product_uos_qty'] = uos_qty_rest
1847                 self.write(cr, uid, [move.id], update_val)
1848
1849             else:
1850                 quantity_rest = quantity
1851                 uos_qty_rest =  uos_qty
1852                 res += [move.id]
1853                 update_val = {
1854                         'product_uom_qty' : quantity_rest,
1855                         'product_uos_qty' : uos_qty_rest,
1856                         'location_id': location_id or move.location_id.id,
1857                 }
1858                 self.write(cr, uid, [move.id], update_val)
1859
1860         self.action_done(cr, uid, res, context=context)
1861         return res
1862
1863     def split(self, cr, uid, move, qty, context=None):
1864         """ Splits qty from move move into a new move """
1865         if move.product_qty == qty:
1866             return move.id
1867         if (move.product_qty < qty) or (qty == 0):
1868             return False
1869
1870         uom_obj = self.pool.get('product.uom')
1871         context = context or {}
1872
1873         uom_qty = uom_obj._compute_qty(cr, uid, move.product_id.uom_id.id, qty, move.product_uom.id)
1874         uos_qty = uom_qty * move.product_uos_qty / move.product_uom_qty
1875
1876         if move.state in ('done', 'cancel'):
1877             raise osv.except_osv(_('Error'), _('You cannot split a move done'))
1878
1879         defaults = {
1880             'product_uom_qty': uom_qty,
1881             'product_uos_qty': uos_qty,
1882             'state': move.state,
1883             'move_dest_id': False,
1884             'reserved_quant_ids': []
1885         }
1886         new_move = self.copy(cr, uid, move.id, defaults)
1887
1888         self.write(cr, uid, [move.id], {
1889             'product_uom_qty': move.product_uom_qty - uom_qty,
1890             'product_uos_qty': move.product_uos_qty - uos_qty,
1891             #'reserved_quant_ids': [(6,0,[])]  SHOULD NOT CHANGE as it has been reserved already
1892         }, context=context)
1893
1894         if move.move_dest_id and move.propagate:
1895             new_move_prop = self.split(cr, uid, move.move_dest_id, qty, context=context)
1896             self.write(cr, uid, [new_move], {'move_dest_id': new_move_prop}, context=context)
1897
1898         self.action_confirm(cr, uid, [new_move], context=context)
1899         return new_move
1900
1901 class stock_inventory(osv.osv):
1902     _name = "stock.inventory"
1903     _description = "Inventory"
1904
1905     def _get_move_ids_exist(self, cr, uid, ids, field_name, arg, context=None):
1906         res = {}
1907         for inv in self.browse(cr, uid, ids, context=context):
1908             res[inv.id] = False
1909             if inv.move_ids:
1910                 res[inv.id] = True
1911         return res
1912
1913     def _get_available_filters(self, cr, uid, context=None):
1914         """
1915            This function will return the list of filter allowed according to the options checked
1916            in 'Settings\Warehouse'.
1917
1918            :rtype: list of tuple
1919         """
1920         #default available choices
1921         res_filter = [('none', ' All products of a whole location'), ('product', 'One product only')]
1922         settings_obj = self.pool.get('stock.config.settings')
1923         config_ids = settings_obj.search(cr, uid, [], limit=1, order='id DESC', context=context)
1924         #If we don't have updated config until now, all fields are by default false and so should be not dipslayed
1925         if not config_ids:
1926             return res_filter
1927
1928         stock_settings = settings_obj.browse(cr, uid, config_ids[0], context=context)
1929         if stock_settings.group_stock_tracking_owner:
1930             res_filter.append(('owner', _('One owner only')))
1931             res_filter.append(('product_owner', _('One product for a specific owner')))
1932         if stock_settings.group_stock_production_lot:
1933             res_filter.append(('lot', _('One Lot/Serial Number')))
1934         if stock_settings.group_stock_tracking_lot:
1935             res_filter.append(('pack', _('A Pack')))
1936         return res_filter
1937
1938     _columns = {
1939         'name': fields.char('Inventory Reference', size=64, required=True, readonly=True, states={'draft': [('readonly', False)]}, help="Inventory Name."),
1940         'date': fields.datetime('Inventory Date', required=True, readonly=True, states={'draft': [('readonly', False)]}, help="Inventory Create Date."),
1941         'date_done': fields.datetime('Date done', help="Inventory Validation Date."),
1942         'line_ids': fields.one2many('stock.inventory.line', 'inventory_id', 'Inventories', readonly=False, states={'done': [('readonly', True)]}, help="Inventory Lines."),
1943         'move_ids': fields.one2many('stock.move', 'inventory_id', 'Created Moves', help="Inventory Moves."),
1944         'state': fields.selection([('draft', 'Draft'), ('cancel', 'Cancelled'), ('confirm', 'In Progress'), ('done', 'Validated')], 'Status', readonly=True, select=True),
1945         'company_id': fields.many2one('res.company', 'Company', required=True, select=True, readonly=True, states={'draft': [('readonly', False)]}),
1946         'location_id': fields.many2one('stock.location', 'Location', required=True),
1947         'product_id': fields.many2one('product.product', 'Product', readonly=True, states={'draft': [('readonly', False)]}, help="Specify Product to focus your inventory on a particular Product."),
1948         'package_id': fields.many2one('stock.quant.package', 'Pack', readonly=True, states={'draft': [('readonly', False)]}, help="Specify Pack to focus your inventory on a particular Pack."),
1949         'partner_id': fields.many2one('res.partner', 'Owner', readonly=True, states={'draft': [('readonly', False)]}, help="Specify Owner to focus your inventory on a particular Owner."),
1950         'lot_id': fields.many2one('stock.production.lot', 'Lot/Serial Number', readonly=True, states={'draft': [('readonly', False)]}, help="Specify Lot/Serial Number to focus your inventory on a particular Lot/Serial Number."),
1951         'move_ids_exist': fields.function(_get_move_ids_exist, type='boolean', string=' Stock Move Exists?', help='technical field for attrs in view'),
1952         'filter': fields.selection(_get_available_filters, 'Selection Filter'),
1953     }
1954
1955     def _default_stock_location(self, cr, uid, context=None):
1956         try:
1957             warehouse = self.pool.get('ir.model.data').get_object(cr, uid, 'stock', 'warehouse0')
1958             return warehouse.lot_stock_id.id
1959         except:
1960             return False
1961
1962     _defaults = {
1963         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
1964         'state': 'draft',
1965         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.inventory', context=c),
1966         'location_id': _default_stock_location,
1967     }
1968
1969     def set_checked_qty(self, cr, uid, ids, context=None):
1970         inventory = self.browse(cr, uid, ids[0], context=context)
1971         line_ids = [line.id for line in inventory.line_ids]
1972         self.pool.get('stock.inventory.line').write(cr, uid, line_ids, {'product_qty': 0})
1973         return True
1974
1975     def copy(self, cr, uid, id, default=None, context=None):
1976         if default is None:
1977             default = {}
1978         default = default.copy()
1979         default.update({'move_ids': [], 'date_done': False})
1980         return super(stock_inventory, self).copy(cr, uid, id, default, context=context)
1981
1982     def _inventory_line_hook(self, cr, uid, inventory_line, move_vals):
1983         """ Creates a stock move from an inventory line
1984         @param inventory_line:
1985         @param move_vals:
1986         @return:
1987         """
1988         return self.pool.get('stock.move').create(cr, uid, move_vals)
1989
1990     def action_done(self, cr, uid, ids, context=None):
1991         """ Finish the inventory
1992         @return: True
1993         """
1994         if context is None:
1995             context = {}
1996         move_obj = self.pool.get('stock.move')
1997         for inv in self.browse(cr, uid, ids, context=context):
1998             if not inv.move_ids:
1999                 self.action_check(cr, uid, [inv.id], context=context)
2000             inv.refresh()
2001             #the action_done on stock_move has to be done in 2 steps:
2002             #first, we start moving the products from stock to inventory loss
2003             move_obj.action_done(cr, uid, [x.id for x in inv.move_ids if x.location_id.usage == 'internal'], context=context)
2004             #then, we move from inventory loss. This 2 steps process is needed because some moved quant may need to be put again in stock
2005             move_obj.action_done(cr, uid, [x.id for x in inv.move_ids if x.location_id.usage != 'internal'], context=context)
2006             self.write(cr, uid, [inv.id], {'state': 'done', 'date_done': time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
2007         return True
2008
2009     def _create_stock_move(self, cr, uid, inventory, todo_line, context=None):
2010         stock_move_obj = self.pool.get('stock.move')
2011         product_obj = self.pool.get('product.product')
2012         inventory_location_id = product_obj.browse(cr, uid, todo_line['product_id'], context=context).property_stock_inventory.id
2013         vals = {
2014             'name': _('INV:') + (inventory.name or ''),
2015             'product_id': todo_line['product_id'],
2016             'product_uom': todo_line['product_uom_id'],
2017             'date': inventory.date,
2018             'company_id': inventory.company_id.id,
2019             'inventory_id': inventory.id,
2020             'state': 'assigned',
2021             'restrict_lot_id': todo_line.get('prod_lot_id'),
2022             'restrict_partner_id': todo_line.get('partner_id'),
2023          }
2024
2025         if todo_line['product_qty'] < 0:
2026             #found more than expected
2027             vals['location_id'] = inventory_location_id
2028             vals['location_dest_id'] = todo_line['location_id']
2029             vals['product_uom_qty'] = -todo_line['product_qty']
2030         else:
2031             #found less than expected
2032             vals['location_id'] = todo_line['location_id']
2033             vals['location_dest_id'] = inventory_location_id
2034             vals['product_uom_qty'] = todo_line['product_qty']
2035         return stock_move_obj.create(cr, uid, vals, context=context)
2036
2037     def action_check(self, cr, uid, ids, context=None):
2038         """ Checks the inventory and computes the stock move to do
2039         @return: True
2040         """
2041         inventory_line_obj = self.pool.get('stock.inventory.line')
2042         stock_move_obj = self.pool.get('stock.move')
2043         for inventory in self.browse(cr, uid, ids, context=context):
2044             #first remove the existing stock moves linked to this inventory
2045             move_ids = [move.id for move in inventory.move_ids]
2046             stock_move_obj.unlink(cr, uid, move_ids, context=context)
2047             #compute what should be in the inventory lines
2048             theorical_lines = self._get_inventory_lines(cr, uid, inventory, context=context)
2049             for line in inventory.line_ids:
2050                 #compare the inventory lines to the theorical ones and store the diff in theorical_lines
2051                 inventory_line_obj._resolve_inventory_line(cr, uid, line, theorical_lines, context=context)
2052             #each theorical_lines where product_qty is not 0 is a difference for which we need to create a stock move
2053             for todo_line in theorical_lines:
2054                 if todo_line['product_qty'] != 0:
2055                     self._create_stock_move(cr, uid, inventory, todo_line, context=context)
2056
2057     def action_cancel_draft(self, cr, uid, ids, context=None):
2058         """ Cancels the stock move and change inventory state to draft.
2059         @return: True
2060         """
2061         for inv in self.browse(cr, uid, ids, context=context):
2062             self.pool.get('stock.move').action_cancel(cr, uid, [x.id for x in inv.move_ids], context=context)
2063             self.write(cr, uid, [inv.id], {'state': 'draft'}, context=context)
2064         return True
2065
2066     def action_cancel_inventory(self, cr, uid, ids, context=None):
2067         #TODO test
2068         self.action_cancel_draft(cr, uid, ids, context=context)
2069
2070     def prepare_inventory(self, cr, uid, ids, context=None):
2071         inventory_line_obj = self.pool.get('stock.inventory.line')
2072         for inventory in self.browse(cr, uid, ids, context=context):
2073             #clean the existing inventory lines before redoing an inventory proposal
2074             line_ids = [line.id for line in inventory.line_ids]
2075             inventory_line_obj.unlink(cr, uid, line_ids, context=context)
2076             #compute the inventory lines and create them
2077             vals = self._get_inventory_lines(cr, uid, inventory, context=context)
2078             for product_line in vals:
2079                 inventory_line_obj.create(cr, uid, product_line, context=context)
2080         return self.write(cr, uid, ids, {'state': 'confirm'})
2081
2082     def _get_inventory_lines(self, cr, uid, inventory, context=None):
2083         location_obj = self.pool.get('stock.location')
2084         product_obj = self.pool.get('product.product')
2085         location_ids = location_obj.search(cr, uid, [('id', 'child_of', [inventory.location_id.id])], context=context)
2086         domain = ' location_id in %s'
2087         args = (tuple(location_ids),)
2088         if inventory.partner_id:
2089             domain += ' and owner_id = %s'
2090             args += (inventory.partner_id.id,)
2091         if inventory.lot_id:
2092             domain += ' and lot_id = %s'
2093             args += (inventory.lot_id.id,)
2094         if inventory.product_id:
2095             domain += 'and product_id = %s'
2096             args += (inventory.product_id.id,)
2097         if inventory.package_id:
2098             domain += ' and package_id = %s'
2099             args += (inventory.package_id.id,)
2100         cr.execute('''
2101            SELECT product_id, sum(qty) as product_qty, location_id, lot_id as prod_lot_id, package_id, owner_id as partner_id
2102            FROM stock_quant WHERE''' + domain + '''
2103            GROUP BY product_id, location_id, lot_id, package_id, partner_id
2104         ''', args)
2105         vals = []
2106         for product_line in cr.dictfetchall():
2107             #replace the None the dictionary by False, because falsy values are tested later on
2108             for key, value in product_line.items():
2109                 if not value:
2110                     product_line[key] = False
2111             product_line['inventory_id'] = inventory.id
2112             product_line['th_qty'] = product_line['product_qty']
2113             if product_line['product_id']:
2114                 product = product_obj.browse(cr, uid, product_line['product_id'], context=context)
2115                 product_line['product_uom_id'] = product.uom_id.id
2116             vals.append(product_line)
2117         return vals
2118
2119 class stock_inventory_line(osv.osv):
2120     _name = "stock.inventory.line"
2121     _description = "Inventory Line"
2122     _rec_name = "inventory_id"
2123     _columns = {
2124         'inventory_id': fields.many2one('stock.inventory', 'Inventory', ondelete='cascade', select=True),
2125         'location_id': fields.many2one('stock.location', 'Location', required=True, select=True),
2126         'product_id': fields.many2one('product.product', 'Product', required=True, select=True),
2127         'package_id': fields.many2one('stock.quant.package', 'Pack', select=True),
2128         'product_uom_id': fields.many2one('product.uom', 'Product Unit of Measure', required=True),
2129         'product_qty': fields.float('Checked Quantity', digits_compute=dp.get_precision('Product Unit of Measure')),
2130         'company_id': fields.related('inventory_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, select=True, readonly=True),
2131         'prod_lot_id': fields.many2one('stock.production.lot', 'Serial Number', domain="[('product_id','=',product_id)]"),
2132         'state': fields.related('inventory_id', 'state', type='char', string='Status', readonly=True),
2133         'th_qty': fields.float('Theoretical Quantity', readonly=True),
2134         'partner_id': fields.many2one('res.partner', 'Owner'),
2135     }
2136
2137     _defaults = {
2138         'product_qty': 1,
2139     }
2140
2141     def _resolve_inventory_line(self, cr, uid, inventory_line, theorical_lines, context=None):
2142         #TODO : package_id management !
2143         found = False
2144         uom_obj = self.pool.get('product.uom')
2145         for th_line in theorical_lines:
2146             #We try to match the inventory line with a theorical line with same product, lot, location and owner
2147             if th_line['location_id'] == inventory_line.location_id.id and th_line['product_id'] == inventory_line.product_id.id and th_line['prod_lot_id'] == inventory_line.prod_lot_id.id and th_line['partner_id'] == inventory_line.partner_id.id:
2148                 uom_reference = inventory_line.product_id.uom_id
2149                 real_qty = uom_obj._compute_qty_obj(cr, uid, inventory_line.product_uom_id, inventory_line.product_qty, uom_reference)
2150                 th_line['product_qty'] -= real_qty
2151                 found = True
2152                 break
2153         #if it was still not found, we add it to the theorical lines so that it will create a stock move for it
2154         if not found:
2155             vals = {
2156                 'inventory_id': inventory_line.inventory_id.id,
2157                 'location_id': inventory_line.location_id.id,
2158                 'product_id': inventory_line.product_id.id,
2159                 'product_uom_id': inventory_line.product_id.uom_id.id,
2160                 'product_qty': -inventory_line.product_qty,
2161                 'prod_lot_id': inventory_line.prod_lot_id.id,
2162                 'partner_id': inventory_line.partner_id.id,
2163             }
2164             theorical_lines.append(vals)
2165
2166     def on_change_product_id(self, cr, uid, ids, location_id, product, uom=False, owner_id=False, lot_id=False, package_id=False, context=None):
2167         """ Changes UoM and name if product_id changes.
2168         @param location_id: Location id
2169         @param product: Changed product_id
2170         @param uom: UoM product
2171         @return:  Dictionary of changed values
2172         """
2173         context = context or {}
2174         if not product:
2175             return {'value': {'product_qty': 0.0, 'product_uom_id': False}}
2176         uom_obj = self.pool.get('product.uom')
2177         ctx = context.copy()
2178         ctx['location'] = location_id
2179         ctx['lot_id'] = lot_id
2180         ctx['owner_id'] = owner_id
2181         ctx['package_id'] = package_id
2182         obj_product = self.pool.get('product.product').browse(cr, uid, product, context=ctx)
2183         th_qty = obj_product.qty_available
2184         if uom and uom != obj_product.uom_id.id:
2185             uom_record = uom_obj.browse(cr, uid, uom, context=context)
2186             th_qty = uom_obj._compute_qty_obj(cr, uid, obj_product.uom_id, th_qty, uom_record)
2187         return {'value': {'th_qty': th_qty, 'product_uom_id': uom or obj_product.uom_id.id}}
2188
2189
2190 #----------------------------------------------------------
2191 # Stock Warehouse
2192 #----------------------------------------------------------
2193 class stock_warehouse(osv.osv):
2194     _name = "stock.warehouse"
2195     _description = "Warehouse"
2196
2197     _columns = {
2198         'name': fields.char('Name', size=128, required=True, select=True),
2199         'company_id': fields.many2one('res.company', 'Company', required=True, select=True),
2200         'partner_id': fields.many2one('res.partner', 'Address'),
2201         'view_location_id': fields.many2one('stock.location', 'View Location', required=True, domain=[('usage', '=', 'view')]),
2202         'lot_stock_id': fields.many2one('stock.location', 'Location Stock', required=True, domain=[('usage', '=', 'internal')]),
2203         'code': fields.char('Short Name', size=5, required=True, help="Short name used to identify your warehouse"),
2204         'route_ids': fields.many2many('stock.location.route', 'stock_route_warehouse', 'warehouse_id', 'route_id', 'Routes', domain="[('warehouse_selectable', '=', True)]", help='Defaults routes through the warehouse'),
2205         'reception_steps': fields.selection([
2206             ('one_step', 'Receive goods directly in stock (1 step)'),
2207             ('two_steps', 'Unload in input location then go to stock (2 steps)'),
2208             ('three_steps', 'Unload in input location, go through a quality control before being admitted in stock (3 steps)')], 'Incoming Shipments', required=True),
2209         'delivery_steps': fields.selection([
2210             ('ship_only', 'Ship directly from stock (Ship only)'),
2211             ('pick_ship', 'Bring goods to output location before shipping (Pick + Ship)'),
2212             ('pick_pack_ship', 'Make packages into a dedicated location, then bring them to the output location for shipping (Pick + Pack + Ship)')], 'Outgoing Shippings', required=True),
2213         'wh_input_stock_loc_id': fields.many2one('stock.location', 'Input Location'),
2214         'wh_qc_stock_loc_id': fields.many2one('stock.location', 'Quality Control Location'),
2215         'wh_output_stock_loc_id': fields.many2one('stock.location', 'Output Location'),
2216         'wh_pack_stock_loc_id': fields.many2one('stock.location', 'Packing Location'),
2217         'mto_pull_id': fields.many2one('procurement.rule', 'MTO rule'),
2218         'pick_type_id': fields.many2one('stock.picking.type', 'Pick Type'),
2219         'pack_type_id': fields.many2one('stock.picking.type', 'Pack Type'),
2220         'out_type_id': fields.many2one('stock.picking.type', 'Out Type'),
2221         'in_type_id': fields.many2one('stock.picking.type', 'In Type'),
2222         'int_type_id': fields.many2one('stock.picking.type', 'Internal Type'),
2223         'crossdock_route_id': fields.many2one('stock.location.route', 'Crossdock Route'),
2224         'reception_route_id': fields.many2one('stock.location.route', 'Reception Route'),
2225         'delivery_route_id': fields.many2one('stock.location.route', 'Delivery Route'),
2226         'resupply_from_wh': fields.boolean('Resupply From Other Warehouses'),
2227         'resupply_wh_ids': fields.many2many('stock.warehouse', 'stock_wh_resupply_table', 'supplied_wh_id', 'supplier_wh_id', 'Resupply Warehouses'),
2228         'resupply_route_ids': fields.one2many('stock.location.route', 'supplied_wh_id', 'Resupply Routes'),
2229         'default_resupply_wh_id': fields.many2one('stock.warehouse', 'Default Resupply Warehouse'),
2230     }
2231
2232     def onchange_filter_default_resupply_wh_id(self, cr, uid, ids, default_resupply_wh_id, resupply_wh_ids, context=None):
2233         resupply_wh_ids = set([x['id'] for x in (self.resolve_2many_commands(cr, uid, 'resupply_wh_ids', resupply_wh_ids, ['id']))])
2234         if default_resupply_wh_id: #If we are removing the default resupply, we don't have default_resupply_wh_id 
2235             resupply_wh_ids.add(default_resupply_wh_id)
2236         resupply_wh_ids = list(resupply_wh_ids)        
2237         return {'value': {'resupply_wh_ids': resupply_wh_ids}}
2238
2239     def _get_inter_wh_location(self, cr, uid, warehouse, context=None):
2240         ''' returns a tuple made of the browse record of customer location and the browse record of supplier location'''
2241         data_obj = self.pool.get('ir.model.data')
2242         try:
2243             inter_wh_loc = data_obj.get_object_reference(cr, uid, 'stock', 'stock_location_inter_wh')[1]
2244         except:
2245             inter_wh_loc = False
2246         return inter_wh_loc
2247
2248     def _get_all_products_to_resupply(self, cr, uid, warehouse, context=None):
2249         return self.pool.get('product.product').search(cr, uid, [], context=context)
2250
2251     def _assign_route_on_products(self, cr, uid, warehouse, inter_wh_route_id, context=None):
2252         product_ids = self._get_all_products_to_resupply(cr, uid, warehouse, context=context)
2253         self.pool.get('product.product').write(cr, uid, product_ids, {'route_ids': [(4, inter_wh_route_id)]}, context=context)
2254
2255     def _unassign_route_on_products(self, cr, uid, warehouse, inter_wh_route_id, context=None):
2256         product_ids = self._get_all_products_to_resupply(cr, uid, warehouse, context=context)
2257         self.pool.get('product.product').write(cr, uid, product_ids, {'route_ids': [(3, inter_wh_route_id)]}, context=context)
2258
2259     def _get_inter_wh_route(self, cr, uid, warehouse, wh, context=None):
2260         return {
2261             'name': _('%s: Supply Product from %s') % (warehouse.name, wh.name),
2262             'warehouse_selectable': False,
2263             'product_selectable': True,
2264             'product_categ_selectable': True,
2265             'supplied_wh_id': warehouse.id,
2266             'supplier_wh_id': wh.id,
2267         }
2268
2269     def _create_resupply_routes(self, cr, uid, warehouse, supplier_warehouses, default_resupply_wh, context=None):
2270         location_obj = self.pool.get('stock.location')
2271         route_obj = self.pool.get('stock.location.route')
2272         pull_obj = self.pool.get('procurement.rule')
2273         #create route selectable on the product to resupply the warehouse from another one
2274         inter_wh_location_id = self._get_inter_wh_location(cr, uid, warehouse, context=context)
2275         if inter_wh_location_id:
2276             input_loc = warehouse.wh_input_stock_loc_id
2277             if warehouse.reception_steps == 'one_step':
2278                 input_loc = warehouse.lot_stock_id
2279             inter_wh_location = location_obj.browse(cr, uid, inter_wh_location_id, context=context)
2280             for wh in supplier_warehouses:
2281                 output_loc = wh.wh_output_stock_loc_id
2282                 if wh.delivery_steps == 'ship_only':
2283                     output_loc = wh.lot_stock_id
2284                 inter_wh_route_vals = self._get_inter_wh_route(cr, uid, warehouse, wh, context=context)
2285                 inter_wh_route_id = route_obj.create(cr, uid, vals=inter_wh_route_vals, context=context)
2286                 values = [(output_loc, inter_wh_location, wh.out_type_id.id, wh), (inter_wh_location, input_loc, warehouse.in_type_id.id, warehouse)]
2287                 pull_rules_list = self._get_supply_pull_rules(cr, uid, warehouse, values, inter_wh_route_id, context=context)
2288                 for pull_rule in pull_rules_list:
2289                     pull_obj.create(cr, uid, vals=pull_rule, context=context)
2290                 #if the warehouse is also set as default resupply method, assign this route automatically to all product
2291                 if default_resupply_wh and default_resupply_wh.id == wh.id:
2292                     self._assign_route_on_products(cr, uid, warehouse, inter_wh_route_id, context=context)
2293                 #finally, save the route on the warehouse
2294                 self.write(cr, uid, [warehouse.id], {'route_ids': [(4, inter_wh_route_id)]}, context=context)
2295
2296     def _default_stock_id(self, cr, uid, context=None):
2297         #lot_input_stock = self.pool.get('ir.model.data').get_object(cr, uid, 'stock', 'stock_location_stock')
2298         try:
2299             warehouse = self.pool.get('ir.model.data').get_object(cr, uid, 'stock', 'warehouse0')
2300             return warehouse.lot_stock_id.id
2301         except:
2302             return False
2303
2304     _defaults = {
2305         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.inventory', context=c),
2306         'lot_stock_id': _default_stock_id,
2307         'reception_steps': 'one_step',
2308         'delivery_steps': 'ship_only',
2309     }
2310     _sql_constraints = [
2311         ('warehouse_name_uniq', 'unique(name, company_id)', 'The name of the warehouse must be unique per company!'),
2312         ('warehouse_code_uniq', 'unique(code, company_id)', 'The code of the warehouse must be unique per company!'),
2313         ('default_resupply_wh_diff', 'check (id != default_resupply_wh_id)', 'The default resupply warehouse should be different that the warehouse itself!'),
2314     ]
2315
2316     def _get_partner_locations(self, cr, uid, ids, context=None):
2317         ''' returns a tuple made of the browse record of customer location and the browse record of supplier location'''
2318         data_obj = self.pool.get('ir.model.data')
2319         location_obj = self.pool.get('stock.location')
2320         try:
2321             customer_loc = data_obj.get_object_reference(cr, uid, 'stock', 'stock_location_customers')[1]
2322             supplier_loc = data_obj.get_object_reference(cr, uid, 'stock', 'stock_location_suppliers')[1]
2323         except:
2324             customer_loc = location_obj.search(cr, uid, [('usage', '=', 'customer')], context=context)
2325             customer_loc = customer_loc and customer_loc[0] or False
2326             supplier_loc = location_obj.search(cr, uid, [('usage', '=', 'supplier')], context=context)
2327             supplier_loc = supplier_loc and supplier_loc[0] or False
2328         if not (customer_loc and supplier_loc):
2329             raise osv.except_osv(_('Error!'), _('Can\'t find any customer or supplier location.'))
2330         return location_obj.browse(cr, uid, [customer_loc, supplier_loc], context=context)
2331
2332     def switch_location(self, cr, uid, ids, warehouse, new_reception_step=False, new_delivery_step=False, context=None):
2333         location_obj = self.pool.get('stock.location')
2334
2335         new_reception_step = new_reception_step or warehouse.reception_steps
2336         new_delivery_step = new_delivery_step or warehouse.delivery_steps
2337         if warehouse.reception_steps != new_reception_step:
2338             location_obj.write(cr, uid, [warehouse.wh_input_stock_loc_id.id, warehouse.wh_qc_stock_loc_id.id], {'active': False}, context=context)
2339             if new_reception_step != 'one_step':
2340                 location_obj.write(cr, uid, warehouse.wh_input_stock_loc_id.id, {'active': True}, context=context)
2341             if new_reception_step == 'three_steps':
2342                 location_obj.write(cr, uid, warehouse.wh_qc_stock_loc_id.id, {'active': True}, context=context)
2343
2344         if warehouse.delivery_steps != new_delivery_step:
2345             location_obj.write(cr, uid, [warehouse.wh_output_stock_loc_id.id, warehouse.wh_pack_stock_loc_id.id], {'active': False}, context=context)
2346             if new_delivery_step != 'ship_only':
2347                 location_obj.write(cr, uid, warehouse.wh_output_stock_loc_id.id, {'active': True}, context=context)
2348             if new_delivery_step == 'pick_pack_ship':
2349                 location_obj.write(cr, uid, warehouse.wh_pack_stock_loc_id.id, {'active': True}, context=context)
2350         return True
2351
2352     def _get_reception_delivery_route(self, cr, uid, warehouse, route_name, context=None):
2353         return {
2354             'name': self._format_routename(cr, uid, warehouse, route_name, context=context),
2355             'product_categ_selectable': True,
2356             'product_selectable': False,
2357             'sequence': 10,
2358         }
2359
2360     def _get_supply_pull_rules(self, cr, uid, supplied_warehouse, values, new_route_id, context=None):
2361         pull_rules_list = []
2362         for from_loc, dest_loc, pick_type_id, warehouse in values:
2363             pull_rules_list.append({
2364                 'name': self._format_rulename(cr, uid, warehouse, from_loc, dest_loc, context=context),
2365                 'location_src_id': from_loc.id,
2366                 'location_id': dest_loc.id,
2367                 'route_id': new_route_id,
2368                 'action': 'move',
2369                 'picking_type_id': pick_type_id,
2370                 'procure_method': 'make_to_order',
2371                 'warehouse_id': supplied_warehouse.id,
2372                 'propagate_warehouse_id': warehouse.id,
2373             })
2374         return pull_rules_list
2375
2376     def _get_push_pull_rules(self, cr, uid, warehouse, active, values, new_route_id, context=None):
2377         first_rule = True
2378         push_rules_list = []
2379         pull_rules_list = []
2380         for from_loc, dest_loc, pick_type_id in values:
2381             push_rules_list.append({
2382                 'name': self._format_rulename(cr, uid, warehouse, from_loc, dest_loc, context=context),
2383                 'location_from_id': from_loc.id,
2384                 'location_dest_id': dest_loc.id,
2385                 'route_id': new_route_id,
2386                 'auto': 'manual',
2387                 'picking_type_id': pick_type_id,
2388                 'active': active,
2389                 'warehouse_id': warehouse.id,
2390             })
2391             pull_rules_list.append({
2392                 'name': self._format_rulename(cr, uid, warehouse, from_loc, dest_loc, context=context),
2393                 'location_src_id': from_loc.id,
2394                 'location_id': dest_loc.id,
2395                 'route_id': new_route_id,
2396                 'action': 'move',
2397                 'picking_type_id': pick_type_id,
2398                 'procure_method': first_rule is True and 'make_to_stock' or 'make_to_order',
2399                 'active': active,
2400                 'warehouse_id': warehouse.id,
2401             })
2402             first_rule = False
2403         return push_rules_list, pull_rules_list
2404
2405     def _get_mto_pull_rule(self, cr, uid, warehouse, values, context=None):
2406         route_obj = self.pool.get('stock.location.route')
2407         data_obj = self.pool.get('ir.model.data')
2408         try:
2409             mto_route_id = data_obj.get_object_reference(cr, uid, 'stock', 'route_warehouse0_mto')[1]
2410         except:
2411             mto_route_id = route_obj.search(cr, uid, [('name', 'like', _('MTO'))], context=context)
2412             mto_route_id = mto_route_id and mto_route_id[0] or False
2413         if not mto_route_id:
2414             raise osv.except_osv(_('Error!'), _('Can\'t find any generic MTO route.'))
2415
2416         from_loc, dest_loc, pick_type_id = values[0]
2417         return {
2418             'name': self._format_rulename(cr, uid, warehouse, from_loc, dest_loc, context=context) + _(' MTO'),
2419             'location_src_id': from_loc.id,
2420             'location_id': dest_loc.id,
2421             'route_id': mto_route_id,
2422             'action': 'move',
2423             'picking_type_id': pick_type_id,
2424             'procure_method': 'make_to_order',
2425             'active': True,
2426             'warehouse_id': warehouse.id,
2427         }
2428
2429     def _get_crossdock_route(self, cr, uid, warehouse, route_name, context=None):
2430         return {
2431             'name': self._format_routename(cr, uid, warehouse, route_name, context=context),
2432             'warehouse_selectable': False,
2433             'product_selectable': True,
2434             'product_categ_selectable': True,
2435             'active': warehouse.delivery_steps != 'ship_only' and warehouse.reception_steps != 'one_step',
2436             'sequence': 20,
2437         }
2438
2439     def create_routes(self, cr, uid, ids, warehouse, context=None):
2440         wh_route_ids = []
2441         route_obj = self.pool.get('stock.location.route')
2442         pull_obj = self.pool.get('procurement.rule')
2443         push_obj = self.pool.get('stock.location.path')
2444         routes_dict = self.get_routes_dict(cr, uid, ids, warehouse, context=context)
2445         #create reception route and rules
2446         route_name, values = routes_dict[warehouse.reception_steps]
2447         route_vals = self._get_reception_delivery_route(cr, uid, warehouse, route_name, context=context)
2448         reception_route_id = route_obj.create(cr, uid, route_vals, context=context)
2449         wh_route_ids.append((4, reception_route_id))
2450         push_rules_list, pull_rules_list = self._get_push_pull_rules(cr, uid, warehouse, True, values, reception_route_id, context=context)
2451         #create the push/pull rules
2452         for push_rule in push_rules_list:
2453             push_obj.create(cr, uid, vals=push_rule, context=context)
2454         for pull_rule in pull_rules_list:
2455             #all pull rules in reception route are mto, because we don't want to wait for the scheduler to trigger an orderpoint on input location
2456             pull_rule['procure_method'] = 'make_to_order'
2457             pull_obj.create(cr, uid, vals=pull_rule, context=context)
2458
2459         #create MTS route and pull rules for delivery a specific route MTO to be set on the product
2460         route_name, values = routes_dict[warehouse.delivery_steps]
2461         route_vals = self._get_reception_delivery_route(cr, uid, warehouse, route_name, context=context)
2462         #create the route and its pull rules
2463         delivery_route_id = route_obj.create(cr, uid, route_vals, context=context)
2464         wh_route_ids.append((4, delivery_route_id))
2465         dummy, pull_rules_list = self._get_push_pull_rules(cr, uid, warehouse, True, values, delivery_route_id, context=context)
2466         for pull_rule in pull_rules_list:
2467             pull_obj.create(cr, uid, vals=pull_rule, context=context)
2468         #create MTO pull rule and link it to the generic MTO route
2469         mto_pull_vals = self._get_mto_pull_rule(cr, uid, warehouse, values, context=context)
2470         mto_pull_id = pull_obj.create(cr, uid, mto_pull_vals, context=context)
2471
2472         #create a route for cross dock operations, that can be set on products and product categories
2473         route_name, values = routes_dict['crossdock']
2474         crossdock_route_vals = self._get_crossdock_route(cr, uid, warehouse, route_name, context=context)
2475         crossdock_route_id = route_obj.create(cr, uid, vals=crossdock_route_vals, context=context)
2476         wh_route_ids.append((4, crossdock_route_id))
2477         dummy, pull_rules_list = self._get_push_pull_rules(cr, uid, warehouse, warehouse.delivery_steps != 'ship_only' and warehouse.reception_steps != 'one_step', values, crossdock_route_id, context=context)
2478         for pull_rule in pull_rules_list:
2479             pull_obj.create(cr, uid, vals=pull_rule, context=context)
2480
2481         #create route selectable on the product to resupply the warehouse from another one
2482         self._create_resupply_routes(cr, uid, warehouse, warehouse.resupply_wh_ids, warehouse.default_resupply_wh_id, context=context)
2483
2484         #return routes and mto pull rule to store on the warehouse
2485         return {
2486             'route_ids': wh_route_ids,
2487             'mto_pull_id': mto_pull_id,
2488             'reception_route_id': reception_route_id,
2489             'delivery_route_id': delivery_route_id,
2490             'crossdock_route_id': crossdock_route_id,
2491         }
2492
2493     def change_route(self, cr, uid, ids, warehouse, new_reception_step=False, new_delivery_step=False, context=None):
2494         picking_type_obj = self.pool.get('stock.picking.type')
2495         pull_obj = self.pool.get('procurement.rule')
2496         push_obj = self.pool.get('stock.location.path')
2497         route_obj = self.pool.get('stock.location.route')
2498         new_reception_step = new_reception_step or warehouse.reception_steps
2499         new_delivery_step = new_delivery_step or warehouse.delivery_steps
2500
2501         #change the default source and destination location and (de)activate picking types
2502         input_loc = warehouse.wh_input_stock_loc_id
2503         if new_reception_step == 'one_step':
2504             input_loc = warehouse.lot_stock_id
2505         output_loc = warehouse.wh_output_stock_loc_id
2506         if new_delivery_step == 'ship_only':
2507             output_loc = warehouse.lot_stock_id
2508         picking_type_obj.write(cr, uid, warehouse.in_type_id.id, {'default_location_dest_id': input_loc.id}, context=context)
2509         picking_type_obj.write(cr, uid, warehouse.out_type_id.id, {'default_location_src_id': output_loc.id}, context=context)
2510         picking_type_obj.write(cr, uid, warehouse.int_type_id.id, {'active': new_reception_step != 'one_step'}, context=context)
2511         picking_type_obj.write(cr, uid, warehouse.pick_type_id.id, {'active': new_delivery_step != 'ship_only'}, context=context)
2512         picking_type_obj.write(cr, uid, warehouse.pack_type_id.id, {'active': new_delivery_step == 'pick_pack_ship'}, context=context)
2513
2514         routes_dict = self.get_routes_dict(cr, uid, ids, warehouse, context=context)
2515         #update delivery route and rules: unlink the existing rules of the warehouse delivery route and recreate it
2516         pull_obj.unlink(cr, uid, [pu.id for pu in warehouse.delivery_route_id.pull_ids], context=context)
2517         route_name, values = routes_dict[new_delivery_step]
2518         route_obj.write(cr, uid, warehouse.delivery_route_id.id, {'name': self._format_routename(cr, uid, warehouse, route_name, context=context)}, context=context)
2519         dummy, pull_rules_list = self._get_push_pull_rules(cr, uid, warehouse, True, values, warehouse.delivery_route_id.id, context=context)
2520         #create the pull rules
2521         for pull_rule in pull_rules_list:
2522             pull_obj.create(cr, uid, vals=pull_rule, context=context)
2523
2524         #update reception route and rules: unlink the existing rules of the warehouse reception route and recreate it
2525         pull_obj.unlink(cr, uid, [pu.id for pu in warehouse.reception_route_id.pull_ids], context=context)
2526         push_obj.unlink(cr, uid, [pu.id for pu in warehouse.reception_route_id.push_ids], context=context)
2527         route_name, values = routes_dict[new_reception_step]
2528         route_obj.write(cr, uid, warehouse.reception_route_id.id, {'name': self._format_routename(cr, uid, warehouse, route_name, context=context)}, context=context)
2529         push_rules_list, pull_rules_list = self._get_push_pull_rules(cr, uid, warehouse, True, values, warehouse.reception_route_id.id, context=context)
2530         #create the push/pull rules
2531         for push_rule in push_rules_list:
2532             push_obj.create(cr, uid, vals=push_rule, context=context)
2533         for pull_rule in pull_rules_list:
2534             #all pull rules in reception route are mto, because we don't want to wait for the scheduler to trigger an orderpoint on input location
2535             pull_rule['procure_method'] = 'make_to_order'
2536             pull_obj.create(cr, uid, vals=pull_rule, context=context)
2537
2538         route_obj.write(cr, uid, warehouse.crossdock_route_id.id, {'active': new_reception_step != 'one_step' and new_delivery_step != 'ship_only'}, context=context)
2539
2540         #change MTO rule
2541         dummy, values = routes_dict[new_delivery_step]
2542         mto_pull_vals = self._get_mto_pull_rule(cr, uid, warehouse, values, context=context)
2543         pull_obj.write(cr, uid, warehouse.mto_pull_id.id, mto_pull_vals, context=context)
2544         return True
2545
2546     def create(self, cr, uid, vals, context=None):
2547         if context is None:
2548             context = {}
2549         if vals is None:
2550             vals = {}
2551         data_obj = self.pool.get('ir.model.data')
2552         seq_obj = self.pool.get('ir.sequence')
2553         picking_type_obj = self.pool.get('stock.picking.type')
2554         location_obj = self.pool.get('stock.location')
2555
2556         #create view location for warehouse
2557         wh_loc_id = location_obj.create(cr, uid, {
2558                 'name': _(vals.get('name')),
2559                 'usage': 'view',
2560                 'location_id': data_obj.get_object_reference(cr, uid, 'stock', 'stock_location_locations')[1]
2561             }, context=context)
2562         vals['view_location_id'] = wh_loc_id
2563         #create all location
2564         def_values = self.default_get(cr, uid, {'reception_steps', 'delivery_steps'})
2565         reception_steps = vals.get('reception_steps',  def_values['reception_steps'])
2566         delivery_steps = vals.get('delivery_steps', def_values['delivery_steps'])
2567         context_with_inactive = context.copy()
2568         context_with_inactive['active_test'] = False
2569         sub_locations = [
2570             {'name': _('Stock'), 'active': True, 'field': 'lot_stock_id'},
2571             {'name': _('Input'), 'active': reception_steps != 'one_step', 'field': 'wh_input_stock_loc_id'},
2572             {'name': _('Quality Control'), 'active': reception_steps == 'three_steps', 'field': 'wh_qc_stock_loc_id'},
2573             {'name': _('Output'), 'active': delivery_steps != 'ship_only', 'field': 'wh_output_stock_loc_id'},
2574             {'name': _('Packing Zone'), 'active': delivery_steps == 'pick_pack_ship', 'field': 'wh_pack_stock_loc_id'},
2575         ]
2576         for values in sub_locations:
2577             location_id = location_obj.create(cr, uid, {
2578                 'name': values['name'],
2579                 'usage': 'internal',
2580                 'location_id': wh_loc_id,
2581                 'active': values['active'],
2582             }, context=context_with_inactive)
2583             vals[values['field']] = location_id
2584
2585         #create new sequences
2586         in_seq_id = seq_obj.create(cr, SUPERUSER_ID, values={'name': vals.get('name', '') + _(' Sequence in'), 'prefix': vals.get('code', '') + '\IN\\', 'padding': 5}, context=context)
2587         out_seq_id = seq_obj.create(cr, SUPERUSER_ID, values={'name': vals.get('name', '') + _(' Sequence out'), 'prefix': vals.get('code', '') + '\OUT\\', 'padding': 5}, context=context)
2588         pack_seq_id = seq_obj.create(cr, SUPERUSER_ID, values={'name': vals.get('name', '') + _(' Sequence packing'), 'prefix': vals.get('code', '') + '\PACK\\', 'padding': 5}, context=context)
2589         pick_seq_id = seq_obj.create(cr, SUPERUSER_ID, values={'name': vals.get('name', '') + _(' Sequence picking'), 'prefix': vals.get('code', '') + '\PICK\\', 'padding': 5}, context=context)
2590         int_seq_id = seq_obj.create(cr, SUPERUSER_ID, values={'name': vals.get('name', '') + _(' Sequence internal'), 'prefix': vals.get('code', '') + '\INT\\', 'padding': 5}, context=context)
2591
2592         #create WH
2593         new_id = super(stock_warehouse, self).create(cr, uid, vals=vals, context=context)
2594
2595         warehouse = self.browse(cr, uid, new_id, context=context)
2596         wh_stock_loc = warehouse.lot_stock_id
2597         wh_input_stock_loc = warehouse.wh_input_stock_loc_id
2598         wh_output_stock_loc = warehouse.wh_output_stock_loc_id
2599         wh_pack_stock_loc = warehouse.wh_pack_stock_loc_id
2600
2601         #fetch customer and supplier locations, for references
2602         customer_loc, supplier_loc = self._get_partner_locations(cr, uid, new_id, context=context)
2603
2604         #create in, out, internal picking types for warehouse
2605         input_loc = wh_input_stock_loc
2606         if warehouse.reception_steps == 'one_step':
2607             input_loc = wh_stock_loc
2608         output_loc = wh_output_stock_loc
2609         if warehouse.delivery_steps == 'ship_only':
2610             output_loc = wh_stock_loc
2611
2612         #choose the next available color for the picking types of this warehouse
2613         all_used_colors = self.pool.get('stock.picking.type').search_read(cr, uid, [('warehouse_id', '!=', False), ('color', '!=', False)], ['color'], order='color')
2614         not_used_colors = list(set(range(0, 9)) - set([x['color'] for x in all_used_colors]))
2615         color = 0
2616         if not_used_colors:
2617             color = not_used_colors[0]
2618
2619         in_type_id = picking_type_obj.create(cr, uid, vals={
2620             'name': _('Receptions'),
2621             'warehouse_id': new_id,
2622             'code_id': 'incoming',
2623             'auto_force_assign': True,
2624             'sequence_id': in_seq_id,
2625             'default_location_src_id': supplier_loc.id,
2626             'default_location_dest_id': input_loc.id,
2627             'color': color}, context=context)
2628         out_type_id = picking_type_obj.create(cr, uid, vals={
2629             'name': _('Delivery Orders'),
2630             'warehouse_id': new_id,
2631             'code_id': 'outgoing',
2632             'sequence_id': out_seq_id,
2633             'delivery': True,
2634             'default_location_src_id': output_loc.id,
2635             'default_location_dest_id': customer_loc.id,
2636             'color': color}, context=context)
2637         int_type_id = picking_type_obj.create(cr, uid, vals={
2638             'name': _('Internal Transfers'),
2639             'warehouse_id': new_id,
2640             'code_id': 'internal',
2641             'sequence_id': int_seq_id,
2642             'default_location_src_id': wh_stock_loc.id,
2643             'default_location_dest_id': wh_stock_loc.id,
2644             'active': reception_steps != 'one_step',
2645             'pack': False,
2646             'color': color}, context=context)
2647         pack_type_id = picking_type_obj.create(cr, uid, vals={
2648             'name': _('Pack'),
2649             'warehouse_id': new_id,
2650             'code_id': 'internal',
2651             'sequence_id': pack_seq_id,
2652             'default_location_src_id': wh_pack_stock_loc.id,
2653             'default_location_dest_id': output_loc.id,
2654             'active': delivery_steps == 'pick_pack_ship',
2655             'pack': True,
2656             'color': color}, context=context)
2657         pick_type_id = picking_type_obj.create(cr, uid, vals={
2658             'name': _('Pick'),
2659             'warehouse_id': new_id,
2660             'code_id': 'internal',
2661             'sequence_id': pick_seq_id,
2662             'default_location_src_id': wh_stock_loc.id,
2663             'default_location_dest_id': wh_pack_stock_loc.id,
2664             'active': delivery_steps != 'ship_only',
2665             'pack': False,
2666             'color': color}, context=context)
2667
2668         #write picking types on WH
2669         vals = {
2670             'in_type_id': in_type_id,
2671             'out_type_id': out_type_id,
2672             'pack_type_id': pack_type_id,
2673             'pick_type_id': pick_type_id,
2674             'int_type_id': int_type_id,
2675         }
2676         super(stock_warehouse, self).write(cr, uid, new_id, vals=vals, context=context)
2677         warehouse.refresh()
2678
2679         #create routes and push/pull rules
2680         new_objects_dict = self.create_routes(cr, uid, new_id, warehouse, context=context)
2681         self.write(cr, uid, warehouse.id, new_objects_dict, context=context)
2682         return new_id
2683
2684     def _format_rulename(self, cr, uid, obj, from_loc, dest_loc, context=None):
2685         return obj.code + ': ' + from_loc.name + ' -> ' + dest_loc.name
2686
2687     def _format_routename(self, cr, uid, obj, name, context=None):
2688         return obj.name + ': ' + name
2689
2690     def get_routes_dict(self, cr, uid, ids, warehouse, context=None):
2691         #fetch customer and supplier locations, for references
2692         customer_loc, supplier_loc = self._get_partner_locations(cr, uid, ids, context=context)
2693
2694         return {
2695             'one_step': (_('Reception in 1 step'), []),
2696             'two_steps': (_('Reception in 2 steps'), [(warehouse.wh_input_stock_loc_id, warehouse.lot_stock_id, warehouse.int_type_id.id)]),
2697             'three_steps': (_('Reception in 3 steps'), [(warehouse.wh_input_stock_loc_id, warehouse.wh_qc_stock_loc_id, warehouse.int_type_id.id), (warehouse.wh_qc_stock_loc_id, warehouse.lot_stock_id, warehouse.int_type_id.id)]),
2698             'crossdock': (_('Cross-Dock'), [(warehouse.wh_input_stock_loc_id, warehouse.wh_output_stock_loc_id, warehouse.int_type_id.id), (warehouse.wh_output_stock_loc_id, customer_loc, warehouse.out_type_id.id)]),
2699             'ship_only': (_('Ship Only'), [(warehouse.lot_stock_id, customer_loc, warehouse.out_type_id.id)]),
2700             'pick_ship': (_('Pick + Ship'), [(warehouse.lot_stock_id, warehouse.wh_output_stock_loc_id, warehouse.pick_type_id.id), (warehouse.wh_output_stock_loc_id, customer_loc, warehouse.out_type_id.id)]),
2701             'pick_pack_ship': (_('Pick + Pack + Ship'), [(warehouse.lot_stock_id, warehouse.wh_pack_stock_loc_id, warehouse.int_type_id.id), (warehouse.wh_pack_stock_loc_id, warehouse.wh_output_stock_loc_id, warehouse.pack_type_id.id), (warehouse.wh_output_stock_loc_id, customer_loc, warehouse.out_type_id.id)]),
2702         }
2703
2704     def _handle_renaming(self, cr, uid, warehouse, name, context=None):
2705         location_obj = self.pool.get('stock.location')
2706         route_obj = self.pool.get('stock.location.route')
2707         pull_obj = self.pool.get('procurement.rule')
2708         push_obj = self.pool.get('stock.location.path')
2709         #rename location
2710         location_id = warehouse.lot_stock_id.location_id.id
2711         location_obj.write(cr, uid, location_id, {'name': name}, context=context)
2712         #rename route and push-pull rules
2713         for route in warehouse.route_ids:
2714             route_obj.write(cr, uid, route.id, {'name': route.name.replace(warehouse.name, name, 1)}, context=context)
2715             for pull in route.pull_ids:
2716                 pull_obj.write(cr, uid, pull.id, {'name': pull.name.replace(warehouse.name, name, 1)}, context=context)
2717             for push in route.push_ids:
2718                 push_obj.write(cr, uid, push.id, {'name': pull.name.replace(warehouse.name, name, 1)}, context=context)
2719         #change the mto pull rule name
2720         pull_obj.write(cr, uid, warehouse.mto_pull_id.id, {'name': warehouse.mto_pull_id.name.replace(warehouse.name, name, 1)}, context=context)
2721
2722     def write(self, cr, uid, ids, vals, context=None):
2723         if context is None:
2724             context = {}
2725         if isinstance(ids, (int, long)):
2726             ids = [ids]
2727         seq_obj = self.pool.get('ir.sequence')
2728         route_obj = self.pool.get('stock.location.route')
2729         warehouse_obj = self.pool.get('stock.warehouse')
2730
2731         context_with_inactive = context.copy()
2732         context_with_inactive['active_test'] = False
2733         for warehouse in self.browse(cr, uid, ids, context=context_with_inactive):
2734             #first of all, check if we need to delete and recreate route
2735             if vals.get('reception_steps') or vals.get('delivery_steps'):
2736                 #activate and deactivate location according to reception and delivery option
2737                 self.switch_location(cr, uid, warehouse.id, warehouse, vals.get('reception_steps', False), vals.get('delivery_steps', False), context=context)
2738                 # switch between route
2739                 self.change_route(cr, uid, ids, warehouse, vals.get('reception_steps', False), vals.get('delivery_steps', False), context=context_with_inactive)
2740             if vals.get('code') or vals.get('name'):
2741                 name = warehouse.name
2742                 #rename sequence
2743                 if vals.get('name'):
2744                     name = vals.get('name')
2745                     self._handle_renaming(cr, uid, warehouse, name, context=context_with_inactive)
2746                 seq_obj.write(cr, uid, warehouse.in_type_id.sequence_id.id, {'name': name + _(' Sequence in'), 'prefix': vals.get('code', warehouse.code) + '\IN\\'}, context=context)
2747                 seq_obj.write(cr, uid, warehouse.out_type_id.sequence_id.id, {'name': name + _(' Sequence out'), 'prefix': vals.get('code', warehouse.code) + '\OUT\\'}, context=context)
2748                 seq_obj.write(cr, uid, warehouse.pack_type_id.sequence_id.id, {'name': name + _(' Sequence packing'), 'prefix': vals.get('code', warehouse.code) + '\PACK\\'}, context=context)
2749                 seq_obj.write(cr, uid, warehouse.pick_type_id.sequence_id.id, {'name': name + _(' Sequence picking'), 'prefix': vals.get('code', warehouse.code) + '\PICK\\'}, context=context)
2750                 seq_obj.write(cr, uid, warehouse.int_type_id.sequence_id.id, {'name': name + _(' Sequence internal'), 'prefix': vals.get('code', warehouse.code) + '\INT\\'}, context=context)
2751         if vals.get('resupply_wh_ids') and not vals.get('resupply_route_ids'):
2752             for cmd in vals.get('resupply_wh_ids'):
2753                 if cmd[0] == 6:
2754                     new_ids = set(cmd[2])
2755                     old_ids = set([wh.id for wh in warehouse.resupply_wh_ids])
2756                     to_add_wh_ids = new_ids - old_ids
2757                     if to_add_wh_ids:
2758                         supplier_warehouses = warehouse_obj.browse(cr, uid, list(to_add_wh_ids), context=context)
2759                         self._create_resupply_routes(cr, uid, warehouse, supplier_warehouses, warehouse.default_resupply_wh_id, context=context)
2760                     to_remove_wh_ids = old_ids - new_ids
2761                     if to_remove_wh_ids:
2762                         to_remove_route_ids = route_obj.search(cr, uid, [('supplied_wh_id', '=', warehouse.id), ('supplier_wh_id', 'in', list(to_remove_wh_ids))], context=context)
2763                         if to_remove_route_ids:
2764                             route_obj.unlink(cr, uid, to_remove_route_ids, context=context)
2765                 else:
2766                     #not implemented
2767                     pass
2768         if 'default_resupply_wh_id' in vals:
2769             if warehouse.default_resupply_wh_id:
2770                 #remove the existing resupplying route on all products
2771                 to_remove_route_ids = route_obj.search(cr, uid, [('supplied_wh_id', '=', warehouse.id), ('supplier_wh_id', '=', warehouse.default_resupply_wh_id.id)], context=context)
2772                 for inter_wh_route_id in to_remove_route_ids:
2773                     self._unassign_route_on_products(cr, uid, warehouse, inter_wh_route_id, context=context)
2774             if vals.get('default_resupply_wh_id'):
2775                 #assign the new resupplying route on all products
2776                 to_assign_route_ids = route_obj.search(cr, uid, [('supplied_wh_id', '=', warehouse.id), ('supplier_wh_id', '=', vals.get('default_resupply_wh_id'))], context=context)
2777                 for inter_wh_route_id in to_assign_route_ids:
2778                     self._assign_route_on_products(cr, uid, warehouse, inter_wh_route_id, context=context)
2779
2780         return super(stock_warehouse, self).write(cr, uid, ids, vals=vals, context=context)
2781
2782     def unlink(self, cr, uid, ids, context=None):
2783         #TODO try to delete location and route and if not possible, put them in inactive
2784         return super(stock_warehouse, self).unlink(cr, uid, ids, context=context)
2785
2786     def get_all_routes_for_wh(self, cr, uid, warehouse, context=None):
2787         all_routes = [route.id for route in warehouse.route_ids]
2788         all_routes += [warehouse.mto_pull_id.route_id.id]
2789         return all_routes
2790
2791     def view_all_routes_for_wh(self, cr, uid, ids, context=None):
2792         all_routes = []
2793         for wh in self.browse(cr, uid, ids, context=context):
2794             all_routes += self.get_all_routes_for_wh(cr, uid, wh, context=context)
2795
2796         domain = [('id', 'in', all_routes)]
2797         return {
2798             'name': _('Warehouse\'s Routes'),
2799             'domain': domain,
2800             'res_model': 'stock.location.route',
2801             'type': 'ir.actions.act_window',
2802             'view_id': False,
2803             'view_mode': 'tree,form',
2804             'view_type': 'form',
2805             'limit': 20
2806         }
2807
2808 class stock_location_path(osv.osv):
2809     _name = "stock.location.path"
2810     _description = "Pushed Flows"
2811     _order = "name"
2812
2813     def _get_route(self, cr, uid, ids, context=None):
2814         #WARNING TODO route_id is not required, so a field related seems a bad idea >-< 
2815         if context is None:
2816             context = {}
2817         result = {}
2818         if context is None:
2819             context = {}
2820         context_with_inactive = context.copy()
2821         context_with_inactive['active_test'] = False
2822         for route in self.pool.get('stock.location.route').browse(cr, uid, ids, context=context_with_inactive):
2823             for push_rule in route.push_ids:
2824                 result[push_rule.id] = True
2825         return result.keys()
2826
2827     _columns = {
2828         'name': fields.char('Operation Name', size=64, required=True),
2829         'company_id': fields.many2one('res.company', 'Company'),
2830         'route_id': fields.many2one('stock.location.route', 'Route'),
2831         'location_from_id': fields.many2one('stock.location', 'Source Location', ondelete='cascade', select=1, required=True),
2832         'location_dest_id': fields.many2one('stock.location', 'Destination Location', ondelete='cascade', select=1, required=True),
2833         'delay': fields.integer('Delay (days)', help="Number of days to do this transition"),
2834         'invoice_state': fields.selection([
2835             ("invoiced", "Invoiced"),
2836             ("2binvoiced", "To Be Invoiced"),
2837             ("none", "Not Applicable")], "Invoice Status",
2838             required=True,), 
2839         'picking_type_id': fields.many2one('stock.picking.type', 'Type of the new Operation', required=True, help="This is the picking type associated with the different pickings"), 
2840         'auto': fields.selection(
2841             [('auto','Automatic Move'), ('manual','Manual Operation'),('transparent','Automatic No Step Added')],
2842             'Automatic Move',
2843             required=True, select=1,
2844             help="This is used to define paths the product has to follow within the location tree.\n" \
2845                 "The 'Automatic Move' value will create a stock move after the current one that will be "\
2846                 "validated automatically. With 'Manual Operation', the stock move has to be validated "\
2847                 "by a worker. With 'Automatic No Step Added', the location is replaced in the original move."
2848             ),
2849         'propagate': fields.boolean('Propagate cancel and split', help='If checked, when the previous move is cancelled or split, the move generated by this move will too'),
2850         'active': fields.related('route_id', 'active', type='boolean', string='Active', store={
2851                     'stock.location.route': (_get_route, ['active'], 20),
2852                     'stock.location.path': (lambda self, cr, uid, ids, c={}: ids, ['route_id'], 20),},
2853                 help="If the active field is set to False, it will allow you to hide the rule without removing it." ),
2854         'warehouse_id': fields.many2one('stock.warehouse', 'Warehouse'),
2855     }
2856     _defaults = {
2857         'auto': 'auto',
2858         'delay': 1,
2859         'invoice_state': 'none',
2860         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'procurement.order', context=c),
2861         'propagate': True,
2862         'active': True,
2863     }
2864     def _apply(self, cr, uid, rule, move, context=None):
2865         move_obj = self.pool.get('stock.move')
2866         newdate = (datetime.strptime(move.date, '%Y-%m-%d %H:%M:%S') + relativedelta.relativedelta(days=rule.delay or 0)).strftime('%Y-%m-%d')
2867         if rule.auto=='transparent':
2868             move_obj.write(cr, uid, [move.id], {
2869                 'date': newdate,
2870                 'location_dest_id': rule.location_dest_id.id
2871             })
2872             if rule.location_dest_id.id<>move.location_dest_id.id:
2873                 move_obj._push_apply(self, cr, uid, move.id, context)
2874             return move.id
2875         else:
2876             move_id = move_obj.copy(cr, uid, move.id, {
2877                 'location_id': move.location_dest_id.id,
2878                 'location_dest_id': rule.location_dest_id.id,
2879                 'date': datetime.now().strftime('%Y-%m-%d'),
2880                 'company_id': rule.company_id and rule.company_id.id or False,
2881                 'date_expected': newdate,
2882                 'picking_id': False,
2883                 'picking_type_id': rule.picking_type_id and rule.picking_type_id.id or False,
2884                 'rule_id': rule.id,
2885                 'propagate': rule.propagate, 
2886                 'warehouse_id': rule.warehouse_id and rule.warehouse_id.id or False,
2887             })
2888             move_obj.write(cr, uid, [move.id], {
2889                 'move_dest_id': move_id,
2890             })
2891             move_obj.action_confirm(cr, uid, [move_id], context=None)
2892             return move_id
2893
2894 class stock_move_putaway(osv.osv):
2895     _name = 'stock.move.putaway'
2896     _description = 'Proposed Destination'
2897     _columns = {
2898         'move_id': fields.many2one('stock.move', required=True),
2899         'location_id': fields.many2one('stock.location', 'Location', required=True),
2900         'lot_id': fields.many2one('stock.production.lot', 'Lot'),
2901         'quantity': fields.float('Quantity', required=True),
2902     }
2903
2904
2905
2906 # -------------------------
2907 # Packaging related stuff
2908 # -------------------------
2909
2910 from openerp.report import report_sxw
2911 report_sxw.report_sxw('report.stock.quant.package.barcode', 'stock.quant.package', 'addons/stock/report/picking_barcode.rml')
2912
2913 class stock_package(osv.osv):
2914     """
2915     These are the packages, containing quants and/or other packages
2916     """
2917     _name = "stock.quant.package"
2918     _description = "Physical Packages"
2919     _parent_name = "parent_id"
2920     _parent_store = True
2921     _parent_order = 'name'
2922     _order = 'parent_left'
2923
2924     def name_get(self, cr, uid, ids, context=None):
2925         res = self._complete_name(cr, uid, ids, 'complete_name', None, context=context)
2926         return res.items()
2927
2928     def _complete_name(self, cr, uid, ids, name, args, context=None):
2929         """ Forms complete name of location from parent location to child location.
2930         @return: Dictionary of values
2931         """
2932         res = {}
2933         for m in self.browse(cr, uid, ids, context=context):
2934             res[m.id] = m.name
2935             parent = m.parent_id
2936             while parent:
2937                 res[m.id] = parent.name + ' / ' + res[m.id]
2938                 parent = parent.parent_id
2939         return res
2940
2941     def _get_packages(self, cr, uid, ids, context=None):
2942         """Returns packages from quants for store"""
2943         res = set()
2944         for quant in self.browse(cr, uid, ids, context=context):
2945             if quant.package_id:
2946                 res.add(quant.package_id.id)
2947         return list(res)
2948
2949     def _get_packages_to_relocate(self, cr, uid, ids, context=None):
2950         res = set()
2951         for pack in self.browse(cr, uid, ids, context=context):
2952             res.add(pack.id)
2953             if pack.parent_id:
2954                 res.add(pack.parent_id.id)
2955         return list(res)
2956
2957     # TODO: Problem when package is empty!
2958     #
2959     def _get_package_info(self, cr, uid, ids, name, args, context=None):
2960         default_company_id = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id.id
2961         res = {}.fromkeys(ids, {'location_id': False, 'company_id': default_company_id})
2962         for pack in self.browse(cr, uid, ids, context=context):
2963             if pack.quant_ids:
2964                 res[pack.id]['location_id'] = pack.quant_ids[0].location_id.id
2965                 res[pack.id]['owner_id'] = pack.quant_ids[0].owner_id and pack.quant_ids[0].owner_id.id or False
2966                 res[pack.id]['company_id'] = pack.quant_ids[0].company_id.id
2967             elif pack.children_ids:
2968                 res[pack.id]['location_id'] = pack.children_ids[0].location_id and pack.children_ids[0].location_id.id or False
2969                 res[pack.id]['owner_id'] = pack.children_ids[0].owner_id and pack.children_ids[0].owner_id.id or False
2970                 res[pack.id]['company_id'] = pack.children_ids[0].company_id and pack.children_ids[0].company_id.id or False
2971         return res
2972
2973     _columns = {
2974         'name': fields.char('Package Reference', size=64, select=True),
2975         'complete_name': fields.function(_complete_name, type='char', string="Package Name",),
2976         'parent_left': fields.integer('Left Parent', select=1),
2977         'parent_right': fields.integer('Right Parent', select=1),
2978         'packaging_id': fields.many2one('product.packaging', 'Type of Packaging'),
2979         'location_id': fields.function(_get_package_info, type='many2one', relation='stock.location', string='Location', multi="package",
2980                                     store={
2981                                        'stock.quant': (_get_packages, ['location_id'], 10),
2982                                        'stock.quant.package': (_get_packages_to_relocate, ['quant_ids', 'children_ids', 'parent_id'], 10),
2983                                     }, readonly=True),
2984         'quant_ids': fields.one2many('stock.quant', 'package_id', 'Bulk Content'),
2985         'parent_id': fields.many2one('stock.quant.package', 'Parent Package', help="The package containing this item", ondelete='restrict'),
2986         'children_ids': fields.one2many('stock.quant.package', 'parent_id', 'Contained Packages'),
2987         'company_id': fields.function(_get_package_info, type="many2one", relation='res.company', string='Company', multi="package", 
2988                                     store={
2989                                        'stock.quant': (_get_packages, ['company_id'], 10),
2990                                        'stock.quant.package': (_get_packages_to_relocate, ['quant_ids', 'children_ids', 'parent_id'], 10),
2991                                     }, readonly=True),
2992         'owner_id': fields.function(_get_package_info, type='many2one', relation='res.partner', string='Owner', multi="package",
2993                                 store={
2994                                        'stock.quant': (_get_packages, ['owner_id'], 10),
2995                                        'stock.quant.package': (_get_packages_to_relocate, ['quant_ids', 'children_ids', 'parent_id'], 10),
2996                                     }, readonly=True),
2997     }
2998     _defaults = {
2999         'name': lambda self, cr, uid, context: self.pool.get('ir.sequence').get(cr, uid, 'stock.quant.package') or _('Unknown Pack')
3000     }
3001     def _check_location(self, cr, uid, ids, context=None):
3002         '''checks that all quants in a package are stored in the same location'''
3003         quant_obj = self.pool.get('stock.quant')
3004         for pack in self.browse(cr, uid, ids, context=context):
3005             parent = pack
3006             while parent.parent_id:
3007                 parent = parent.parent_id
3008             quant_ids = self.get_content(cr, uid, [parent.id], context=context)
3009             quants = quant_obj.browse(cr, uid, quant_ids, context=context)
3010             location_id = quants and quants[0].location_id.id or False
3011             if not all([quant.location_id.id == location_id for quant in quants]):
3012                 return False
3013         return True
3014
3015     _constraints = [
3016         (_check_location, 'Everything inside a package should be in the same location', ['location_id']),
3017     ]
3018
3019     def action_print(self, cr, uid, ids, context=None):
3020         if context is None:
3021             context = {}
3022         datas = {
3023             'ids': context.get('active_id') and [context.get('active_id')] or ids,
3024             'model': 'stock.quant.package',
3025             'form': self.read(cr, uid, ids)[0]
3026         }
3027         return {
3028             'type': 'ir.actions.report.xml',
3029             'report_name': 'stock.quant.package.barcode',
3030             'datas': datas
3031         }
3032
3033     def unpack(self, cr, uid, ids, context=None):
3034         quant_obj = self.pool.get('stock.quant')
3035         for package in self.browse(cr, uid, ids, context=context):
3036             quant_ids = [quant.id for quant in package.quant_ids]
3037             quant_obj.write(cr, uid, quant_ids, {'package_id': package.parent_id.id or False}, context=context)
3038             children_package_ids = [child_package.id for child_package in package.children_ids]
3039             self.write(cr, uid, children_package_ids, {'parent_id': package.parent_id.id or False}, context=context)
3040         #delete current package since it contains nothing anymore
3041         self.unlink(cr, uid, ids, context=context)
3042         return self.pool.get('ir.actions.act_window').for_xml_id(cr, uid, 'stock', 'action_package_view', context=context)
3043
3044     def get_content(self, cr, uid, ids, context=None):
3045         child_package_ids = self.search(cr, uid, [('id', 'child_of', ids)], context=context)
3046         return self.pool.get('stock.quant').search(cr, uid, [('package_id', 'in', child_package_ids)], context=context)
3047
3048     def get_content_package(self, cr, uid, ids, context=None):
3049         quants_ids = self.get_content(cr, uid, ids, context=context)
3050         res = self.pool.get('ir.actions.act_window').for_xml_id(cr, uid, 'stock', 'quantsact', context=context)
3051         res['domain'] = [('id', 'in', quants_ids)]
3052         return res
3053
3054     def _get_product_total_qty(self, cr, uid, package_record, product_id, context=None):
3055         ''' find the total of given product 'product_id' inside the given package 'package_id'''
3056         quant_obj = self.pool.get('stock.quant')
3057         all_quant_ids = self.get_content(cr, uid, [package_record.id], context=context)
3058         total = 0
3059         for quant in quant_obj.browse(cr, uid, all_quant_ids, context=context):
3060             if quant.product_id.id == product_id:
3061                 total += quant.qty
3062         return total
3063
3064     def _get_all_products_quantities(self, cr, uid, package_id, context=None):
3065         '''This function computes the different product quantities for the given package
3066         '''
3067         quant_obj = self.pool.get('stock.quant')
3068         res = {}
3069         for quant in quant_obj.browse(cr, uid, self.get_content(cr, uid, package_id, context=context)):
3070             if quant.product_id.id not in res:
3071                 res[quant.product_id.id] = 0
3072             res[quant.product_id.id] += quant.qty
3073         return res
3074
3075 class stock_pack_operation(osv.osv):
3076     _name = "stock.pack.operation"
3077     _description = "Packing Operation"
3078
3079     def _get_remaining_prod_quantities(self, cr, uid, operation, context=None):
3080         '''Get the remaining quantities per product on an operation with a package. This function returns a dictionary'''
3081         #if the operation doesn't concern a package, it's not relevant to call this function
3082         if not operation.package_id or operation.product_id:
3083             return {operation.product_id.id: operation.remaining_qty}
3084         #get the total of products the package contains
3085         res = self.pool.get('stock.quant.package')._get_all_products_quantities(cr, uid, operation.package_id.id, context=context)
3086         #reduce by the quantities linked to a move
3087         for record in operation.linked_move_operation_ids:
3088             if record.move_id.product_id.id not in res:
3089                 res[record.move_id.product_id.id] = 0
3090             res[record.move_id.product_id.id] -= record.qty
3091         return res
3092
3093     def _get_remaining_qty(self, cr, uid, ids, name, args, context=None):
3094         uom_obj = self.pool.get('product.uom')
3095         res = {}
3096         for ops in self.browse(cr, uid, ids, context=context):
3097             res[ops.id] = 0
3098             if ops.package_id:
3099                 #dont try to compute the remaining quantity for packages because it's not relevant (a package could include different products).
3100                 #should use _get_remaining_prod_quantities instead
3101                 continue
3102             elif ops.product_id:
3103                 qty = ops.product_qty
3104                 if ops.product_uom_id:
3105                     qty = uom_obj._compute_qty(cr, uid, ops.product_uom_id.id, ops.product_qty, ops.product_id.uom_id.id)
3106                 for record in ops.linked_move_operation_ids:
3107                     qty -= record.qty
3108                 #converting the remaining quantity in the pack operation UoM
3109                 if ops.product_uom_id:
3110                     qty = uom_obj._compute_qty(cr, uid, ops.product_id.uom_id.id, qty, ops.product_uom_id.id)
3111                 res[ops.id] = qty
3112         return res
3113
3114     def product_id_change(self, cr, uid, ids, product_id, product_uom_id, product_qty, context=None):
3115         res = self.on_change_tests(cr, uid, ids, product_id, product_uom_id, product_qty, context=context)
3116         if product_id and not product_uom_id:
3117             product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
3118             res['value']['product_uom_id'] = product.uom_id.id
3119         return res
3120
3121     def on_change_tests(self, cr, uid, ids, product_id, product_uom_id, product_qty, context=None):
3122         res = {'value': {}}
3123         uom_obj = self.pool.get('product.uom')
3124         if product_id:
3125             product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
3126             product_uom_id = product_uom_id or product.uom_id.id
3127             selected_uom = uom_obj.browse(cr, uid, product_uom_id, context=context)
3128             if selected_uom.category_id.id != product.uom_id.category_id.id:
3129                 res['warning'] = {
3130                     'title': _('Warning: wrong UoM!'),
3131                     'message': _('The selected UoM for product %s is not compatible with the UoM set on the product form. \nPlease choose an UoM within the same UoM category.') % (product.name)
3132                 }
3133             if product_qty and 'warning' not in res:
3134                 rounded_qty = uom_obj._compute_qty(cr, uid, product_uom_id, product_qty, product_uom_id, round=True)
3135                 if rounded_qty != product_qty:
3136                     res['warning'] = {
3137                         'title': _('Warning: wrong quantity!'),
3138                         'message': _('The chosen quantity for product %s is not compatible with the UoM rounding. It will be automatically converted at confirmation') % (product.name)
3139                     }
3140         return res
3141
3142     _columns = {
3143         'picking_id': fields.many2one('stock.picking', 'Stock Picking', help='The stock operation where the packing has been made', required=True),
3144         'product_id': fields.many2one('product.product', 'Product', ondelete="CASCADE"),  # 1
3145         'product_uom_id': fields.many2one('product.uom', 'Product Unit of Measure'),
3146         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product Unit of Measure'), required=True),
3147         'package_id': fields.many2one('stock.quant.package', 'Package'),  # 2
3148         'quant_id': fields.many2one('stock.quant', 'Quant'),  # 3
3149         'lot_id': fields.many2one('stock.production.lot', 'Lot/Serial Number'),
3150         'result_package_id': fields.many2one('stock.quant.package', 'Container Package', help="If set, the operations are packed into this package", required=False, ondelete='cascade'),
3151         'date': fields.datetime('Date', required=True),
3152         'owner_id': fields.many2one('res.partner', 'Owner', help="Owner of the quants"),
3153         #'update_cost': fields.boolean('Need cost update'),
3154         'cost': fields.float("Cost", help="Unit Cost for this product line"),
3155         'currency': fields.many2one('res.currency', string="Currency", help="Currency in which Unit cost is expressed", ondelete='CASCADE'),
3156         'linked_move_operation_ids': fields.one2many('stock.move.operation.link', 'operation_id', string='Linked Moves', readonly=True, help='Moves impacted by this operation for the computation of the remaining quantities'),
3157         'remaining_qty': fields.function(_get_remaining_qty, type='float', string='Remaining Qty'),
3158     }
3159
3160     _defaults = {
3161         'date': fields.date.context_today,
3162     }
3163
3164     def process_packaging(self, cr, uid, operation, quants, context=None):
3165         ''' Process the packaging of a given operation, after the quants have been moved. If there was not enough quants found
3166         a quant already has been with the good package information so we don't consider that case in this method'''
3167         quant_obj = self.pool.get("stock.quant")
3168         pack_obj = self.pool.get("stock.quant.package")
3169         for quant, qty in quants:
3170             if quant:
3171                 if operation.product_id:
3172                     #if a product + a package information is given, we consider that we took a part of an existing package (unpacking)
3173                     quant_obj.write(cr, uid, quant.id, {'package_id': operation.result_package_id.id}, context=context)
3174                 elif operation.package_id and operation.result_package_id:
3175                     #move the whole pack into the final package if any
3176                     pack_obj.write(cr, uid, [operation.package_id.id], {'parent_id': operation.result_package_id.id}, context=context)
3177
3178
3179
3180
3181     #TODO: this function can be refactored
3182     def _search_and_increment(self, cr, uid, picking_id, key, context=None):
3183         '''Search for an operation on an existing key in a picking, if it exists increment the qty (+1) otherwise create it
3184
3185         :param key: tuple directly reusable in a domain
3186         context can receive a key 'current_package_id' with the package to consider for this operation
3187         returns True
3188
3189         previously: returns the update to do in stock.move one2many field of picking (adapt remaining quantities) and to the list of package in the classic one2many syntax
3190                  (0, 0,  { values })    link to a new record that needs to be created with the given values dictionary
3191                  (1, ID, { values })    update the linked record with id = ID (write *values* on it)
3192                  (2, ID)                remove and delete the linked record with id = ID (calls unlink on ID, that will delete the object completely, and the link to it as well)
3193         '''
3194         quant_obj = self.pool.get('stock.quant')
3195         if context is None:
3196             context = {}
3197
3198         #if current_package_id is given in the context, we increase the number of items in this package
3199         package_clause = [('result_package_id', '=', context.get('current_package_id', False))]
3200         existing_operation_ids = self.search(cr, uid, [('picking_id', '=', picking_id), key] + package_clause, context=context)
3201         if existing_operation_ids:
3202             #existing operation found for the given key and picking => increment its quantity
3203             operation_id = existing_operation_ids[0]
3204             qty = self.browse(cr, uid, operation_id, context=context).product_qty + 1
3205             self.write(cr, uid, operation_id, {'product_qty': qty}, context=context)
3206         else:
3207             #no existing operation found for the given key and picking => create a new one
3208             var_name, dummy, value = key
3209             uom_id = False
3210             if var_name == 'product_id':
3211                 uom_id = self.pool.get('product.product').browse(cr, uid, value, context=context).uom_id.id
3212             elif var_name == 'quant_id':
3213                 quant = quant_obj.browse(cr, uid, value, context=context)
3214                 uom_id = quant.product_id.uom_id.id
3215             values = {
3216                 'picking_id': picking_id,
3217                 var_name: value,
3218                 'product_qty': 1,
3219                 'product_uom_id': uom_id,
3220             }
3221             operation_id = self.create(cr, uid, values, context=context)
3222             values.update({'id': operation_id})
3223         return True
3224
3225
3226 class stock_move_operation_link(osv.osv):
3227     """
3228     Table making the link between stock.moves and stock.pack.operations to compute the remaining quantities on each of these objects
3229     """
3230     _name = "stock.move.operation.link"
3231     _description = "Link between stock moves and pack operations"
3232
3233     _columns = {
3234         'qty': fields.float('Quantity', help="Quantity of products to consider when talking about the contribution of this pack operation towards the remaining quantity of the move (and inverse). Given in the product main uom."),
3235         'operation_id': fields.many2one('stock.pack.operation', 'Operation', required=True, ondelete="cascade"),
3236         'move_id': fields.many2one('stock.move', 'Move', required=True, ondelete="cascade"),
3237         'reserved_quant_ids': fields.one2many('stock.quant', 'link_move_operation_id', 'Reserved quants'),
3238     }
3239
3240     def get_specific_domain(self, cr, uid, record, context=None):
3241         '''Returns the specific domain to consider for quant selection in action_assign() or action_done() of stock.move,
3242         having the record given as parameter making the link between the stock move and a pack operation'''
3243         package_obj = self.pool.get('stock.quant.package')
3244
3245         op = record.operation_id
3246         domain = []
3247         if op.package_id:
3248             domain.append(('id', 'in', package_obj.get_content(cr, uid, [op.package_id.id], context=context)))
3249         if op.quant_id:
3250             domain.append(('id', '=', op.quant_id.id))
3251         if op.lot_id:
3252             domain.append(('lot_id', '=', op.lot_id.id))
3253         if op.owner_id:
3254             domain.append(('owner_id', '=', op.owner_id.id))
3255         else:
3256             domain.append(('owner_id', '=', False))
3257         return domain
3258
3259 class stock_warehouse_orderpoint(osv.osv):
3260     """
3261     Defines Minimum stock rules.
3262     """
3263     _name = "stock.warehouse.orderpoint"
3264     _description = "Minimum Inventory Rule"
3265
3266     def get_draft_procurements(self, cr, uid, ids, context=None):
3267         if context is None:
3268             context = {}
3269         result = {}
3270         if not isinstance(ids, list):
3271             ids = [ids]
3272         procurement_obj = self.pool.get('procurement.order')
3273         for orderpoint in self.browse(cr, uid, ids, context=context):
3274             procurement_ids = procurement_obj.search(cr, uid, [('state', 'not in', ('cancel', 'done')), ('product_id', '=', orderpoint.product_id.id), ('location_id', '=', orderpoint.location_id.id)], context=context)            
3275         return list(set(procurement_ids))
3276
3277     def _check_product_uom(self, cr, uid, ids, context=None):
3278         '''
3279         Check if the UoM has the same category as the product standard UoM
3280         '''
3281         if not context:
3282             context = {}
3283
3284         for rule in self.browse(cr, uid, ids, context=context):
3285             if rule.product_id.uom_id.category_id.id != rule.product_uom.category_id.id:
3286                 return False
3287
3288         return True
3289
3290     def action_view_proc_to_process(self, cr, uid, ids, context=None):        
3291         act_obj = self.pool.get('ir.actions.act_window')
3292         mod_obj = self.pool.get('ir.model.data')
3293         draft_ids = self.get_draft_procurements(cr, uid, ids, context=context)
3294         result = mod_obj.get_object_reference(cr, uid, 'procurement', 'do_view_procurements')
3295         if not result:
3296             return False
3297  
3298         result = act_obj.read(cr, uid, [result[1]], context=context)[0]
3299         result['domain'] = "[('id', 'in', [" + ','.join(map(str, draft_ids)) + "])]"
3300         return result
3301
3302     _columns = {
3303         'name': fields.char('Name', size=32, required=True),
3304         'active': fields.boolean('Active', help="If the active field is set to False, it will allow you to hide the orderpoint without removing it."),
3305         'logic': fields.selection([('max', 'Order to Max'), ('price', 'Best price (not yet active!)')], 'Reordering Mode', required=True),
3306         'warehouse_id': fields.many2one('stock.warehouse', 'Warehouse', required=True, ondelete="cascade"),
3307         'location_id': fields.many2one('stock.location', 'Location', required=True, ondelete="cascade"),
3308         'product_id': fields.many2one('product.product', 'Product', required=True, ondelete='cascade', domain=[('type', '!=', 'service')]),
3309         'product_uom': fields.many2one('product.uom', 'Product Unit of Measure', required=True),
3310         'product_min_qty': fields.float('Minimum Quantity', required=True,
3311             help="When the virtual stock goes below the Min Quantity specified for this field, OpenERP generates "\
3312             "a procurement to bring the forecasted quantity to the Max Quantity."),
3313         'product_max_qty': fields.float('Maximum Quantity', required=True,
3314             help="When the virtual stock goes below the Min Quantity, OpenERP generates "\
3315             "a procurement to bring the forecasted quantity to the Quantity specified as Max Quantity."),
3316         'qty_multiple': fields.integer('Qty Multiple', required=True,
3317             help="The procurement quantity will be rounded up to this multiple."),
3318         'procurement_id': fields.many2one('procurement.order', 'Latest procurement', ondelete="set null"),
3319         'company_id': fields.many2one('res.company', 'Company', required=True)        
3320     }
3321     _defaults = {
3322         'active': lambda *a: 1,
3323         'logic': lambda *a: 'max',
3324         'qty_multiple': lambda *a: 1,
3325         'name': lambda self, cr, uid, context: self.pool.get('ir.sequence').get(cr, uid, 'stock.orderpoint') or '',
3326         'product_uom': lambda self, cr, uid, context: context.get('product_uom', False),
3327         'company_id': lambda self, cr, uid, context: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.warehouse.orderpoint', context=context)
3328     }
3329     _sql_constraints = [
3330         ('qty_multiple_check', 'CHECK( qty_multiple > 0 )', 'Qty Multiple must be greater than zero.'),
3331     ]
3332     _constraints = [
3333         (_check_product_uom, 'You have to select a product unit of measure in the same category than the default unit of measure of the product', ['product_id', 'product_uom']),
3334     ]
3335
3336     def default_get(self, cr, uid, fields, context=None):
3337         res = super(stock_warehouse_orderpoint, self).default_get(cr, uid, fields, context)
3338         # default 'warehouse_id' and 'location_id'
3339         if 'warehouse_id' not in res:
3340             warehouse = self.pool.get('ir.model.data').get_object(cr, uid, 'stock', 'warehouse0', context)
3341             res['warehouse_id'] = warehouse.id
3342         if 'location_id' not in res:
3343             warehouse = self.pool.get('stock.warehouse').browse(cr, uid, res['warehouse_id'], context)
3344             res['location_id'] = warehouse.lot_stock_id.id
3345         return res
3346
3347     def onchange_warehouse_id(self, cr, uid, ids, warehouse_id, context=None):
3348         """ Finds location id for changed warehouse.
3349         @param warehouse_id: Changed id of warehouse.
3350         @return: Dictionary of values.
3351         """
3352         if warehouse_id:
3353             w = self.pool.get('stock.warehouse').browse(cr, uid, warehouse_id, context=context)
3354             v = {'location_id': w.lot_stock_id.id}
3355             return {'value': v}
3356         return {}
3357
3358     def onchange_product_id(self, cr, uid, ids, product_id, context=None):
3359         """ Finds UoM for changed product.
3360         @param product_id: Changed id of product.
3361         @return: Dictionary of values.
3362         """
3363         if product_id:
3364             prod = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
3365             d = {'product_uom': [('category_id', '=', prod.uom_id.category_id.id)]}
3366             v = {'product_uom': prod.uom_id.id}
3367             return {'value': v, 'domain': d}
3368         return {'domain': {'product_uom': []}}
3369
3370     def copy(self, cr, uid, id, default=None, context=None):
3371         if not default:
3372             default = {}
3373         default.update({
3374             'name': self.pool.get('ir.sequence').get(cr, uid, 'stock.orderpoint') or '',
3375         })
3376         return super(stock_warehouse_orderpoint, self).copy(cr, uid, id, default, context=context)
3377
3378
3379 class stock_picking_type(osv.osv):
3380     _name = "stock.picking.type"
3381     _description = "The picking type determines the picking view"
3382
3383     def __get_bar_values(self, cr, uid, obj, domain, read_fields, value_field, groupby_field, context=None):
3384         """ Generic method to generate data for bar chart values using SparklineBarWidget.
3385             This method performs obj.read_group(cr, uid, domain, read_fields, groupby_field).
3386
3387             :param obj: the target model (i.e. crm_lead)
3388             :param domain: the domain applied to the read_group
3389             :param list read_fields: the list of fields to read in the read_group
3390             :param str value_field: the field used to compute the value of the bar slice
3391             :param str groupby_field: the fields used to group
3392
3393             :return list section_result: a list of dicts: [
3394                                                 {   'value': (int) bar_column_value,
3395                                                     'tootip': (str) bar_column_tooltip,
3396                                                 }
3397                                             ]
3398         """
3399         month_begin = date.today().replace(day=1)
3400         section_result = [{
3401                             'value': 0,
3402                             'tooltip': (month_begin + relativedelta.relativedelta(months=-i)).strftime('%B'),
3403                             } for i in range(10, -1, -1)]
3404         group_obj = obj.read_group(cr, uid, domain, read_fields, groupby_field, context=context)
3405         for group in group_obj:
3406             group_begin_date = datetime.strptime(group['__domain'][0][2], tools.DEFAULT_SERVER_DATE_FORMAT)
3407             month_delta = relativedelta.relativedelta(month_begin, group_begin_date)
3408             section_result[10 - (month_delta.months + 1)] = {'value': group.get(value_field, 0), 'tooltip': group_begin_date.strftime('%B')}
3409         return section_result
3410
3411     def _get_picking_data(self, cr, uid, ids, field_name, arg, context=None):
3412         obj = self.pool.get('stock.picking')
3413         res = dict.fromkeys(ids, False)
3414         month_begin = date.today().replace(day=1)
3415         groupby_begin = (month_begin + relativedelta.relativedelta(months=-4)).strftime(tools.DEFAULT_SERVER_DATE_FORMAT)
3416         groupby_end = (month_begin + relativedelta.relativedelta(months=3)).strftime(tools.DEFAULT_SERVER_DATE_FORMAT)
3417         for id in ids:
3418             created_domain = [
3419                 ('picking_type_id', '=', id),
3420                 ('state', 'not in', ['done', 'cancel']),
3421                 ('date', '>=', groupby_begin),
3422                 ('date', '<', groupby_end),
3423             ]
3424             res[id] = self.__get_bar_values(cr, uid, obj, created_domain, ['date'], 'picking_type_id_count', 'date', context=context)
3425         return res
3426
3427     def _get_picking_count(self, cr, uid, ids, field_names, arg, context=None):
3428         obj = self.pool.get('stock.picking')
3429         domains = {
3430             'count_picking_draft': [('state', '=', 'draft')],
3431             'count_picking_waiting': [('state','=','confirmed')],
3432             'count_picking_ready': [('state','=','assigned')],
3433             'count_picking': [('state','in',('assigned','waiting','confirmed'))],
3434             'count_picking_late': [('min_date','<', time.strftime('%Y-%m-%d %H:%M:%S')), ('state','in',('assigned','waiting','confirmed'))],
3435             'count_picking_backorders': [('backorder_id','<>', False), ('state','!=','done')],
3436         }
3437         result = {}
3438         for field in domains:
3439             data = obj.read_group(cr, uid, domains[field] +
3440                 [('state', 'not in',('done','cancel')), ('picking_type_id', 'in', ids)],
3441                 ['picking_type_id'], ['picking_type_id'], context=context)
3442             count = dict(map(lambda x: (x['picking_type_id'] and x['picking_type_id'][0], x['picking_type_id_count']), data))
3443             for tid in ids:
3444                 result.setdefault(tid, {})[field] = count.get(tid, 0)
3445         for tid in ids:
3446             if result[tid]['count_picking']:
3447                 result[tid]['rate_picking_late'] = result[tid]['count_picking_late'] *100 / result[tid]['count_picking']
3448                 result[tid]['rate_picking_backorders'] = result[tid]['count_picking_backorders'] *100 / (result[tid]['count_picking'] + result[tid]['count_picking_draft'])
3449             else:
3450                 result[tid]['rate_picking_late'] = 0
3451                 result[tid]['rate_picking_backorders'] = 0
3452         return result
3453
3454     #TODO: not returning valus in required format to show in sparkline library,just added latest_picking_waiting need to add proper logic.
3455     def _get_picking_history(self, cr, uid, ids, field_names, arg, context=None):
3456         obj = self.pool.get('stock.picking')
3457         result = {}
3458         for id in ids:
3459             result[id] = {
3460                 'latest_picking_late': [],
3461                 'latest_picking_backorders': [],
3462                 'latest_picking_waiting': []
3463             }
3464         for type_id in ids:
3465             pick_ids = obj.search(cr, uid, [('state', '=','done'), ('picking_type_id','=',type_id)], limit=12, order="date desc", context=context)
3466             for pick in obj.browse(cr, uid, pick_ids, context=context):
3467                 result[type_id]['latest_picking_late'] = cmp(pick.date[:10], time.strftime('%Y-%m-%d'))
3468                 result[type_id]['latest_picking_backorders'] = bool(pick.backorder_id)
3469                 result[type_id]['latest_picking_waiting'] = cmp(pick.date[:10], time.strftime('%Y-%m-%d'))
3470         return result
3471
3472     def onchange_picking_code(self, cr, uid, ids, picking_code=False):
3473         if not picking_code:
3474             return False
3475         
3476         obj_data = self.pool.get('ir.model.data')
3477         stock_loc = obj_data.get_object_reference(cr, uid, 'stock','stock_location_stock')[1]
3478         
3479         result = {
3480             'default_location_src_id': stock_loc,
3481             'default_location_dest_id': stock_loc,
3482         }
3483         if picking_code == 'incoming':
3484             result['default_location_src_id'] = obj_data.get_object_reference(cr, uid, 'stock','stock_location_suppliers')[1]
3485             return {'value': result}
3486         if picking_code == 'outgoing':
3487             result['default_location_dest_id'] = obj_data.get_object_reference(cr, uid, 'stock','stock_location_customers')[1]
3488             return {'value': result}
3489         else:
3490             return {'value': result}
3491
3492     def _get_name(self, cr, uid, ids, field_names, arg, context=None):
3493         return dict(self.name_get(cr, uid, ids, context=context))
3494
3495     def name_get(self, cr, uid, ids, context=None):
3496         """Overides orm name_get method to display 'Warehouse_name: PickingType_name' """
3497         if context is None:
3498             context = {}
3499         if not isinstance(ids, list):
3500             ids = [ids]
3501         res = []
3502         if not ids:
3503             return res
3504         for record in self.browse(cr, uid, ids, context=context):
3505             name = record.name
3506             if record.warehouse_id:
3507                 name = record.warehouse_id.name + ': ' +name
3508             if context.get('special_shortened_wh_name'):
3509                 if record.warehouse_id:
3510                     name = record.warehouse_id.name
3511                 else:
3512                     name = _('Customer') + ' (' + record.name + ')'
3513             res.append((record.id, name))
3514         return res
3515
3516     def _default_warehouse(self, cr, uid, context=None):
3517         user = self.pool.get('res.users').browse(cr, uid, uid, context)
3518         res = self.pool.get('stock.warehouse').search(cr, uid, [('company_id', '=', user.company_id.id)], limit=1, context=context)
3519         return res and res[0] or False
3520
3521     _columns = {
3522         'name': fields.char('Name', translate=True, required=True),
3523         'complete_name': fields.function(_get_name, type='char', string='Name'),
3524         'pack': fields.boolean('Prefill Pack Operations', help='This picking type needs packing interface'),
3525         'auto_force_assign': fields.boolean('Automatic Availability', help='This picking type does\'t need to check for the availability in source location.'),
3526         'color': fields.integer('Color'),
3527         'delivery': fields.boolean('Print delivery'),
3528         'sequence_id': fields.many2one('ir.sequence', 'Reference Sequence', required=True),
3529         'default_location_src_id': fields.many2one('stock.location', 'Default Source Location'),
3530         'default_location_dest_id': fields.many2one('stock.location', 'Default Destination Location'),
3531         #TODO: change field name to "code" as it's not a many2one anymore
3532         'code_id': fields.selection([('incoming', 'Suppliers'), ('outgoing', 'Customers'), ('internal', 'Internal')], 'Picking type code', required=True),
3533         'return_picking_type_id': fields.many2one('stock.picking.type', 'Picking Type for Returns'),
3534         'warehouse_id': fields.many2one('stock.warehouse', 'Warehouse', ondelete='cascade'),
3535         'active': fields.boolean('Active'),
3536
3537         # Statistics for the kanban view
3538         'weekly_picking': fields.function(_get_picking_data,
3539             type='string',
3540             string='Scheduled pickings per week'),
3541
3542         'count_picking_draft': fields.function(_get_picking_count,
3543             type='integer', multi='_get_picking_count'),
3544         'count_picking_ready': fields.function(_get_picking_count,
3545             type='integer', multi='_get_picking_count'),
3546         'count_picking': fields.function(_get_picking_count,
3547             type='integer', multi='_get_picking_count'),
3548         'count_picking_waiting': fields.function(_get_picking_count,
3549             type='integer', multi='_get_picking_count'),
3550         'count_picking_late': fields.function(_get_picking_count,
3551             type='integer', multi='_get_picking_count'),
3552         'count_picking_backorders': fields.function(_get_picking_count,
3553             type='integer', multi='_get_picking_count'),
3554
3555         'rate_picking_late': fields.function(_get_picking_count,
3556             type='integer', multi='_get_picking_count'),
3557         'rate_picking_backorders': fields.function(_get_picking_count,
3558             type='integer', multi='_get_picking_count'),
3559
3560         'latest_picking_late': fields.function(_get_picking_history,
3561             type='string', multi='_get_picking_history'),
3562         'latest_picking_backorders': fields.function(_get_picking_history,
3563             type='string', multi='_get_picking_history'),
3564         'latest_picking_waiting': fields.function(_get_picking_history,
3565             type='string', multi='_get_picking_history'),
3566
3567     }
3568     _defaults = {
3569         'warehouse_id': _default_warehouse,
3570         'active': True,
3571     }
3572
3573
3574 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: