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