[IMP] stock : Improved the onchange method.
[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 lxml import etree
23 from datetime import datetime
24 from dateutil.relativedelta import relativedelta
25 import time
26 from operator import itemgetter
27 from itertools import groupby
28
29 from osv import fields, osv
30 from tools.translate import _
31 import netsvc
32 import tools
33 from tools import float_compare
34 import decimal_precision as dp
35 import logging
36 _logger = logging.getLogger(__name__)
37
38 #----------------------------------------------------------
39 # Incoterms
40 #----------------------------------------------------------
41 class stock_incoterms(osv.osv):
42     _name = "stock.incoterms"
43     _description = "Incoterms"
44     _columns = {
45         '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."),
46         'code': fields.char('Code', size=3, required=True, help="Code for Incoterms"),
47         'active': fields.boolean('Active', help="By unchecking the active field, you may hide an INCOTERM without deleting it."),
48     }
49     _defaults = {
50         'active': True,
51     }
52
53 stock_incoterms()
54
55 class stock_journal(osv.osv):
56     _name = "stock.journal"
57     _description = "Stock Journal"
58     _columns = {
59         'name': fields.char('Stock Journal', size=32, required=True),
60         'user_id': fields.many2one('res.users', 'Responsible'),
61     }
62     _defaults = {
63         'user_id': lambda s, c, u, ctx: u
64     }
65
66 stock_journal()
67
68 #----------------------------------------------------------
69 # Stock Location
70 #----------------------------------------------------------
71 class stock_location(osv.osv):
72     _name = "stock.location"
73     _description = "Location"
74     _parent_name = "location_id"
75     _parent_store = True
76     _parent_order = 'posz,name'
77     _order = 'parent_left'
78
79     def name_get(self, cr, uid, ids, context=None):
80         # always return the full hierarchical name
81         res = self._complete_name(cr, uid, ids, 'complete_name', None, context=context)
82         return res.items()
83
84     def _complete_name(self, cr, uid, ids, name, args, context=None):
85         """ Forms complete name of location from parent location to child location.
86         @return: Dictionary of values
87         """
88         res = {}
89         for m in self.browse(cr, uid, ids, context=context):
90             names = [m.name]
91             parent = m.location_id
92             while parent:
93                 names.append(parent.name)
94                 parent = parent.location_id
95             res[m.id] = ' / '.join(reversed(names))
96         return res
97
98     def _get_sublocations(self, cr, uid, ids, context=None):
99         """ return all sublocations of the given stock locations (included) """
100         return self.search(cr, uid, [('id', 'child_of', ids)], context=context)
101
102     def _product_value(self, cr, uid, ids, field_names, arg, context=None):
103         """Computes stock value (real and virtual) for a product, as well as stock qty (real and virtual).
104         @param field_names: Name of field
105         @return: Dictionary of values
106         """
107         prod_id = context and context.get('product_id', False)
108
109         if not prod_id:
110             return dict([(i, {}.fromkeys(field_names, 0.0)) for i in ids])
111
112         product_product_obj = self.pool.get('product.product')
113
114         cr.execute('select distinct product_id, location_id from stock_move where location_id in %s', (tuple(ids), ))
115         dict1 = cr.dictfetchall()
116         cr.execute('select distinct product_id, location_dest_id as location_id from stock_move where location_dest_id in %s', (tuple(ids), ))
117         dict2 = cr.dictfetchall()
118         res_products_by_location = sorted(dict1+dict2, key=itemgetter('location_id'))
119         products_by_location = dict((k, [v['product_id'] for v in itr]) for k, itr in groupby(res_products_by_location, itemgetter('location_id')))
120
121         result = dict([(i, {}.fromkeys(field_names, 0.0)) for i in ids])
122         result.update(dict([(i, {}.fromkeys(field_names, 0.0)) for i in list(set([aaa['location_id'] for aaa in res_products_by_location]))]))
123
124         currency_id = self.pool.get('res.users').browse(cr, uid, uid).company_id.currency_id.id
125         currency_obj = self.pool.get('res.currency')
126         currency = currency_obj.browse(cr, uid, currency_id, context=context)
127         for loc_id, product_ids in products_by_location.items():
128             if prod_id:
129                 product_ids = [prod_id]
130             c = (context or {}).copy()
131             c['location'] = loc_id
132             for prod in product_product_obj.browse(cr, uid, product_ids, context=c):
133                 for f in field_names:
134                     if f == 'stock_real':
135                         if loc_id not in result:
136                             result[loc_id] = {}
137                         result[loc_id][f] += prod.qty_available
138                     elif f == 'stock_virtual':
139                         result[loc_id][f] += prod.virtual_available
140                     elif f == 'stock_real_value':
141                         amount = prod.qty_available * prod.standard_price
142                         amount = currency_obj.round(cr, uid, currency, amount)
143                         result[loc_id][f] += amount
144                     elif f == 'stock_virtual_value':
145                         amount = prod.virtual_available * prod.standard_price
146                         amount = currency_obj.round(cr, uid, currency, amount)
147                         result[loc_id][f] += amount
148         return result
149
150     _columns = {
151         'name': fields.char('Location Name', size=64, required=True, translate=True),
152         'active': fields.boolean('Active', help="By unchecking the active field, you may hide a location without deleting it."),
153         '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,
154                  help="""* Supplier Location: Virtual location representing the source location for products coming from your suppliers
155                        \n* View: Virtual location used to create a hierarchical structures for your warehouse, aggregating its child locations ; can't directly contain products
156                        \n* Internal Location: Physical locations inside your own warehouses,
157                        \n* Customer Location: Virtual location representing the destination location for products sent to your customers
158                        \n* Inventory: Virtual location serving as counterpart for inventory operations used to correct stock levels (Physical inventories)
159                        \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.
160                        \n* Production: Virtual counterpart location for production operations: this location consumes the raw material and produces finished products
161                       """, select = True),
162          # temporarily removed, as it's unused: 'allocation_method': fields.selection([('fifo', 'FIFO'), ('lifo', 'LIFO'), ('nearest', 'Nearest')], 'Allocation Method', required=True),
163         'complete_name': fields.function(_complete_name, type='char', size=256, string="Location Name",
164                             store={'stock.location': (_get_sublocations, ['name', 'location_id'], 10)}),
165
166         'stock_real': fields.function(_product_value, type='float', string='Real Stock', multi="stock"),
167         'stock_virtual': fields.function(_product_value, type='float', string='Virtual Stock', multi="stock"),
168
169         'location_id': fields.many2one('stock.location', 'Parent Location', select=True, ondelete='cascade'),
170         'child_ids': fields.one2many('stock.location', 'location_id', 'Contains'),
171
172         'chained_journal_id': fields.many2one('stock.journal', 'Chaining Journal',help="Inventory Journal in which the chained move will be written, if the Chaining Type is not Transparent (no journal is used if left empty)"),
173         'chained_location_id': fields.many2one('stock.location', 'Chained Location If Fixed'),
174         'chained_location_type': fields.selection([('none', 'None'), ('customer', 'Customer'), ('fixed', 'Fixed Location')],
175             'Chained Location Type', required=True,
176             help="Determines whether this location is chained to another location, i.e. any incoming product in this location \n" \
177                 "should next go to the chained location. The chained location is determined according to the type :"\
178                 "\n* None: No chaining at all"\
179                 "\n* Customer: The chained location will be taken from the Customer Location field on the Partner form of the Partner that is specified in the Picking list of the incoming products." \
180                 "\n* Fixed Location: The chained location is taken from the next field: Chained Location if Fixed." \
181                 ),
182         'chained_auto_packing': fields.selection(
183             [('auto', 'Automatic Move'), ('manual', 'Manual Operation'), ('transparent', 'Automatic No Step Added')],
184             'Chaining Type',
185             required=True,
186             help="This is used only if you select a chained location type.\n" \
187                 "The 'Automatic Move' value will create a stock move after the current one that will be "\
188                 "validated automatically. With 'Manual Operation', the stock move has to be validated "\
189                 "by a worker. With 'Automatic No Step Added', the location is replaced in the original move."
190             ),
191         'chained_picking_type': fields.selection([('out', 'Sending Goods'), ('in', 'Getting Goods'), ('internal', 'Internal')], 'Shipping Type', help="Shipping Type of the Picking List that will contain the chained move (leave empty to automatically detect the type based on the source and destination locations)."),
192         'chained_company_id': fields.many2one('res.company', 'Chained Company', help='The company the Picking List containing the chained move will belong to (leave empty to use the default company determination rules'),
193         'chained_delay': fields.integer('Chaining Lead Time',help="Delay between original move and chained move in days"),
194         'partner_id': fields.many2one('res.partner', 'Location Address',help="Address of  customer or supplier."),
195         'icon': fields.selection(tools.icons, 'Icon', size=64,help="Icon show in  hierarchical tree view"),
196
197         'comment': fields.text('Additional Information'),
198         'posx': fields.integer('Corridor (X)',help="Optional localization details, for information purpose only"),
199         'posy': fields.integer('Shelves (Y)', help="Optional localization details, for information purpose only"),
200         'posz': fields.integer('Height (Z)', help="Optional localization details, for information purpose only"),
201
202         'parent_left': fields.integer('Left Parent', select=1),
203         'parent_right': fields.integer('Right Parent', select=1),
204         'stock_real_value': fields.function(_product_value, type='float', string='Real Stock Value', multi="stock", digits_compute=dp.get_precision('Account')),
205         'stock_virtual_value': fields.function(_product_value, type='float', string='Virtual Stock Value', multi="stock", digits_compute=dp.get_precision('Account')),
206         'company_id': fields.many2one('res.company', 'Company', select=1, help='Let this field empty if this location is shared between all companies'),
207         'scrap_location': fields.boolean('Scrap Location', help='Check this box to allow using this location to put scrapped/damaged goods.'),
208         'valuation_in_account_id': fields.many2one('account.account', 'Stock Valuation Account (Incoming)', domain = [('type','=','other')],
209                                                    help="Used for real-time inventory valuation. When set on a virtual location (non internal type), "
210                                                         "this account will be used to hold the value of products being moved from an internal location "
211                                                         "into this location, instead of the generic Stock Output Account set on the product. "
212                                                         "This has no effect for internal locations."),
213         'valuation_out_account_id': fields.many2one('account.account', 'Stock Valuation Account (Outgoing)', domain = [('type','=','other')],
214                                                    help="Used for real-time inventory valuation. When set on a virtual location (non internal type), "
215                                                         "this account will be used to hold the value of products being moved out of this location "
216                                                         "and into an internal location, instead of the generic Stock Output Account set on the product. "
217                                                         "This has no effect for internal locations."),
218     }
219     _defaults = {
220         'active': True,
221         'usage': 'internal',
222         'chained_location_type': 'none',
223         'chained_auto_packing': 'manual',
224         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.location', context=c),
225         'posx': 0,
226         'posy': 0,
227         'posz': 0,
228         'icon': False,
229         'scrap_location': False,
230     }
231
232     def chained_location_get(self, cr, uid, location, partner=None, product=None, context=None):
233         """ Finds chained location
234         @param location: Location id
235         @param partner: Partner id
236         @param product: Product id
237         @return: List of values
238         """
239         result = None
240         if location.chained_location_type == 'customer':
241             if partner:
242                 result = partner.property_stock_customer
243         elif location.chained_location_type == 'fixed':
244             result = location.chained_location_id
245         if result:
246             return result, location.chained_auto_packing, location.chained_delay, location.chained_journal_id and location.chained_journal_id.id or False, location.chained_company_id and location.chained_company_id.id or False, location.chained_picking_type
247         return result
248
249     def picking_type_get(self, cr, uid, from_location, to_location, context=None):
250         """ Gets type of picking.
251         @param from_location: Source location
252         @param to_location: Destination location
253         @return: Location type
254         """
255         result = 'internal'
256         if (from_location.usage=='internal') and (to_location and to_location.usage in ('customer', 'supplier')):
257             result = 'out'
258         elif (from_location.usage in ('supplier', 'customer')) and (to_location.usage == 'internal'):
259             result = 'in'
260         return result
261
262     def _product_get_all_report(self, cr, uid, ids, product_ids=False, context=None):
263         return self._product_get_report(cr, uid, ids, product_ids, context, recursive=True)
264
265     def _product_get_report(self, cr, uid, ids, product_ids=False,
266             context=None, recursive=False):
267         """ Finds the product quantity and price for particular location.
268         @param product_ids: Ids of product
269         @param recursive: True or False
270         @return: Dictionary of values
271         """
272         if context is None:
273             context = {}
274         product_obj = self.pool.get('product.product')
275         # Take the user company and pricetype
276         context['currency_id'] = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id.currency_id.id
277
278         # To be able to offer recursive or non-recursive reports we need to prevent recursive quantities by default
279         context['compute_child'] = False
280
281         if not product_ids:
282             product_ids = product_obj.search(cr, uid, [], context={'active_test': False})
283
284         products = product_obj.browse(cr, uid, product_ids, context=context)
285         products_by_uom = {}
286         products_by_id = {}
287         for product in products:
288             products_by_uom.setdefault(product.uom_id.id, [])
289             products_by_uom[product.uom_id.id].append(product)
290             products_by_id.setdefault(product.id, [])
291             products_by_id[product.id] = product
292
293         result = {}
294         result['product'] = []
295         for id in ids:
296             quantity_total = 0.0
297             total_price = 0.0
298             for uom_id in products_by_uom.keys():
299                 fnc = self._product_get
300                 if recursive:
301                     fnc = self._product_all_get
302                 ctx = context.copy()
303                 ctx['uom'] = uom_id
304                 qty = fnc(cr, uid, id, [x.id for x in products_by_uom[uom_id]],
305                         context=ctx)
306                 for product_id in qty.keys():
307                     if not qty[product_id]:
308                         continue
309                     product = products_by_id[product_id]
310                     quantity_total += qty[product_id]
311
312                     # Compute based on pricetype
313                     # Choose the right filed standard_price to read
314                     amount_unit = product.price_get('standard_price', context=context)[product.id]
315                     price = qty[product_id] * amount_unit
316
317                     total_price += price
318                     result['product'].append({
319                         'price': amount_unit,
320                         'prod_name': product.name,
321                         'code': product.default_code, # used by lot_overview_all report!
322                         'variants': product.variants or '',
323                         'uom': product.uom_id.name,
324                         'prod_qty': qty[product_id],
325                         'price_value': price,
326                     })
327         result['total'] = quantity_total
328         result['total_price'] = total_price
329         return result
330
331     def _product_get_multi_location(self, cr, uid, ids, product_ids=False, context=None,
332                                     states=['done'], what=('in', 'out')):
333         """
334         @param product_ids: Ids of product
335         @param states: List of states
336         @param what: Tuple of
337         @return:
338         """
339         product_obj = self.pool.get('product.product')
340         if context is None:
341             context = {}
342         context.update({
343             'states': states,
344             'what': what,
345             'location': ids
346         })
347         return product_obj.get_product_available(cr, uid, product_ids, context=context)
348
349     def _product_get(self, cr, uid, id, product_ids=False, context=None, states=['done']):
350         """
351         @param product_ids:
352         @param states:
353         @return:
354         """
355         ids = id and [id] or []
356         return self._product_get_multi_location(cr, uid, ids, product_ids, context=context, states=states)
357
358     def _product_all_get(self, cr, uid, id, product_ids=False, context=None, states=['done']):
359         # build the list of ids of children of the location given by id
360         ids = id and [id] or []
361         location_ids = self.search(cr, uid, [('location_id', 'child_of', ids)])
362         return self._product_get_multi_location(cr, uid, location_ids, product_ids, context, states)
363
364     def _product_virtual_get(self, cr, uid, id, product_ids=False, context=None, states=['done']):
365         return self._product_all_get(cr, uid, id, product_ids, context, ['confirmed', 'waiting', 'assigned', 'done'])
366
367     def _product_reserve(self, cr, uid, ids, product_id, product_qty, context=None, lock=False):
368         """
369         Attempt to find a quantity ``product_qty`` (in the product's default uom or the uom passed in ``context``) of product ``product_id``
370         in locations with id ``ids`` and their child locations. If ``lock`` is True, the stock.move lines
371         of product with id ``product_id`` in the searched location will be write-locked using Postgres's
372         "FOR UPDATE NOWAIT" option until the transaction is committed or rolled back, to prevent reservin
373         twice the same products.
374         If ``lock`` is True and the lock cannot be obtained (because another transaction has locked some of
375         the same stock.move lines), a log line will be output and False will be returned, as if there was
376         not enough stock.
377
378         :param product_id: Id of product to reserve
379         :param product_qty: Quantity of product to reserve (in the product's default uom or the uom passed in ``context``)
380         :param lock: if True, the stock.move lines of product with id ``product_id`` in all locations (and children locations) with ``ids`` will
381                      be write-locked using postgres's "FOR UPDATE NOWAIT" option until the transaction is committed or rolled back. This is
382                      to prevent reserving twice the same products.
383         :param context: optional context dictionary: if a 'uom' key is present it will be used instead of the default product uom to
384                         compute the ``product_qty`` and in the return value.
385         :return: List of tuples in the form (qty, location_id) with the (partial) quantities that can be taken in each location to
386                  reach the requested product_qty (``qty`` is expressed in the default uom of the product), of False if enough
387                  products could not be found, or the lock could not be obtained (and ``lock`` was True).
388         """
389         result = []
390         amount = 0.0
391         if context is None:
392             context = {}
393         uom_obj = self.pool.get('product.uom')
394         uom_rounding = self.pool.get('product.product').browse(cr, uid, product_id, context=context).uom_id.rounding
395         if context.get('uom'):
396             uom_rounding = uom_obj.browse(cr, uid, context.get('uom'), context=context).rounding
397         for id in self.search(cr, uid, [('location_id', 'child_of', ids)]):
398             if lock:
399                 try:
400                     # Must lock with a separate select query because FOR UPDATE can't be used with
401                     # aggregation/group by's (when individual rows aren't identifiable).
402                     # We use a SAVEPOINT to be able to rollback this part of the transaction without
403                     # failing the whole transaction in case the LOCK cannot be acquired.
404                     cr.execute("SAVEPOINT stock_location_product_reserve")
405                     cr.execute("""SELECT id FROM stock_move
406                                   WHERE product_id=%s AND
407                                           (
408                                             (location_dest_id=%s AND
409                                              location_id<>%s AND
410                                              state='done')
411                                             OR
412                                             (location_id=%s AND
413                                              location_dest_id<>%s AND
414                                              state in ('done', 'assigned'))
415                                           )
416                                   FOR UPDATE of stock_move NOWAIT""", (product_id, id, id, id, id), log_exceptions=False)
417                 except Exception:
418                     # Here it's likely that the FOR UPDATE NOWAIT failed to get the LOCK,
419                     # so we ROLLBACK to the SAVEPOINT to restore the transaction to its earlier
420                     # state, we return False as if the products were not available, and log it:
421                     cr.execute("ROLLBACK TO stock_location_product_reserve")
422                     _logger.warn("Failed attempt to reserve %s x product %s, likely due to another transaction already in progress. Next attempt is likely to work. Detailed error available at DEBUG level.", product_qty, product_id)
423                     _logger.debug("Trace of the failed product reservation attempt: ", exc_info=True)
424                     return False
425
426             # XXX TODO: rewrite this with one single query, possibly even the quantity conversion
427             cr.execute("""SELECT product_uom, sum(product_qty) AS product_qty
428                           FROM stock_move
429                           WHERE location_dest_id=%s AND
430                                 location_id<>%s AND
431                                 product_id=%s AND
432                                 state='done'
433                           GROUP BY product_uom
434                        """,
435                        (id, id, product_id))
436             results = cr.dictfetchall()
437             cr.execute("""SELECT product_uom,-sum(product_qty) AS product_qty
438                           FROM stock_move
439                           WHERE location_id=%s AND
440                                 location_dest_id<>%s AND
441                                 product_id=%s AND
442                                 state in ('done', 'assigned')
443                           GROUP BY product_uom
444                        """,
445                        (id, id, product_id))
446             results += cr.dictfetchall()
447             total = 0.0
448             results2 = 0.0
449             for r in results:
450                 amount = uom_obj._compute_qty(cr, uid, r['product_uom'], r['product_qty'], context.get('uom', False))
451                 results2 += amount
452                 total += amount
453             if total <= 0.0:
454                 continue
455
456             amount = results2
457             compare_qty = float_compare(amount, 0, precision_rounding=uom_rounding)
458             if compare_qty == 1:
459                 if amount > min(total, product_qty):
460                     amount = min(product_qty, total)
461                 result.append((amount, id))
462                 product_qty -= amount
463                 total -= amount
464                 if product_qty <= 0.0:
465                     return result
466                 if total <= 0.0:
467                     continue
468         return False
469
470 stock_location()
471
472
473 class stock_tracking(osv.osv):
474     _name = "stock.tracking"
475     _description = "Packs"
476
477     def checksum(sscc):
478         salt = '31' * 8 + '3'
479         sum = 0
480         for sscc_part, salt_part in zip(sscc, salt):
481             sum += int(sscc_part) * int(salt_part)
482         return (10 - (sum % 10)) % 10
483     checksum = staticmethod(checksum)
484
485     def make_sscc(self, cr, uid, context=None):
486         sequence = self.pool.get('ir.sequence').get(cr, uid, 'stock.lot.tracking')
487         try:
488             return sequence + str(self.checksum(sequence))
489         except Exception:
490             return sequence
491
492     _columns = {
493         'name': fields.char('Pack Reference', size=64, required=True, select=True, help="By default, the pack reference is generated following the sscc standard. (Serial number + 1 check digit)"),
494         'active': fields.boolean('Active', help="By unchecking the active field, you may hide a pack without deleting it."),
495         'serial': fields.char('Additional Reference', size=64, select=True, help="Other reference or serial number"),
496         'move_ids': fields.one2many('stock.move', 'tracking_id', 'Moves for this pack', readonly=True),
497         'date': fields.datetime('Creation Date', required=True),
498     }
499     _defaults = {
500         'active': 1,
501         'name': make_sscc,
502         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
503     }
504
505     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=100):
506         if not args:
507             args = []
508         ids = self.search(cr, user, [('serial', '=', name)]+ args, limit=limit, context=context)
509         ids += self.search(cr, user, [('name', operator, name)]+ args, limit=limit, context=context)
510         return self.name_get(cr, user, ids, context)
511
512     def name_get(self, cr, uid, ids, context=None):
513         if not len(ids):
514             return []
515         res = [(r['id'], r['name']+' ['+(r['serial'] or '')+']') for r in self.read(cr, uid, ids, ['name', 'serial'], context)]
516         return res
517
518     def unlink(self, cr, uid, ids, context=None):
519         raise osv.except_osv(_('Error!'), _('You cannot remove a lot line.'))
520
521     def action_traceability(self, cr, uid, ids, context={}):
522         """ It traces the information of a product
523         @param self: The object pointer.
524         @param cr: A database cursor
525         @param uid: ID of the user currently logged in
526         @param ids: List of IDs selected
527         @param context: A standard dictionary
528         @return: A dictionary of values
529         """
530         return self.pool.get('action.traceability').action_traceability(cr,uid,ids,context)
531
532 stock_tracking()
533
534 #----------------------------------------------------------
535 # Stock Picking
536 #----------------------------------------------------------
537 class stock_picking(osv.osv):
538     _name = "stock.picking"
539     _inherit = ['mail.thread']
540     _description = "Picking List"
541
542     def _set_maximum_date(self, cr, uid, ids, name, value, arg, context=None):
543         """ Calculates planned date if it is greater than 'value'.
544         @param name: Name of field
545         @param value: Value of field
546         @param arg: User defined argument
547         @return: True or False
548         """
549         if not value:
550             return False
551         if isinstance(ids, (int, long)):
552             ids = [ids]
553         for pick in self.browse(cr, uid, ids, context=context):
554             sql_str = """update stock_move set
555                     date='%s'
556                 where
557                     picking_id=%d """ % (value, pick.id)
558
559             if pick.max_date:
560                 sql_str += " and (date='" + pick.max_date + "' or date>'" + value + "')"
561             cr.execute(sql_str)
562         return True
563
564     def _set_minimum_date(self, cr, uid, ids, name, value, arg, context=None):
565         """ Calculates planned date if it is less than 'value'.
566         @param name: Name of field
567         @param value: Value of field
568         @param arg: User defined argument
569         @return: True or False
570         """
571         if not value:
572             return False
573         if isinstance(ids, (int, long)):
574             ids = [ids]
575         for pick in self.browse(cr, uid, ids, context=context):
576             sql_str = """update stock_move set
577                     date='%s'
578                 where
579                     picking_id=%s """ % (value, pick.id)
580             if pick.min_date:
581                 sql_str += " and (date='" + pick.min_date + "' or date<'" + value + "')"
582             cr.execute(sql_str)
583         return True
584
585     def get_min_max_date(self, cr, uid, ids, field_name, arg, context=None):
586         """ Finds minimum and maximum dates for picking.
587         @return: Dictionary of values
588         """
589         res = {}
590         for id in ids:
591             res[id] = {'min_date': False, 'max_date': False}
592         if not ids:
593             return res
594         cr.execute("""select
595                 picking_id,
596                 min(date_expected),
597                 max(date_expected)
598             from
599                 stock_move
600             where
601                 picking_id IN %s
602             group by
603                 picking_id""",(tuple(ids),))
604         for pick, dt1, dt2 in cr.fetchall():
605             res[pick]['min_date'] = dt1
606             res[pick]['max_date'] = dt2
607         return res
608
609     def create(self, cr, user, vals, context=None):
610         if ('name' not in vals) or (vals.get('name')=='/'):
611             seq_obj_name =  'stock.picking.' + vals['type']
612             vals['name'] = self.pool.get('ir.sequence').get(cr, user, seq_obj_name)
613         new_id = super(stock_picking, self).create(cr, user, vals, context)
614         if new_id:
615             self.create_send_note(cr, user, [new_id], context=context)
616         return new_id
617
618     _columns = {
619         'name': fields.char('Reference', size=64, select=True, states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
620         'origin': fields.char('Source', size=64, states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}, help="Reference of the document", select=True),
621         '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),
622         'type': fields.selection([('out', 'Sending Goods'), ('in', 'Getting Goods'), ('internal', 'Internal')], 'Shipping Type', required=True, select=True, readonly=True, help="Shipping type specify, goods coming in or going out."),
623         'note': fields.text('Notes', states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
624         'stock_journal_id': fields.many2one('stock.journal','Stock Journal', select=True, states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
625         'location_id': fields.many2one('stock.location', 'Location', states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}, help="Keep empty if you produce at the location where the finished products are needed." \
626                 "Set a location if you produce at a fixed location. This can be a partner location " \
627                 "if you subcontract the manufacturing operations.", select=True),
628         'location_dest_id': fields.many2one('stock.location', 'Dest. Location', states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}, help="Location where the system will stock the finished products.", select=True),
629         '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"),
630         'state': fields.selection([
631             ('draft', 'Draft'),
632             ('cancel', 'Cancelled'),
633             ('auto', 'Waiting Another Operation'),
634             ('confirmed', 'Waiting Availability'),
635             ('assigned', 'Ready to Transfer'),
636             ('done', 'Transferred'),
637             ], 'Status', readonly=True, select=True, help="""
638             * Draft: not confirmed yet and will not be scheduled until confirmed\n
639             * Waiting Another Operation: waiting for another move to proceed before it becomes automatically available (e.g. in Make-To-Order flows)\n
640             * Waiting Availability: still waiting for the availability of products\n
641             * Ready to Transfer: products reserved, simply waiting for confirmation.\n
642             * Transferred: has been processed, can't be modified or cancelled anymore\n
643             * Cancelled: has been cancelled, can't be confirmed anymore"""
644         ),
645         'min_date': fields.function(get_min_max_date, fnct_inv=_set_minimum_date, multi="min_max_date",
646                  store=True, type='datetime', string='Scheduled Date', select=1, help="Scheduled date for the shipment to be processed"),
647         'date': fields.datetime('Order Date', help="Date of order", select=True, states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
648         'date_done': fields.datetime('Date Done', help="Date of Completion", states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
649         'max_date': fields.function(get_min_max_date, fnct_inv=_set_maximum_date, multi="min_max_date",
650                  store=True, type='datetime', string='Max. Expected Date', select=2),
651         'move_lines': fields.one2many('stock.move', 'picking_id', 'Internal Moves', states={'done': [('readonly', True)], 'cancel': [('readonly', True)]}),
652         'product_id': fields.related('move_lines', 'product_id', type='many2one', relation='product.product', string='Product'),
653         'auto_picking': fields.boolean('Auto-Picking', states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
654         'partner_id': fields.many2one('res.partner', 'Partner', states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
655         'invoice_state': fields.selection([
656             ("invoiced", "Invoiced"),
657             ("2binvoiced", "To Be Invoiced"),
658             ("none", "Not Applicable")], "Invoice Control",
659             select=True, required=True, readonly=True, states={'draft': [('readonly', False)]}),
660         'company_id': fields.many2one('res.company', 'Company', required=True, select=True, states={'done':[('readonly', True)], 'cancel':[('readonly',True)]}),
661     }
662     _defaults = {
663         'name': lambda self, cr, uid, context: '/',
664         'state': 'draft',
665         'move_type': 'direct',
666         'type': 'internal',
667         'invoice_state': 'none',
668         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
669         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.picking', context=c)
670     }
671     _sql_constraints = [
672         ('name_uniq', 'unique(name, company_id)', 'Reference must be unique per Company!'),
673     ]
674
675     def action_process(self, cr, uid, ids, context=None):
676         if context is None:
677             context = {}
678         context.update({
679             'active_model': self._name,
680             'active_ids': ids,
681             'active_id': len(ids) and ids[0] or False
682         })
683         return {
684             'view_type': 'form',
685             'view_mode': 'form',
686             'res_model': 'stock.partial.picking',
687             'type': 'ir.actions.act_window',
688             'target': 'new',
689             'context': context,
690             'nodestroy': True,
691         }
692
693     def copy(self, cr, uid, id, default=None, context=None):
694         if default is None:
695             default = {}
696         default = default.copy()
697         picking_obj = self.browse(cr, uid, id, context=context)
698         move_obj=self.pool.get('stock.move')
699         if ('name' not in default) or (picking_obj.name=='/'):
700             seq_obj_name =  'stock.picking.' + picking_obj.type
701             default['name'] = self.pool.get('ir.sequence').get(cr, uid, seq_obj_name)
702             default['origin'] = ''
703             default['backorder_id'] = False
704         if picking_obj.invoice_state == 'invoiced':
705             default['invoice_state'] = '2binvoiced'
706         res=super(stock_picking, self).copy(cr, uid, id, default, context)
707         if res:
708             picking_obj = self.browse(cr, uid, res, context=context)
709             for move in picking_obj.move_lines:
710                 move_obj.write(cr, uid, [move.id], {'tracking_id': False,'prodlot_id':False, 'move_history_ids2': [(6, 0, [])], 'move_history_ids': [(6, 0, [])]})
711         return res
712     
713     def fields_view_get(self, cr, uid, view_id=None, view_type=False, context=None, toolbar=False, submenu=False):
714         if view_type == 'form' and not view_id:
715             mod_obj = self.pool.get('ir.model.data')
716             if self._name == "stock.picking.in":
717                 model,view_id = mod_obj.get_object_reference(cr, uid, 'stock', 'view_picking_in_form')
718             if self._name == "stock.picking.out":
719                 model,view_id = mod_obj.get_object_reference(cr, uid, 'stock', 'view_picking_out_form')
720         return super(stock_picking,self).fields_view_get(cr, uid, view_id=view_id, view_type=view_type, context=context, toolbar=toolbar, submenu=submenu)
721
722     def onchange_partner_in(self, cr, uid, context=None, partner_id=None):
723         return {}
724
725     def action_explode(self, cr, uid, moves, context=None):
726         return moves
727
728     def action_confirm(self, cr, uid, ids, context=None):
729         """ Confirms picking.
730         @return: True
731         """
732         pickings = self.browse(cr, uid, ids, context=context)
733         for picking in pickings:
734             if picking.state <> 'confirmed':
735                 self.confirm_send_note(cr, uid, [picking.id], context=context)
736         self.write(cr, uid, ids, {'state': 'confirmed'})
737         todo = []
738         for picking in pickings:
739             for r in picking.move_lines:
740                 if r.state == 'draft':
741                     todo.append(r.id)
742         todo = self.action_explode(cr, uid, todo, context)
743         if len(todo):
744             self.pool.get('stock.move').action_confirm(cr, uid, todo, context=context)
745         return True
746
747     def test_auto_picking(self, cr, uid, ids):
748         # TODO: Check locations to see if in the same location ?
749         return True
750
751     def action_assign(self, cr, uid, ids, *args):
752         """ Changes state of picking to available if all moves are confirmed.
753         @return: True
754         """
755         wf_service = netsvc.LocalService("workflow")
756         for pick in self.browse(cr, uid, ids):
757             if pick.state == 'draft':
758                 wf_service.trg_validate(uid, 'stock.picking', pick.id, 'button_confirm', cr)
759             move_ids = [x.id for x in pick.move_lines if x.state == 'confirmed']
760             if not move_ids:
761                 raise osv.except_osv(_('Warning!'),_('Not enough stock, unable to reserve the products.'))
762             self.pool.get('stock.move').action_assign(cr, uid, move_ids)
763         return True
764
765     def force_assign(self, cr, uid, ids, *args):
766         """ Changes state of picking to available if moves are confirmed or waiting.
767         @return: True
768         """
769         wf_service = netsvc.LocalService("workflow")
770         for pick in self.browse(cr, uid, ids):
771             move_ids = [x.id for x in pick.move_lines if x.state in ['confirmed','waiting']]
772             self.pool.get('stock.move').force_assign(cr, uid, move_ids)
773             wf_service.trg_write(uid, 'stock.picking', pick.id, cr)
774         return True
775
776     def draft_force_assign(self, cr, uid, ids, *args):
777         """ Confirms picking directly from draft state.
778         @return: True
779         """
780         wf_service = netsvc.LocalService("workflow")
781         for pick in self.browse(cr, uid, ids):
782             if not pick.move_lines:
783                 raise osv.except_osv(_('Error!'),_('You cannot process picking without stock moves.'))
784             wf_service.trg_validate(uid, 'stock.picking', pick.id,
785                 'button_confirm', cr)
786         return True
787
788     def draft_validate(self, cr, uid, ids, context=None):
789         """ Validates picking directly from draft state.
790         @return: True
791         """
792         wf_service = netsvc.LocalService("workflow")
793         self.draft_force_assign(cr, uid, ids)
794         for pick in self.browse(cr, uid, ids, context=context):
795             move_ids = [x.id for x in pick.move_lines]
796             self.pool.get('stock.move').force_assign(cr, uid, move_ids)
797             wf_service.trg_write(uid, 'stock.picking', pick.id, cr)
798         return self.action_process(
799             cr, uid, ids, context=context)
800     def cancel_assign(self, cr, uid, ids, *args):
801         """ Cancels picking and moves.
802         @return: True
803         """
804         wf_service = netsvc.LocalService("workflow")
805         for pick in self.browse(cr, uid, ids):
806             move_ids = [x.id for x in pick.move_lines]
807             self.pool.get('stock.move').cancel_assign(cr, uid, move_ids)
808             wf_service.trg_write(uid, 'stock.picking', pick.id, cr)
809         return True
810
811     def action_assign_wkf(self, cr, uid, ids, context=None):
812         """ Changes picking state to assigned.
813         @return: True
814         """
815         self.write(cr, uid, ids, {'state': 'assigned'})
816         return True
817
818     def test_finished(self, cr, uid, ids):
819         """ Tests whether the move is in done or cancel state or not.
820         @return: True or False
821         """
822         move_ids = self.pool.get('stock.move').search(cr, uid, [('picking_id', 'in', ids)])
823         for move in self.pool.get('stock.move').browse(cr, uid, move_ids):
824             if move.state not in ('done', 'cancel'):
825
826                 if move.product_qty != 0.0:
827                     return False
828                 else:
829                     move.write({'state': 'done'})
830         return True
831
832     def test_assigned(self, cr, uid, ids):
833         """ Tests whether the move is in assigned state or not.
834         @return: True or False
835         """
836         #TOFIX: assignment of move lines should be call before testing assigment otherwise picking never gone in assign state
837         ok = True
838         for pick in self.browse(cr, uid, ids):
839             mt = pick.move_type
840             # incomming shipments are always set as available if they aren't chained
841             if pick.type == 'in':
842                 if all([x.state != 'waiting' for x in pick.move_lines]):
843                     return True
844             for move in pick.move_lines:
845                 if (move.state in ('confirmed', 'draft')) and (mt == 'one'):
846                     return False
847                 if (mt == 'direct') and (move.state == 'assigned') and (move.product_qty):
848                     return True
849                 ok = ok and (move.state in ('cancel', 'done', 'assigned'))
850         return ok
851
852     def action_cancel(self, cr, uid, ids, context=None):
853         """ Changes picking state to cancel.
854         @return: True
855         """
856         for pick in self.browse(cr, uid, ids, context=context):
857             ids2 = [move.id for move in pick.move_lines]
858             self.pool.get('stock.move').action_cancel(cr, uid, ids2, context)
859         self.write(cr, uid, ids, {'state': 'cancel', 'invoice_state': 'none'})
860         self.ship_cancel_send_note(cr, uid, ids, context)
861         return True
862
863     #
864     # TODO: change and create a move if not parents
865     #
866     def action_done(self, cr, uid, ids, context=None):
867         """ Changes picking state to done.
868         @return: True
869         """
870         self.write(cr, uid, ids, {'state': 'done', 'date_done': time.strftime('%Y-%m-%d %H:%M:%S')})
871         return True
872
873     def action_move(self, cr, uid, ids, context=None):
874         """ Changes move state to assigned.
875         @return: True
876         """
877         for pick in self.browse(cr, uid, ids, context=context):
878             todo = []
879             for move in pick.move_lines:
880                 if move.state == 'draft':
881                     self.pool.get('stock.move').action_confirm(cr, uid, [move.id],
882                         context=context)
883                     todo.append(move.id)
884                 elif move.state in ('assigned','confirmed'):
885                     todo.append(move.id)
886             if len(todo):
887                 self.pool.get('stock.move').action_done(cr, uid, todo,
888                         context=context)
889         return True
890
891     def get_currency_id(self, cr, uid, picking):
892         return False
893
894     def _get_partner_to_invoice(self, cr, uid, picking, context=None):
895         """ Gets the partner that will be invoiced
896             Note that this function is inherited in the sale and purchase modules
897             @param picking: object of the picking for which we are selecting the partner to invoice
898             @return: object of the partner to invoice
899         """
900         return picking.partner_id and picking.partner_id.id
901
902     def _get_comment_invoice(self, cr, uid, picking):
903         """
904         @return: comment string for invoice
905         """
906         return picking.note or ''
907
908     def _get_price_unit_invoice(self, cr, uid, move_line, type, context=None):
909         """ Gets price unit for invoice
910         @param move_line: Stock move lines
911         @param type: Type of invoice
912         @return: The price unit for the move line
913         """
914         if context is None:
915             context = {}
916
917         if type in ('in_invoice', 'in_refund'):
918             # Take the user company and pricetype
919             context['currency_id'] = move_line.company_id.currency_id.id
920             amount_unit = move_line.product_id.price_get('standard_price', context=context)[move_line.product_id.id]
921             return amount_unit
922         else:
923             return move_line.product_id.list_price
924
925     def _get_discount_invoice(self, cr, uid, move_line):
926         '''Return the discount for the move line'''
927         return 0.0
928
929     def _get_taxes_invoice(self, cr, uid, move_line, type):
930         """ Gets taxes on invoice
931         @param move_line: Stock move lines
932         @param type: Type of invoice
933         @return: Taxes Ids for the move line
934         """
935         if type in ('in_invoice', 'in_refund'):
936             taxes = move_line.product_id.supplier_taxes_id
937         else:
938             taxes = move_line.product_id.taxes_id
939
940         if move_line.picking_id and move_line.picking_id.partner_id and move_line.picking_id.partner_id.id:
941             return self.pool.get('account.fiscal.position').map_tax(
942                 cr,
943                 uid,
944                 move_line.picking_id.partner_id.property_account_position,
945                 taxes
946             )
947         else:
948             return map(lambda x: x.id, taxes)
949
950     def _get_account_analytic_invoice(self, cr, uid, picking, move_line):
951         return False
952
953     def _invoice_line_hook(self, cr, uid, move_line, invoice_line_id):
954         '''Call after the creation of the invoice line'''
955         return
956
957     def _invoice_hook(self, cr, uid, picking, invoice_id):
958         '''Call after the creation of the invoice'''
959         return
960
961     def _get_invoice_type(self, pick):
962         src_usage = dest_usage = None
963         inv_type = None
964         if pick.invoice_state == '2binvoiced':
965             if pick.move_lines:
966                 src_usage = pick.move_lines[0].location_id.usage
967                 dest_usage = pick.move_lines[0].location_dest_id.usage
968             if pick.type == 'out' and dest_usage == 'supplier':
969                 inv_type = 'in_refund'
970             elif pick.type == 'out' and dest_usage == 'customer':
971                 inv_type = 'out_invoice'
972             elif pick.type == 'in' and src_usage == 'supplier':
973                 inv_type = 'in_invoice'
974             elif pick.type == 'in' and src_usage == 'customer':
975                 inv_type = 'out_refund'
976             else:
977                 inv_type = 'out_invoice'
978         return inv_type
979
980     def _prepare_invoice_group(self, cr, uid, picking, partner, invoice, context=None):
981         """ Builds the dict for grouped invoices
982             @param picking: picking object
983             @param partner: object of the partner to invoice (not used here, but may be usefull if this function is inherited)
984             @param invoice: object of the invoice that we are updating
985             @return: dict that will be used to update the invoice
986         """
987         comment = self._get_comment_invoice(cr, uid, picking)
988         return {
989             'name': (invoice.name or '') + ', ' + (picking.name or ''),
990             'origin': (invoice.origin or '') + ', ' + (picking.name or '') + (picking.origin and (':' + picking.origin) or ''),
991             'comment': (comment and (invoice.comment and invoice.comment + "\n" + comment or comment)) or (invoice.comment and invoice.comment or ''),
992             'date_invoice': context.get('date_inv', False),
993             'user_id': uid,
994         }
995
996     def _prepare_invoice(self, cr, uid, picking, partner, inv_type, journal_id, context=None):
997         """ Builds the dict containing the values for the invoice
998             @param picking: picking object
999             @param partner: object of the partner to invoice
1000             @param inv_type: type of the invoice ('out_invoice', 'in_invoice', ...)
1001             @param journal_id: ID of the accounting journal
1002             @return: dict that will be used to create the invoice object
1003         """
1004         if isinstance(partner, int):
1005             partner = self.pool.get('res.partner').browse(cr, uid, partner, context=context)
1006         if inv_type in ('out_invoice', 'out_refund'):
1007             account_id = partner.property_account_receivable.id
1008         else:
1009             account_id = partner.property_account_payable.id
1010         comment = self._get_comment_invoice(cr, uid, picking)
1011         invoice_vals = {
1012             'name': picking.name,
1013             'origin': (picking.name or '') + (picking.origin and (':' + picking.origin) or ''),
1014             'type': inv_type,
1015             'account_id': account_id,
1016             'partner_id': partner.id,
1017             'comment': comment,
1018             'payment_term': partner.property_payment_term and partner.property_payment_term.id or False,
1019             'fiscal_position': partner.property_account_position.id,
1020             'date_invoice': context.get('date_inv', False),
1021             'company_id': picking.company_id.id,
1022             'user_id': uid,
1023         }
1024         cur_id = self.get_currency_id(cr, uid, picking)
1025         if cur_id:
1026             invoice_vals['currency_id'] = cur_id
1027         if journal_id:
1028             invoice_vals['journal_id'] = journal_id
1029         return invoice_vals
1030
1031     def _prepare_invoice_line(self, cr, uid, group, picking, move_line, invoice_id,
1032         invoice_vals, context=None):
1033         """ Builds the dict containing the values for the invoice line
1034             @param group: True or False
1035             @param picking: picking object
1036             @param: move_line: move_line object
1037             @param: invoice_id: ID of the related invoice
1038             @param: invoice_vals: dict used to created the invoice
1039             @return: dict that will be used to create the invoice line
1040         """
1041         if group:
1042             name = (picking.name or '') + '-' + move_line.name
1043         else:
1044             name = move_line.name
1045         origin = move_line.picking_id.name or ''
1046         if move_line.picking_id.origin:
1047             origin += ':' + move_line.picking_id.origin
1048
1049         if invoice_vals['type'] in ('out_invoice', 'out_refund'):
1050             account_id = move_line.product_id.product_tmpl_id.\
1051                     property_account_income.id
1052             if not account_id:
1053                 account_id = move_line.product_id.categ_id.\
1054                         property_account_income_categ.id
1055         else:
1056             account_id = move_line.product_id.product_tmpl_id.\
1057                     property_account_expense.id
1058             if not account_id:
1059                 account_id = move_line.product_id.categ_id.\
1060                         property_account_expense_categ.id
1061         if invoice_vals['fiscal_position']:
1062             fp_obj = self.pool.get('account.fiscal.position')
1063             fiscal_position = fp_obj.browse(cr, uid, invoice_vals['fiscal_position'], context=context)
1064             account_id = fp_obj.map_account(cr, uid, fiscal_position, account_id)
1065         # set UoS if it's a sale and the picking doesn't have one
1066         uos_id = move_line.product_uos and move_line.product_uos.id or False
1067         if not uos_id and invoice_vals['type'] in ('out_invoice', 'out_refund'):
1068             uos_id = move_line.product_uom.id
1069
1070         return {
1071             'name': name,
1072             'origin': origin,
1073             'invoice_id': invoice_id,
1074             'uos_id': uos_id,
1075             'product_id': move_line.product_id.id,
1076             'account_id': account_id,
1077             'price_unit': self._get_price_unit_invoice(cr, uid, move_line, invoice_vals['type']),
1078             'discount': self._get_discount_invoice(cr, uid, move_line),
1079             'quantity': move_line.product_uos_qty or move_line.product_qty,
1080             'invoice_line_tax_id': [(6, 0, self._get_taxes_invoice(cr, uid, move_line, invoice_vals['type']))],
1081             'account_analytic_id': self._get_account_analytic_invoice(cr, uid, picking, move_line),
1082         }
1083
1084     def action_invoice_create(self, cr, uid, ids, journal_id=False,
1085             group=False, type='out_invoice', context=None):
1086         """ Creates invoice based on the invoice state selected for picking.
1087         @param journal_id: Id of journal
1088         @param group: Whether to create a group invoice or not
1089         @param type: Type invoice to be created
1090         @return: Ids of created invoices for the pickings
1091         """
1092         if context is None:
1093             context = {}
1094
1095         invoice_obj = self.pool.get('account.invoice')
1096         invoice_line_obj = self.pool.get('account.invoice.line')
1097         partner_obj = self.pool.get('res.partner')
1098         invoices_group = {}
1099         res = {}
1100         inv_type = type
1101         for picking in self.browse(cr, uid, ids, context=context):
1102             if picking.invoice_state != '2binvoiced':
1103                 continue
1104             partner = self._get_partner_to_invoice(cr, uid, picking, context=context)
1105             if isinstance(partner, int):
1106                 partner = partner_obj.browse(cr, uid, [partner], context=context)[0]
1107             if not partner:
1108                 raise osv.except_osv(_('Error, no partner !'),
1109                     _('Please put a partner on the picking list if you want to generate invoice.'))
1110
1111             if not inv_type:
1112                 inv_type = self._get_invoice_type(picking)
1113
1114             if group and partner.id in invoices_group:
1115                 invoice_id = invoices_group[partner.id]
1116                 invoice = invoice_obj.browse(cr, uid, invoice_id)
1117                 invoice_vals_group = self._prepare_invoice_group(cr, uid, picking, partner, invoice, context=context)
1118                 invoice_obj.write(cr, uid, [invoice_id], invoice_vals_group, context=context)
1119             else:
1120                 invoice_vals = self._prepare_invoice(cr, uid, picking, partner, inv_type, journal_id, context=context)
1121                 invoice_id = invoice_obj.create(cr, uid, invoice_vals, context=context)
1122                 invoices_group[partner.id] = invoice_id
1123             res[picking.id] = invoice_id
1124             for move_line in picking.move_lines:
1125                 if move_line.state == 'cancel':
1126                     continue
1127                 vals = self._prepare_invoice_line(cr, uid, group, picking, move_line,
1128                                 invoice_id, invoice_vals, context=context)
1129                 if vals:
1130                     invoice_line_id = invoice_line_obj.create(cr, uid, vals, context=context)
1131                     self._invoice_line_hook(cr, uid, move_line, invoice_line_id)
1132
1133             invoice_obj.button_compute(cr, uid, [invoice_id], context=context,
1134                     set_total=(inv_type in ('in_invoice', 'in_refund')))
1135             self.write(cr, uid, [picking.id], {
1136                 'invoice_state': 'invoiced',
1137                 }, context=context)
1138             self._invoice_hook(cr, uid, picking, invoice_id)
1139         self.write(cr, uid, res.keys(), {
1140             'invoice_state': 'invoiced',
1141             }, context=context)
1142         return res
1143
1144     def test_done(self, cr, uid, ids, context=None):
1145         """ Test whether the move lines are done or not.
1146         @return: True or False
1147         """
1148         ok = False
1149         for pick in self.browse(cr, uid, ids, context=context):
1150             if not pick.move_lines:
1151                 return True
1152             for move in pick.move_lines:
1153                 if move.state not in ('cancel','done'):
1154                     return False
1155                 if move.state=='done':
1156                     ok = True
1157         return ok
1158
1159     def test_cancel(self, cr, uid, ids, context=None):
1160         """ Test whether the move lines are canceled or not.
1161         @return: True or False
1162         """
1163         for pick in self.browse(cr, uid, ids, context=context):
1164             for move in pick.move_lines:
1165                 if move.state not in ('cancel',):
1166                     return False
1167         return True
1168
1169     def allow_cancel(self, cr, uid, ids, context=None):
1170         for pick in self.browse(cr, uid, ids, context=context):
1171             if not pick.move_lines:
1172                 return True
1173             for move in pick.move_lines:
1174                 if move.state == 'done':
1175                     raise osv.except_osv(_('Error!'), _('You cannot cancel the picking as some moves have been done. You should cancel the picking lines.'))
1176         return True
1177
1178     def unlink(self, cr, uid, ids, context=None):
1179         move_obj = self.pool.get('stock.move')
1180         if context is None:
1181             context = {}
1182         for pick in self.browse(cr, uid, ids, context=context):
1183             if pick.state in ['done','cancel']:
1184                 raise osv.except_osv(_('Error!'), _('You cannot remove the picking which is in %s state!')%(pick.state,))
1185             else:
1186                 ids2 = [move.id for move in pick.move_lines]
1187                 ctx = context.copy()
1188                 ctx.update({'call_unlink':True})
1189                 if pick.state != 'draft':
1190                     #Cancelling the move in order to affect Virtual stock of product
1191                     move_obj.action_cancel(cr, uid, ids2, ctx)
1192                 #Removing the move
1193                 move_obj.unlink(cr, uid, ids2, ctx)
1194
1195         return super(stock_picking, self).unlink(cr, uid, ids, context=context)
1196
1197     # FIXME: needs refactoring, this code is partially duplicated in stock_move.do_partial()!
1198     def do_partial(self, cr, uid, ids, partial_datas, context=None):
1199         """ Makes partial picking and moves done.
1200         @param partial_datas : Dictionary containing details of partial picking
1201                           like partner_id, partner_id, delivery_date,
1202                           delivery moves with product_id, product_qty, uom
1203         @return: Dictionary of values
1204         """
1205         if context is None:
1206             context = {}
1207         else:
1208             context = dict(context)
1209         res = {}
1210         move_obj = self.pool.get('stock.move')
1211         product_obj = self.pool.get('product.product')
1212         currency_obj = self.pool.get('res.currency')
1213         uom_obj = self.pool.get('product.uom')
1214         sequence_obj = self.pool.get('ir.sequence')
1215         wf_service = netsvc.LocalService("workflow")
1216         for pick in self.browse(cr, uid, ids, context=context):
1217             new_picking = None
1218             complete, too_many, too_few = [], [], []
1219             move_product_qty, prodlot_ids, product_avail, partial_qty, product_uoms = {}, {}, {}, {}, {}
1220             for move in pick.move_lines:
1221                 if move.state in ('done', 'cancel'):
1222                     continue
1223                 partial_data = partial_datas.get('move%s'%(move.id), {})
1224                 product_qty = partial_data.get('product_qty',0.0)
1225                 move_product_qty[move.id] = product_qty
1226                 product_uom = partial_data.get('product_uom',False)
1227                 product_price = partial_data.get('product_price',0.0)
1228                 product_currency = partial_data.get('product_currency',False)
1229                 prodlot_id = partial_data.get('prodlot_id')
1230                 prodlot_ids[move.id] = prodlot_id
1231                 product_uoms[move.id] = product_uom
1232                 partial_qty[move.id] = uom_obj._compute_qty(cr, uid, product_uoms[move.id], product_qty, move.product_uom.id)
1233                 if move.product_qty == partial_qty[move.id]:
1234                     complete.append(move)
1235                 elif move.product_qty > partial_qty[move.id]:
1236                     too_few.append(move)
1237                 else:
1238                     too_many.append(move)
1239
1240                 # Average price computation
1241                 if (pick.type == 'in') and (move.product_id.cost_method == 'average'):
1242                     product = product_obj.browse(cr, uid, move.product_id.id)
1243                     move_currency_id = move.company_id.currency_id.id
1244                     context['currency_id'] = move_currency_id
1245                     qty = uom_obj._compute_qty(cr, uid, product_uom, product_qty, product.uom_id.id)
1246
1247                     if product.id in product_avail:
1248                         product_avail[product.id] += qty
1249                     else:
1250                         product_avail[product.id] = product.qty_available
1251
1252                     if qty > 0:
1253                         new_price = currency_obj.compute(cr, uid, product_currency,
1254                                 move_currency_id, product_price)
1255                         new_price = uom_obj._compute_price(cr, uid, product_uom, new_price,
1256                                 product.uom_id.id)
1257                         if product.qty_available <= 0:
1258                             new_std_price = new_price
1259                         else:
1260                             # Get the standard price
1261                             amount_unit = product.price_get('standard_price', context=context)[product.id]
1262                             new_std_price = ((amount_unit * product_avail[product.id])\
1263                                 + (new_price * qty))/(product_avail[product.id] + qty)
1264                         # Write the field according to price type field
1265                         product_obj.write(cr, uid, [product.id], {'standard_price': new_std_price})
1266
1267                         # Record the values that were chosen in the wizard, so they can be
1268                         # used for inventory valuation if real-time valuation is enabled.
1269                         move_obj.write(cr, uid, [move.id],
1270                                 {'price_unit': product_price,
1271                                  'price_currency_id': product_currency})
1272
1273
1274             for move in too_few:
1275                 product_qty = move_product_qty[move.id]
1276                 if not new_picking:
1277                     new_picking = self.copy(cr, uid, pick.id,
1278                             {
1279                                 'name': sequence_obj.get(cr, uid, 'stock.picking.%s'%(pick.type)),
1280                                 'move_lines' : [],
1281                                 'state':'draft',
1282                             })
1283                 if product_qty != 0:
1284                     defaults = {
1285                             'product_qty' : product_qty,
1286                             'product_uos_qty': product_qty, #TODO: put correct uos_qty
1287                             'picking_id' : new_picking,
1288                             'state': 'assigned',
1289                             'move_dest_id': False,
1290                             'price_unit': move.price_unit,
1291                             'product_uom': product_uoms[move.id]
1292                     }
1293                     prodlot_id = prodlot_ids[move.id]
1294                     if prodlot_id:
1295                         defaults.update(prodlot_id=prodlot_id)
1296                     move_obj.copy(cr, uid, move.id, defaults)
1297                 move_obj.write(cr, uid, [move.id],
1298                         {
1299                             'product_qty' : move.product_qty - partial_qty[move.id],
1300                             'product_uos_qty': move.product_qty - partial_qty[move.id], #TODO: put correct uos_qty
1301
1302                         })
1303
1304             if new_picking:
1305                 move_obj.write(cr, uid, [c.id for c in complete], {'picking_id': new_picking})
1306             for move in complete:
1307                 defaults = {'product_uom': product_uoms[move.id], 'product_qty': move_product_qty[move.id]}
1308                 if prodlot_ids.get(move.id):
1309                     defaults.update({'prodlot_id': prodlot_ids[move.id]})
1310                 move_obj.write(cr, uid, [move.id], defaults)
1311             for move in too_many:
1312                 product_qty = move_product_qty[move.id]
1313                 defaults = {
1314                     'product_qty' : product_qty,
1315                     'product_uos_qty': product_qty, #TODO: put correct uos_qty
1316                     'product_uom': product_uoms[move.id]
1317                 }
1318                 prodlot_id = prodlot_ids.get(move.id)
1319                 if prodlot_ids.get(move.id):
1320                     defaults.update(prodlot_id=prodlot_id)
1321                 if new_picking:
1322                     defaults.update(picking_id=new_picking)
1323                 move_obj.write(cr, uid, [move.id], defaults)
1324
1325             # At first we confirm the new picking (if necessary)
1326             if new_picking:
1327                 wf_service.trg_validate(uid, 'stock.picking', new_picking, 'button_confirm', cr)
1328                 # Then we finish the good picking
1329                 self.write(cr, uid, [pick.id], {'backorder_id': new_picking})
1330                 self.action_move(cr, uid, [new_picking])
1331                 wf_service.trg_validate(uid, 'stock.picking', new_picking, 'button_done', cr)
1332                 wf_service.trg_write(uid, 'stock.picking', pick.id, cr)
1333                 delivered_pack_id = new_picking
1334                 back_order_name = self.browse(cr, uid, delivered_pack_id, context=context).name
1335                 self.back_order_send_note(cr, uid, ids, back_order_name, context)
1336             else:
1337                 self.action_move(cr, uid, [pick.id])
1338                 wf_service.trg_validate(uid, 'stock.picking', pick.id, 'button_done', cr)
1339                 delivered_pack_id = pick.id
1340                 self.ship_done_send_note(cr, uid, ids, context)
1341
1342             delivered_pack = self.browse(cr, uid, delivered_pack_id, context=context)
1343             res[pick.id] = {'delivered_picking': delivered_pack.id or False}
1344
1345         return res
1346
1347     def log_picking(self, cr, uid, ids, context=None):
1348         """ This function will create log messages for picking.
1349         @param cr: the database cursor
1350         @param uid: the current user's ID for security checks,
1351         @param ids: List of Picking Ids
1352         @param context: A standard dictionary for contextual values
1353         """
1354         if context is None:
1355             context = {}
1356         data_obj = self.pool.get('ir.model.data')
1357         for pick in self.browse(cr, uid, ids, context=context):
1358             msg=''
1359             if pick.auto_picking:
1360                 continue
1361             type_list = {
1362                 'out':_("Delivery Order"),
1363                 'in':_('Reception'),
1364                 'internal': _('Internal picking'),
1365             }
1366             view_list = {
1367                 'out': 'view_picking_out_form',
1368                 'in': 'view_picking_in_form',
1369                 'internal': 'view_picking_form',
1370             }
1371             message = type_list.get(pick.type, _('Document')) + " '" + (pick.name or '?') + "' "
1372             if pick.min_date:
1373                 msg= _(' for the ')+ datetime.strptime(pick.min_date, '%Y-%m-%d %H:%M:%S').strftime('%m/%d/%Y')
1374             state_list = {
1375                 'confirmed': _('is scheduled %s.') % msg,
1376                 'assigned': _('is ready to process.'),
1377                 'cancel': _('is cancelled.'),
1378                 'done': _('is done.'),
1379                 'auto': _('is waiting.'),
1380                 'draft': _('is in draft state.'),
1381             }
1382             res = data_obj.get_object_reference(cr, uid, 'stock', view_list.get(pick.type, 'view_picking_form'))
1383             context.update({'view_id': res and res[1] or False})
1384             message += state_list[pick.state]
1385         return True
1386
1387     # -----------------------------------------
1388     # OpenChatter methods and notifications
1389     # -----------------------------------------
1390
1391     def _get_document_type(self, cr, uid, obj, context=None):
1392         type_dict = {
1393                 'out': _('Delivery order'),
1394                 'in': _('Shipment'),
1395                 'internal': _('Internal picking'),
1396         }
1397         return type_dict.get(obj.type, _('Picking'))
1398
1399     def create_send_note(self, cr, uid, ids, context=None):
1400         for obj in self.browse(cr, uid, ids, context=context):
1401             self.message_post(cr, uid, [obj.id], body=_("%s has been <b>created</b>.") % (self._get_document_type(cr, uid, obj, context=context)), context=context)
1402
1403     def confirm_send_note(self, cr, uid, ids, context=None):
1404         for obj in self.browse(cr, uid, ids, context=context):
1405             self.message_post(cr, uid, [obj.id], body=_("%s has been <b>confirmed</b>.") % (self._get_document_type(cr, uid, obj, context=context)), context=context)
1406
1407     def scrap_send_note(self, cr, uid, ids, quantity, uom, name, context=None):
1408         return self.message_post(cr, uid, ids, body= _("%s %s %s has been <b>moved to</b> scrap.") % (quantity, uom, name), context=context)
1409
1410     def back_order_send_note(self, cr, uid, ids, back_name, context=None):
1411         return self.message_post(cr, uid, ids, body=_("Back order <em>%s</em> has been <b>created</b>.") % (back_name), context=context)
1412
1413     def ship_done_send_note(self, cr, uid, ids, context=None):
1414         type_dict = {
1415                 'out': _("Products have been <b>delivered</b>."),
1416                 'in': _("Products have been <b>received</b>."),
1417                 'internal': _("Products have been <b>moved</b>."),
1418         }
1419         for obj in self.browse(cr, uid, ids, context=context):
1420             self.message_post(cr, uid, [obj.id], body=type_dict.get(obj.type, _('Products have been moved.')), context=context)
1421
1422     def ship_cancel_send_note(self, cr, uid, ids, context=None):
1423         for obj in self.browse(cr, uid, ids, context=context):
1424             self.message_post(cr, uid, [obj.id], body=_("%s has been <b>cancelled</b>.") % (self._get_document_type(cr, uid, obj, context=context)), context=context)
1425
1426
1427 stock_picking()
1428
1429 class stock_production_lot(osv.osv):
1430
1431     def name_get(self, cr, uid, ids, context=None):
1432         if not ids:
1433             return []
1434         reads = self.read(cr, uid, ids, ['name', 'prefix', 'ref'], context)
1435         res = []
1436         for record in reads:
1437             name = record['name']
1438             prefix = record['prefix']
1439             if prefix:
1440                 name = prefix + '/' + name
1441             if record['ref']:
1442                 name = '%s [%s]' % (name, record['ref'])
1443             res.append((record['id'], name))
1444         return res
1445
1446     def name_search(self, cr, uid, name, args=None, operator='ilike', context=None, limit=100):
1447         args = args or []
1448         ids = []
1449         if name:
1450             ids = self.search(cr, uid, [('prefix', '=', name)] + args, limit=limit, context=context)
1451             if not ids:
1452                 ids = self.search(cr, uid, [('name', operator, name)] + args, limit=limit, context=context)
1453         else:
1454             ids = self.search(cr, uid, args, limit=limit, context=context)
1455         return self.name_get(cr, uid, ids, context)
1456
1457     _name = 'stock.production.lot'
1458     _description = 'Serial Number'
1459
1460     def _get_stock(self, cr, uid, ids, field_name, arg, context=None):
1461         """ Gets stock of products for locations
1462         @return: Dictionary of values
1463         """
1464         if context is None:
1465             context = {}
1466         if 'location_id' not in context:
1467             locations = self.pool.get('stock.location').search(cr, uid, [('usage', '=', 'internal')], context=context)
1468         else:
1469             locations = context['location_id'] and [context['location_id']] or []
1470
1471         if isinstance(ids, (int, long)):
1472             ids = [ids]
1473
1474         res = {}.fromkeys(ids, 0.0)
1475         if locations:
1476             cr.execute('''select
1477                     prodlot_id,
1478                     sum(qty)
1479                 from
1480                     stock_report_prodlots
1481                 where
1482                     location_id IN %s and prodlot_id IN %s group by prodlot_id''',(tuple(locations),tuple(ids),))
1483             res.update(dict(cr.fetchall()))
1484
1485         return res
1486
1487     def _stock_search(self, cr, uid, obj, name, args, context=None):
1488         """ Searches Ids of products
1489         @return: Ids of locations
1490         """
1491         locations = self.pool.get('stock.location').search(cr, uid, [('usage', '=', 'internal')])
1492         cr.execute('''select
1493                 prodlot_id,
1494                 sum(qty)
1495             from
1496                 stock_report_prodlots
1497             where
1498                 location_id IN %s group by prodlot_id
1499             having  sum(qty) '''+ str(args[0][1]) + str(args[0][2]),(tuple(locations),))
1500         res = cr.fetchall()
1501         ids = [('id', 'in', map(lambda x: x[0], res))]
1502         return ids
1503
1504     _columns = {
1505         'name': fields.char('Serial Number', size=64, required=True, help="Unique Serial Number, will be displayed as: PREFIX/SERIAL [INT_REF]"),
1506         'ref': fields.char('Internal Reference', size=256, help="Internal reference number in case it differs from the manufacturer's serial number"),
1507         'prefix': fields.char('Prefix', size=64, help="Optional prefix to prepend when displaying this serial number: PREFIX/SERIAL [INT_REF]"),
1508         'product_id': fields.many2one('product.product', 'Product', required=True, domain=[('type', '<>', 'service')]),
1509         'date': fields.datetime('Creation Date', required=True),
1510         'stock_available': fields.function(_get_stock, fnct_search=_stock_search, type="float", string="Available", select=True,
1511             help="Current quantity of products with this Serial Number available in company warehouses",
1512             digits_compute=dp.get_precision('Product Unit of Measure')),
1513         'revisions': fields.one2many('stock.production.lot.revision', 'lot_id', 'Revisions'),
1514         'company_id': fields.many2one('res.company', 'Company', select=True),
1515         'move_ids': fields.one2many('stock.move', 'prodlot_id', 'Moves for this serial number', readonly=True),
1516     }
1517     _defaults = {
1518         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
1519         'name': lambda x, y, z, c: x.pool.get('ir.sequence').get(y, z, 'stock.lot.serial'),
1520         'product_id': lambda x, y, z, c: c.get('product_id', False),
1521     }
1522     _sql_constraints = [
1523         ('name_ref_uniq', 'unique (name, ref)', 'The combination of serial number and internal reference must be unique !'),
1524     ]
1525     def action_traceability(self, cr, uid, ids, context=None):
1526         """ It traces the information of a product
1527         @param self: The object pointer.
1528         @param cr: A database cursor
1529         @param uid: ID of the user currently logged in
1530         @param ids: List of IDs selected
1531         @param context: A standard dictionary
1532         @return: A dictionary of values
1533         """
1534         value=self.pool.get('action.traceability').action_traceability(cr,uid,ids,context)
1535         return value
1536 stock_production_lot()
1537
1538 class stock_production_lot_revision(osv.osv):
1539     _name = 'stock.production.lot.revision'
1540     _description = 'Serial Number Revision'
1541
1542     _columns = {
1543         'name': fields.char('Revision Name', size=64, required=True),
1544         'description': fields.text('Description'),
1545         'date': fields.date('Revision Date'),
1546         'indice': fields.char('Revision Number', size=16),
1547         'author_id': fields.many2one('res.users', 'Author'),
1548         'lot_id': fields.many2one('stock.production.lot', 'Serial Number', select=True, ondelete='cascade'),
1549         'company_id': fields.related('lot_id','company_id',type='many2one',relation='res.company',string='Company', store=True, readonly=True),
1550     }
1551
1552     _defaults = {
1553         'author_id': lambda x, y, z, c: z,
1554         'date': fields.date.context_today,
1555     }
1556
1557 stock_production_lot_revision()
1558
1559 # ----------------------------------------------------
1560 # Move
1561 # ----------------------------------------------------
1562
1563 #
1564 # Fields:
1565 #   location_dest_id is only used for predicting futur stocks
1566 #
1567 class stock_move(osv.osv):
1568
1569     def _getSSCC(self, cr, uid, context=None):
1570         cr.execute('select id from stock_tracking where create_uid=%s order by id desc limit 1', (uid,))
1571         res = cr.fetchone()
1572         return (res and res[0]) or False
1573
1574     _name = "stock.move"
1575     _description = "Stock Move"
1576     _order = 'date_expected desc, id'
1577     _log_create = False
1578
1579     def action_partial_move(self, cr, uid, ids, context=None):
1580         if context is None: context = {}
1581         if context.get('active_model') != self._name:
1582             context.update(active_ids=ids, active_model=self._name)
1583         partial_id = self.pool.get("stock.partial.move").create(
1584             cr, uid, {}, context=context)
1585         return {
1586             'name':_("Products to Process"),
1587             'view_mode': 'form',
1588             'view_id': False,
1589             'view_type': 'form',
1590             'res_model': 'stock.partial.move',
1591             'res_id': partial_id,
1592             'type': 'ir.actions.act_window',
1593             'nodestroy': True,
1594             'target': 'new',
1595             'domain': '[]',
1596             'context': context
1597         }
1598
1599
1600     def name_get(self, cr, uid, ids, context=None):
1601         res = []
1602         for line in self.browse(cr, uid, ids, context=context):
1603             name = line.location_id.name+' > '+line.location_dest_id.name
1604             # optional prefixes
1605             if line.product_id.code:
1606                 name = line.product_id.code + ': ' + name
1607             if line.picking_id.origin:
1608                 name = line.picking_id.origin + '/ ' + name
1609             res.append((line.id, name))
1610         return res
1611
1612     def _check_tracking(self, cr, uid, ids, context=None):
1613         """ Checks if serial number is assigned to stock move or not.
1614         @return: True or False
1615         """
1616         for move in self.browse(cr, uid, ids, context=context):
1617             if not move.prodlot_id and \
1618                (move.state == 'done' and \
1619                ( \
1620                    (move.product_id.track_production and move.location_id.usage == 'production') or \
1621                    (move.product_id.track_production and move.location_dest_id.usage == 'production') or \
1622                    (move.product_id.track_incoming and move.location_id.usage == 'supplier') or \
1623                    (move.product_id.track_outgoing and move.location_dest_id.usage == 'customer') \
1624                )):
1625                 return False
1626         return True
1627
1628     def _check_product_lot(self, cr, uid, ids, context=None):
1629         """ Checks whether move is done or not and production lot is assigned to that move.
1630         @return: True or False
1631         """
1632         for move in self.browse(cr, uid, ids, context=context):
1633             if move.prodlot_id and move.state == 'done' and (move.prodlot_id.product_id.id != move.product_id.id):
1634                 return False
1635         return True
1636
1637     _columns = {
1638         'name': fields.char('Name', size=250, required=True, select=True),
1639         'priority': fields.selection([('0', 'Not urgent'), ('1', 'Urgent')], 'Priority'),
1640         'create_date': fields.datetime('Creation Date', readonly=True, select=True),
1641         '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)]}),
1642         'date_expected': fields.datetime('Scheduled Date', states={'done': [('readonly', True)]},required=True, select=True, help="Scheduled date for the processing of this move"),
1643         'product_id': fields.many2one('product.product', 'Product', required=True, select=True, domain=[('type','<>','service')],states={'done': [('readonly', True)]}),
1644
1645         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product Unit of Measure'), required=True,states={'done': [('readonly', True)]}),
1646         'product_uom': fields.many2one('product.uom', 'Unit of Measure', required=True,states={'done': [('readonly', True)]}),
1647         'product_uos_qty': fields.float('Quantity (UOS)', digits_compute=dp.get_precision('Product Unit of Measure'), states={'done': [('readonly', True)]}),
1648         'product_uos': fields.many2one('product.uom', 'Product UOS', states={'done': [('readonly', True)]}),
1649         'product_packaging': fields.many2one('product.packaging', 'Packaging', help="It specifies attributes of packaging like type, quantity of packaging,etc."),
1650
1651         '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."),
1652         '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."),
1653         '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"),
1654
1655         'prodlot_id': fields.many2one('stock.production.lot', 'Serial Number', states={'done': [('readonly', True)]}, help="Serial number is used to put a serial number on the production", select=True),
1656         'tracking_id': fields.many2one('stock.tracking', 'Pack', select=True, states={'done': [('readonly', True)]}, help="Logistical shipping unit: pallet, box, pack ..."),
1657
1658         'auto_validate': fields.boolean('Auto Validate'),
1659
1660         'move_dest_id': fields.many2one('stock.move', 'Destination Move', help="Optional: next stock move when chaining them", select=True),
1661         'move_history_ids': fields.many2many('stock.move', 'stock_move_history_ids', 'parent_id', 'child_id', 'Move History (child moves)'),
1662         'move_history_ids2': fields.many2many('stock.move', 'stock_move_history_ids', 'child_id', 'parent_id', 'Move History (parent moves)'),
1663         'picking_id': fields.many2one('stock.picking', 'Reference', select=True,states={'done': [('readonly', True)]}),
1664         'note': fields.text('Notes'),
1665         'state': fields.selection([('draft', 'New'),
1666                                    ('cancel', 'Cancelled'),
1667                                    ('waiting', 'Waiting Another Move'),
1668                                    ('confirmed', 'Waiting Availability'),
1669                                    ('assigned', 'Available'),
1670                                    ('done', 'Done'),
1671                                    ], 'Status', readonly=True, select=True,
1672                  help= "* New: When the stock move is created and not yet confirmed.\n"\
1673                        "* Waiting Another Move: This state can be seen when a move is waiting for another one, for example in a chained flow.\n"\
1674                        "* 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"\
1675                        "* Available: When products are reserved, it is set to \'Available\'.\n"\
1676                        "* Done: When the shipment is processed, the state is \'Done\'."),
1677         'price_unit': fields.float('Unit Price', digits_compute= dp.get_precision('Account'), help="Technical field used to record the product cost set by the user during a picking confirmation (when average price costing method is used)"),
1678         'price_currency_id': fields.many2one('res.currency', 'Currency for average price', help="Technical field used to record the currency chosen by the user during a picking confirmation (when average price costing method is used)"),
1679         'company_id': fields.many2one('res.company', 'Company', required=True, select=True),
1680         'backorder_id': fields.related('picking_id','backorder_id',type='many2one', relation="stock.picking", string="Back Order", select=True),
1681         'origin': fields.related('picking_id','origin',type='char', size=64, relation="stock.picking", string="Source", store=True),
1682
1683         # used for colors in tree views:
1684         'scrapped': fields.related('location_dest_id','scrap_location',type='boolean',relation='stock.location',string='Scrapped', readonly=True),
1685         'type': fields.related('picking_id', 'type', type='selection', selection=[('out', 'Sending Goods'), ('in', 'Getting Goods'), ('internal', 'Internal')], string='Shipping Type'),
1686     }
1687
1688     def _check_location(self, cr, uid, ids, context=None):
1689         for record in self.browse(cr, uid, ids, context=context):
1690             if (record.state=='done') and (record.location_id.usage == 'view'):
1691                 raise osv.except_osv(_('Error'), _('You cannot move product %s from a location of type view %s.')% (record.product_id.name, record.location_id.name))
1692             if (record.state=='done') and (record.location_dest_id.usage == 'view' ):
1693                 raise osv.except_osv(_('Error'), _('You cannot move product %s to a location of type view %s.')% (record.product_id.name, record.location_dest_id.name))
1694         return True
1695
1696     _constraints = [
1697         (_check_tracking,
1698             'You must assign a serial number for this product.',
1699             ['prodlot_id']),
1700         (_check_location, 'You cannot move products from or to a location of the type view.',
1701             ['location_id','location_dest_id']),
1702         (_check_product_lot,
1703             'You try to assign a lot which is not from the same product.',
1704             ['prodlot_id'])]
1705
1706     def _default_location_destination(self, cr, uid, context=None):
1707         """ Gets default address of partner for destination location
1708         @return: Address id or False
1709         """
1710         mod_obj = self.pool.get('ir.model.data')
1711         picking_type = context.get('picking_type')
1712         location_id = False
1713         if context is None:
1714             context = {}
1715         if context.get('move_line', []):
1716             if context['move_line'][0]:
1717                 if isinstance(context['move_line'][0], (tuple, list)):
1718                     location_id = context['move_line'][0][2] and context['move_line'][0][2].get('location_dest_id',False)
1719                 else:
1720                     move_list = self.pool.get('stock.move').read(cr, uid, context['move_line'][0], ['location_dest_id'])
1721                     location_id = move_list and move_list['location_dest_id'][0] or False
1722         elif context.get('address_out_id', False):
1723             property_out = self.pool.get('res.partner').browse(cr, uid, context['address_out_id'], context).property_stock_customer
1724             location_id = property_out and property_out.id or False
1725         else:
1726             location_xml_id = False
1727             if picking_type == 'in':
1728                 location_xml_id = 'stock_location_stock'
1729             elif picking_type == 'out':
1730                 location_xml_id = 'stock_location_customers'
1731             if location_xml_id:
1732                 location_model, location_id = mod_obj.get_object_reference(cr, uid, 'stock', location_xml_id)
1733         return location_id
1734
1735     def _default_location_source(self, cr, uid, context=None):
1736         """ Gets default address of partner for source location
1737         @return: Address id or False
1738         """
1739         mod_obj = self.pool.get('ir.model.data')
1740         picking_type = context.get('picking_type')
1741         location_id = False
1742
1743         if context is None:
1744             context = {}
1745         if context.get('move_line', []):
1746             try:
1747                 location_id = context['move_line'][0][2]['location_id']
1748             except:
1749                 pass
1750         elif context.get('address_in_id', False):
1751             part_obj_add = self.pool.get('res.partner').browse(cr, uid, context['address_in_id'], context=context)
1752             if part_obj_add:
1753                 location_id = part_obj_add.property_stock_supplier.id
1754         else:
1755             location_xml_id = False
1756             if picking_type == 'in':
1757                 location_xml_id = 'stock_location_suppliers'
1758             elif picking_type == 'out':
1759                 location_xml_id = 'stock_location_stock'
1760             if location_xml_id:
1761                 location_model, location_id = mod_obj.get_object_reference(cr, uid, 'stock', location_xml_id)
1762         return location_id
1763
1764     def _default_destination_address(self, cr, uid, context=None):
1765         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
1766         return user.company_id.partner_id.id
1767
1768     def _default_move_type(self, cr, uid, context=None):
1769         """ Gets default type of move
1770         @return: type
1771         """
1772         if context is None:
1773             context = {}
1774         picking_type = context.get('picking_type')
1775         type = 'internal'
1776         if picking_type == 'in':
1777             type = 'in'
1778         elif picking_type == 'out':
1779             type = 'out'
1780         return type
1781
1782     _defaults = {
1783         'location_id': _default_location_source,
1784         'location_dest_id': _default_location_destination,
1785         'partner_id': _default_destination_address,
1786         'type': _default_move_type,
1787         'state': 'draft',
1788         'priority': '1',
1789         'product_qty': 1.0,
1790         'scrapped' :  False,
1791         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
1792         'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.move', context=c),
1793         'date_expected': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
1794     }
1795
1796     def write(self, cr, uid, ids, vals, context=None):
1797         if isinstance(ids, (int, long)):
1798             ids = [ids]
1799         if uid != 1:
1800             frozen_fields = set(['product_qty', 'product_uom', 'product_uos_qty', 'product_uos', 'location_id', 'location_dest_id', 'product_id'])
1801             for move in self.browse(cr, uid, ids, context=context):
1802                 if move.state == 'done':
1803                     if frozen_fields.intersection(vals):
1804                         raise osv.except_osv(_('Operation forbidden !'),
1805                                              _('Quantities, Unit of Measures, Products and Locations cannot be modified on stock moves that have already been processed (except by the Administrator).'))
1806         return  super(stock_move, self).write(cr, uid, ids, vals, context=context)
1807
1808     def copy(self, cr, uid, id, default=None, context=None):
1809         if default is None:
1810             default = {}
1811         default = default.copy()
1812         default.update({'move_history_ids2': [], 'move_history_ids': []})
1813         return super(stock_move, self).copy(cr, uid, id, default, context=context)
1814
1815     def _auto_init(self, cursor, context=None):
1816         res = super(stock_move, self)._auto_init(cursor, context=context)
1817         cursor.execute('SELECT indexname \
1818                 FROM pg_indexes \
1819                 WHERE indexname = \'stock_move_location_id_location_dest_id_product_id_state\'')
1820         if not cursor.fetchone():
1821             cursor.execute('CREATE INDEX stock_move_location_id_location_dest_id_product_id_state \
1822                     ON stock_move (product_id, state, location_id, location_dest_id)')
1823         return res
1824
1825     def onchange_lot_id(self, cr, uid, ids, prodlot_id=False, product_qty=False,
1826                         loc_id=False, product_id=False, uom_id=False, context=None):
1827         """ On change of production lot gives a warning message.
1828         @param prodlot_id: Changed production lot id
1829         @param product_qty: Quantity of product
1830         @param loc_id: Location id
1831         @param product_id: Product id
1832         @return: Warning message
1833         """
1834         if not prodlot_id or not loc_id:
1835             return {}
1836         ctx = context and context.copy() or {}
1837         ctx['location_id'] = loc_id
1838         ctx.update({'raise-exception': True})
1839         uom_obj = self.pool.get('product.uom')
1840         product_obj = self.pool.get('product.product')
1841         product_uom = product_obj.browse(cr, uid, product_id, context=ctx).uom_id
1842         prodlot = self.pool.get('stock.production.lot').browse(cr, uid, prodlot_id, context=ctx)
1843         location = self.pool.get('stock.location').browse(cr, uid, loc_id, context=ctx)
1844         uom = uom_obj.browse(cr, uid, uom_id, context=ctx)
1845         amount_actual = uom_obj._compute_qty_obj(cr, uid, product_uom, prodlot.stock_available, uom, context=ctx)
1846         warning = {}
1847         if (location.usage == 'internal') and (product_qty > (amount_actual or 0.0)):
1848             warning = {
1849                 'title': _('Insufficient Stock for Serial Number !'),
1850                 'message': _('You are moving %.2f %s but only %.2f %s available for this serial number.') % (product_qty, uom.name, amount_actual, uom.name)
1851             }
1852         return {'warning': warning}
1853
1854     def onchange_quantity(self, cr, uid, ids, product_id, product_qty,
1855                           product_uom, product_uos):
1856         """ On change of product quantity finds UoM and UoS quantities
1857         @param product_id: Product id
1858         @param product_qty: Changed Quantity of product
1859         @param product_uom: Unit of measure of product
1860         @param product_uos: Unit of sale of product
1861         @return: Dictionary of values
1862         """
1863         result = {
1864                   'product_uos_qty': 0.00
1865           }
1866
1867         if (not product_id) or (product_qty <=0.0):
1868             return {'value': result}
1869
1870         product_obj = self.pool.get('product.product')
1871         uos_coeff = product_obj.read(cr, uid, product_id, ['uos_coeff'])
1872
1873         if product_uos and product_uom and (product_uom != product_uos):
1874             result['product_uos_qty'] = product_qty * uos_coeff['uos_coeff']
1875         else:
1876             result['product_uos_qty'] = product_qty
1877
1878         return {'value': result}
1879
1880     def onchange_uos_quantity(self, cr, uid, ids, product_id, product_uos_qty,
1881                           product_uos, product_uom):
1882         """ On change of product quantity finds UoM and UoS quantities
1883         @param product_id: Product id
1884         @param product_uos_qty: Changed UoS Quantity of product
1885         @param product_uom: Unit of measure of product
1886         @param product_uos: Unit of sale of product
1887         @return: Dictionary of values
1888         """
1889         result = {
1890                   'product_qty': 0.00
1891           }
1892
1893         if (not product_id) or (product_uos_qty <=0.0):
1894             return {'value': result}
1895
1896         product_obj = self.pool.get('product.product')
1897         uos_coeff = product_obj.read(cr, uid, product_id, ['uos_coeff'])
1898
1899         if product_uos and product_uom and (product_uom != product_uos):
1900             result['product_qty'] = product_uos_qty / uos_coeff['uos_coeff']
1901         else:
1902             result['product_qty'] = product_uos_qty
1903
1904         return {'value': result}
1905
1906     def onchange_product_id(self, cr, uid, ids, prod_id=False, loc_id=False,
1907                             loc_dest_id=False, partner_id=False):
1908         """ On change of product id, if finds UoM, UoS, quantity and UoS quantity.
1909         @param prod_id: Changed Product id
1910         @param loc_id: Source location id
1911         @param loc_dest_id: Destination location id
1912         @param partner_id: Address id of partner
1913         @return: Dictionary of values
1914         """
1915         if not prod_id:
1916             return {}
1917         lang = False
1918         if partner_id:
1919             addr_rec = self.pool.get('res.partner').browse(cr, uid, partner_id)
1920             if addr_rec:
1921                 lang = addr_rec and addr_rec.lang or False
1922         ctx = {'lang': lang}
1923
1924         product = self.pool.get('product.product').browse(cr, uid, [prod_id], context=ctx)[0]
1925         uos_id  = product.uos_id and product.uos_id.id or False
1926         result = {
1927             'product_uom': product.uom_id.id,
1928             'product_uos': uos_id,
1929             'product_qty': 1.00,
1930             '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']
1931         }
1932         if not ids:
1933             result['name'] = product.partner_ref
1934         if loc_id:
1935             result['location_id'] = loc_id
1936         if loc_dest_id:
1937             result['location_dest_id'] = loc_dest_id
1938         return {'value': result}
1939
1940     def onchange_move_type(self, cr, uid, ids, type, context=None):
1941         """ On change of move type gives sorce and destination location.
1942         @param type: Move Type
1943         @return: Dictionary of values
1944         """
1945         mod_obj = self.pool.get('ir.model.data')
1946         location_source_id = 'stock_location_stock'
1947         location_dest_id = 'stock_location_stock'
1948         if type == 'in':
1949             location_source_id = 'stock_location_suppliers'
1950             location_dest_id = 'stock_location_stock'
1951         elif type == 'out':
1952             location_source_id = 'stock_location_stock'
1953             location_dest_id = 'stock_location_customers'
1954         source_location = mod_obj.get_object_reference(cr, uid, 'stock', location_source_id)
1955         dest_location = mod_obj.get_object_reference(cr, uid, 'stock', location_dest_id)
1956         return {'value':{'location_id': source_location and source_location[1] or False, 'location_dest_id': dest_location and dest_location[1] or False}}
1957
1958     def onchange_date(self, cr, uid, ids, date, date_expected, context=None):
1959         """ On change of Scheduled Date gives a Move date.
1960         @param date_expected: Scheduled Date
1961         @param date: Move Date
1962         @return: Move Date
1963         """
1964         if not date_expected:
1965             date_expected = time.strftime('%Y-%m-%d %H:%M:%S')
1966         return {'value':{'date': date_expected}}
1967
1968     def _chain_compute(self, cr, uid, moves, context=None):
1969         """ Finds whether the location has chained location type or not.
1970         @param moves: Stock moves
1971         @return: Dictionary containing destination location with chained location type.
1972         """
1973         result = {}
1974         for m in moves:
1975             dest = self.pool.get('stock.location').chained_location_get(
1976                 cr,
1977                 uid,
1978                 m.location_dest_id,
1979                 m.picking_id and m.picking_id.partner_id and m.picking_id.partner_id,
1980                 m.product_id,
1981                 context
1982             )
1983             if dest:
1984                 if dest[1] == 'transparent':
1985                     newdate = (datetime.strptime(m.date, '%Y-%m-%d %H:%M:%S') + relativedelta(days=dest[2] or 0)).strftime('%Y-%m-%d')
1986                     self.write(cr, uid, [m.id], {
1987                         'date': newdate,
1988                         'location_dest_id': dest[0].id})
1989                     if m.picking_id and (dest[3] or dest[5]):
1990                         self.pool.get('stock.picking').write(cr, uid, [m.picking_id.id], {
1991                             'stock_journal_id': dest[3] or m.picking_id.stock_journal_id.id,
1992                             'type': dest[5] or m.picking_id.type
1993                         }, context=context)
1994                     m.location_dest_id = dest[0]
1995                     res2 = self._chain_compute(cr, uid, [m], context=context)
1996                     for pick_id in res2.keys():
1997                         result.setdefault(pick_id, [])
1998                         result[pick_id] += res2[pick_id]
1999                 else:
2000                     result.setdefault(m.picking_id, [])
2001                     result[m.picking_id].append( (m, dest) )
2002         return result
2003
2004     def _prepare_chained_picking(self, cr, uid, picking_name, picking, picking_type, moves_todo, context=None):
2005         """Prepare the definition (values) to create a new chained picking.
2006
2007            :param str picking_name: desired new picking name
2008            :param browse_record picking: source picking (being chained to)
2009            :param str picking_type: desired new picking type
2010            :param list moves_todo: specification of the stock moves to be later included in this
2011                picking, in the form::
2012
2013                    [[move, (dest_location, auto_packing, chained_delay, chained_journal,
2014                                   chained_company_id, chained_picking_type)],
2015                     ...
2016                    ]
2017
2018                See also :meth:`stock_location.chained_location_get`.
2019         """
2020         res_company = self.pool.get('res.company')
2021         return {
2022                     'name': picking_name,
2023                     'origin': tools.ustr(picking.origin or ''),
2024                     'type': picking_type,
2025                     'note': picking.note,
2026                     'move_type': picking.move_type,
2027                     'auto_picking': moves_todo[0][1][1] == 'auto',
2028                     'stock_journal_id': moves_todo[0][1][3],
2029                     'company_id': moves_todo[0][1][4] or res_company._company_default_get(cr, uid, 'stock.company', context=context),
2030                     'partner_id': picking.partner_id.id,
2031                     'invoice_state': 'none',
2032                     'date': picking.date,
2033                 }
2034
2035     def _create_chained_picking(self, cr, uid, picking_name, picking, picking_type, moves_todo, context=None):
2036         picking_obj = self.pool.get('stock.picking')
2037         return picking_obj.create(cr, uid, self._prepare_chained_picking(cr, uid, picking_name, picking, picking_type, moves_todo, context=context))
2038
2039     def create_chained_picking(self, cr, uid, moves, context=None):
2040         res_obj = self.pool.get('res.company')
2041         location_obj = self.pool.get('stock.location')
2042         move_obj = self.pool.get('stock.move')
2043         wf_service = netsvc.LocalService("workflow")
2044         new_moves = []
2045         if context is None:
2046             context = {}
2047         seq_obj = self.pool.get('ir.sequence')
2048         for picking, todo in self._chain_compute(cr, uid, moves, context=context).items():
2049             ptype = todo[0][1][5] and todo[0][1][5] or location_obj.picking_type_get(cr, uid, todo[0][0].location_dest_id, todo[0][1][0])
2050             if picking:
2051                 # name of new picking according to its type
2052                 new_pick_name = seq_obj.get(cr, uid, 'stock.picking.' + ptype)
2053                 pickid = self._create_chained_picking(cr, uid, new_pick_name, picking, ptype, todo, context=context)
2054                 # Need to check name of old picking because it always considers picking as "OUT" when created from Sale Order
2055                 old_ptype = location_obj.picking_type_get(cr, uid, picking.move_lines[0].location_id, picking.move_lines[0].location_dest_id)
2056                 if old_ptype != picking.type:
2057                     old_pick_name = seq_obj.get(cr, uid, 'stock.picking.' + old_ptype)
2058                     self.pool.get('stock.picking').write(cr, uid, [picking.id], {'name': old_pick_name}, context=context)
2059             else:
2060                 pickid = False
2061             for move, (loc, dummy, delay, dummy, company_id, ptype) in todo:
2062                 new_id = move_obj.copy(cr, uid, move.id, {
2063                     'location_id': move.location_dest_id.id,
2064                     'location_dest_id': loc.id,
2065                     'date_moved': time.strftime('%Y-%m-%d'),
2066                     'picking_id': pickid,
2067                     'state': 'waiting',
2068                     'company_id': company_id or res_obj._company_default_get(cr, uid, 'stock.company', context=context)  ,
2069                     'move_history_ids': [],
2070                     'date': (datetime.strptime(move.date, '%Y-%m-%d %H:%M:%S') + relativedelta(days=delay or 0)).strftime('%Y-%m-%d'),
2071                     'move_history_ids2': []}
2072                 )
2073                 move_obj.write(cr, uid, [move.id], {
2074                     'move_dest_id': new_id,
2075                     'move_history_ids': [(4, new_id)]
2076                 })
2077                 new_moves.append(self.browse(cr, uid, [new_id])[0])
2078             if pickid:
2079                 wf_service.trg_validate(uid, 'stock.picking', pickid, 'button_confirm', cr)
2080         if new_moves:
2081             new_moves += self.create_chained_picking(cr, uid, new_moves, context)
2082         return new_moves
2083
2084     def action_confirm(self, cr, uid, ids, context=None):
2085         """ Confirms stock move.
2086         @return: List of ids.
2087         """
2088         moves = self.browse(cr, uid, ids, context=context)
2089         self.write(cr, uid, ids, {'state': 'confirmed'})
2090         self.create_chained_picking(cr, uid, moves, context)
2091         return []
2092
2093     def action_assign(self, cr, uid, ids, *args):
2094         """ Changes state to confirmed or waiting.
2095         @return: List of values
2096         """
2097         todo = []
2098         for move in self.browse(cr, uid, ids):
2099             if move.state in ('confirmed', 'waiting'):
2100                 todo.append(move.id)
2101         res = self.check_assign(cr, uid, todo)
2102         return res
2103
2104     def force_assign(self, cr, uid, ids, context=None):
2105         """ Changes the state to assigned.
2106         @return: True
2107         """
2108         self.write(cr, uid, ids, {'state': 'assigned'})
2109         return True
2110
2111     def cancel_assign(self, cr, uid, ids, context=None):
2112         """ Changes the state to confirmed.
2113         @return: True
2114         """
2115         self.write(cr, uid, ids, {'state': 'confirmed'})
2116
2117         # fix for bug lp:707031
2118         # called write of related picking because changing move availability does
2119         # not trigger workflow of picking in order to change the state of picking
2120         wf_service = netsvc.LocalService('workflow')
2121         for move in self.browse(cr, uid, ids, context):
2122             if move.picking_id:
2123                 wf_service.trg_write(uid, 'stock.picking', move.picking_id.id, cr)
2124         return True
2125
2126     #
2127     # Duplicate stock.move
2128     #
2129     def check_assign(self, cr, uid, ids, context=None):
2130         """ Checks the product type and accordingly writes the state.
2131         @return: No. of moves done
2132         """
2133         done = []
2134         count = 0
2135         pickings = {}
2136         if context is None:
2137             context = {}
2138         for move in self.browse(cr, uid, ids, context=context):
2139             if move.product_id.type == 'consu' or move.location_id.usage == 'supplier':
2140                 if move.state in ('confirmed', 'waiting'):
2141                     done.append(move.id)
2142                 pickings[move.picking_id.id] = 1
2143                 continue
2144             if move.state in ('confirmed', 'waiting'):
2145                 # Important: we must pass lock=True to _product_reserve() to avoid race conditions and double reservations
2146                 res = self.pool.get('stock.location')._product_reserve(cr, uid, [move.location_id.id], move.product_id.id, move.product_qty, {'uom': move.product_uom.id}, lock=True)
2147                 if res:
2148                     #_product_available_test depends on the next status for correct functioning
2149                     #the test does not work correctly if the same product occurs multiple times
2150                     #in the same order. This is e.g. the case when using the button 'split in two' of
2151                     #the stock outgoing form
2152                     self.write(cr, uid, [move.id], {'state':'assigned'})
2153                     done.append(move.id)
2154                     pickings[move.picking_id.id] = 1
2155                     r = res.pop(0)
2156                     product_uos_qty = self.pool.get('stock.move').onchange_quantity(cr, uid, ids, move.product_id.id, r[0], move.product_id.uom_id.id, move.product_id.uos_id.id)['value']['product_uos_qty']
2157                     cr.execute('update stock_move set location_id=%s, product_qty=%s, product_uos_qty=%s where id=%s', (r[1], r[0],product_uos_qty, move.id))
2158
2159                     while res:
2160                         r = res.pop(0)
2161                         move_id = self.copy(cr, uid, move.id, {'product_uos_qty': product_uos_qty, 'product_qty': r[0], 'location_id': r[1]})
2162                         done.append(move_id)
2163         if done:
2164             count += len(done)
2165             self.write(cr, uid, done, {'state': 'assigned'})
2166
2167         if count:
2168             for pick_id in pickings:
2169                 wf_service = netsvc.LocalService("workflow")
2170                 wf_service.trg_write(uid, 'stock.picking', pick_id, cr)
2171         return count
2172
2173     def setlast_tracking(self, cr, uid, ids, context=None):
2174         tracking_obj = self.pool.get('stock.tracking')
2175         picking = self.browse(cr, uid, ids, context=context)[0].picking_id
2176         if picking:
2177             last_track = [line.tracking_id.id for line in picking.move_lines if line.tracking_id]
2178             if not last_track:
2179                 last_track = tracking_obj.create(cr, uid, {}, context=context)
2180             else:
2181                 last_track.sort()
2182                 last_track = last_track[-1]
2183             self.write(cr, uid, ids, {'tracking_id': last_track})
2184         return True
2185
2186     #
2187     # Cancel move => cancel others move and pickings
2188     #
2189     def action_cancel(self, cr, uid, ids, context=None):
2190         """ Cancels the moves and if all moves are cancelled it cancels the picking.
2191         @return: True
2192         """
2193         if not len(ids):
2194             return True
2195         if context is None:
2196             context = {}
2197         pickings = set()
2198         for move in self.browse(cr, uid, ids, context=context):
2199             if move.state in ('confirmed', 'waiting', 'assigned', 'draft'):
2200                 if move.picking_id:
2201                     pickings.add(move.picking_id.id)
2202             if move.move_dest_id and move.move_dest_id.state == 'waiting':
2203                 self.write(cr, uid, [move.move_dest_id.id], {'state': 'assigned'})
2204                 if context.get('call_unlink',False) and move.move_dest_id.picking_id:
2205                     wf_service = netsvc.LocalService("workflow")
2206                     wf_service.trg_write(uid, 'stock.picking', move.move_dest_id.picking_id.id, cr)
2207         self.write(cr, uid, ids, {'state': 'cancel', 'move_dest_id': False})
2208         if not context.get('call_unlink',False):
2209             for pick in self.pool.get('stock.picking').browse(cr, uid, list(pickings), context=context):
2210                 if all(move.state == 'cancel' for move in pick.move_lines):
2211                     self.pool.get('stock.picking').write(cr, uid, [pick.id], {'state': 'cancel'})
2212
2213         wf_service = netsvc.LocalService("workflow")
2214         for id in ids:
2215             wf_service.trg_trigger(uid, 'stock.move', id, cr)
2216         return True
2217
2218     def _get_accounting_data_for_valuation(self, cr, uid, move, context=None):
2219         """
2220         Return the accounts and journal to use to post Journal Entries for the real-time
2221         valuation of the move.
2222
2223         :param context: context dictionary that can explicitly mention the company to consider via the 'force_company' key
2224         :raise: osv.except_osv() is any mandatory account or journal is not defined.
2225         """
2226         product_obj=self.pool.get('product.product')
2227         accounts = product_obj.get_product_accounts(cr, uid, move.product_id.id, context)
2228         if move.location_id.valuation_out_account_id:
2229             acc_src = move.location_id.valuation_out_account_id.id
2230         else:
2231             acc_src = accounts['stock_account_input']
2232
2233         if move.location_dest_id.valuation_in_account_id:
2234             acc_dest = move.location_dest_id.valuation_in_account_id.id
2235         else:
2236             acc_dest = accounts['stock_account_output']
2237
2238         acc_valuation = accounts.get('property_stock_valuation_account_id', False)
2239         journal_id = accounts['stock_journal']
2240
2241         if acc_dest == acc_valuation:
2242             raise osv.except_osv(_('Error!'),  _('Cannot create Journal Entry, Output Account of this product and Valuation account on category of this product are same.'))
2243
2244         if acc_src == acc_valuation:
2245             raise osv.except_osv(_('Error!'),  _('Cannot create Journal Entry, Input Account of this product and Valuation account on category of this product are same.'))
2246
2247         if not acc_src:
2248             raise osv.except_osv(_('Error!'),  _('Please define stock input account for this product or its category: "%s" (id: %d)') % \
2249                                     (move.product_id.name, move.product_id.id,))
2250         if not acc_dest:
2251             raise osv.except_osv(_('Error!'),  _('Please define stock output account for this product or its category: "%s" (id: %d)') % \
2252                                     (move.product_id.name, move.product_id.id,))
2253         if not journal_id:
2254             raise osv.except_osv(_('Error!'), _('Please define journal on the product category: "%s" (id: %d)') % \
2255                                     (move.product_id.categ_id.name, move.product_id.categ_id.id,))
2256         if not acc_valuation:
2257             raise osv.except_osv(_('Error!'), _('Please define inventory valuation account on the product category: "%s" (id: %d)') % \
2258                                     (move.product_id.categ_id.name, move.product_id.categ_id.id,))
2259         return journal_id, acc_src, acc_dest, acc_valuation
2260
2261     def _get_reference_accounting_values_for_valuation(self, cr, uid, move, context=None):
2262         """
2263         Return the reference amount and reference currency representing the inventory valuation for this move.
2264         These reference values should possibly be converted before being posted in Journals to adapt to the primary
2265         and secondary currencies of the relevant accounts.
2266         """
2267         product_uom_obj = self.pool.get('product.uom')
2268
2269         # by default the reference currency is that of the move's company
2270         reference_currency_id = move.company_id.currency_id.id
2271
2272         default_uom = move.product_id.uom_id.id
2273         qty = product_uom_obj._compute_qty(cr, uid, move.product_uom.id, move.product_qty, default_uom)
2274
2275         # if product is set to average price and a specific value was entered in the picking wizard,
2276         # we use it
2277         if move.product_id.cost_method == 'average' and move.price_unit:
2278             reference_amount = qty * move.price_unit
2279             reference_currency_id = move.price_currency_id.id or reference_currency_id
2280
2281         # Otherwise we default to the company's valuation price type, considering that the values of the
2282         # valuation field are expressed in the default currency of the move's company.
2283         else:
2284             if context is None:
2285                 context = {}
2286             currency_ctx = dict(context, currency_id = move.company_id.currency_id.id)
2287             amount_unit = move.product_id.price_get('standard_price', context=currency_ctx)[move.product_id.id]
2288             reference_amount = amount_unit * qty
2289
2290         return reference_amount, reference_currency_id
2291
2292
2293     def _create_product_valuation_moves(self, cr, uid, move, context=None):
2294         """
2295         Generate the appropriate accounting moves if the product being moves is subject
2296         to real_time valuation tracking, and the source or destination location is
2297         a transit location or is outside of the company.
2298         """
2299         if move.product_id.valuation == 'real_time': # FIXME: product valuation should perhaps be a property?
2300             if context is None:
2301                 context = {}
2302             src_company_ctx = dict(context,force_company=move.location_id.company_id.id)
2303             dest_company_ctx = dict(context,force_company=move.location_dest_id.company_id.id)
2304             account_moves = []
2305             # Outgoing moves (or cross-company output part)
2306             if move.location_id.company_id \
2307                 and (move.location_id.usage == 'internal' and move.location_dest_id.usage != 'internal'\
2308                      or move.location_id.company_id != move.location_dest_id.company_id):
2309                 journal_id, acc_src, acc_dest, acc_valuation = self._get_accounting_data_for_valuation(cr, uid, move, src_company_ctx)
2310                 reference_amount, reference_currency_id = self._get_reference_accounting_values_for_valuation(cr, uid, move, src_company_ctx)
2311                 #returning goods to supplier
2312                 if move.location_dest_id.usage == 'supplier':
2313                     account_moves += [(journal_id, self._create_account_move_line(cr, uid, move, acc_valuation, acc_src, reference_amount, reference_currency_id, context))]
2314                 else:
2315                     account_moves += [(journal_id, self._create_account_move_line(cr, uid, move, acc_valuation, acc_dest, reference_amount, reference_currency_id, context))]
2316
2317             # Incoming moves (or cross-company input part)
2318             if move.location_dest_id.company_id \
2319                 and (move.location_id.usage != 'internal' and move.location_dest_id.usage == 'internal'\
2320                      or move.location_id.company_id != move.location_dest_id.company_id):
2321                 journal_id, acc_src, acc_dest, acc_valuation = self._get_accounting_data_for_valuation(cr, uid, move, dest_company_ctx)
2322                 reference_amount, reference_currency_id = self._get_reference_accounting_values_for_valuation(cr, uid, move, src_company_ctx)
2323                 #goods return from customer
2324                 if move.location_id.usage == 'customer':
2325                     account_moves += [(journal_id, self._create_account_move_line(cr, uid, move, acc_dest, acc_valuation, reference_amount, reference_currency_id, context))]
2326                 else:
2327                     account_moves += [(journal_id, self._create_account_move_line(cr, uid, move, acc_src, acc_valuation, reference_amount, reference_currency_id, context))]
2328
2329             move_obj = self.pool.get('account.move')
2330             for j_id, move_lines in account_moves:
2331                 move_obj.create(cr, uid,
2332                         {
2333                          'journal_id': j_id,
2334                          'line_id': move_lines,
2335                          'ref': move.picking_id and move.picking_id.name})
2336
2337
2338     def action_done(self, cr, uid, ids, context=None):
2339         """ Makes the move done and if all moves are done, it will finish the picking.
2340         @return:
2341         """
2342         picking_ids = []
2343         move_ids = []
2344         wf_service = netsvc.LocalService("workflow")
2345         if context is None:
2346             context = {}
2347
2348         todo = []
2349         for move in self.browse(cr, uid, ids, context=context):
2350             if move.state=="draft":
2351                 todo.append(move.id)
2352         if todo:
2353             self.action_confirm(cr, uid, todo, context=context)
2354             todo = []
2355
2356         for move in self.browse(cr, uid, ids, context=context):
2357             if move.state in ['done','cancel']:
2358                 continue
2359             move_ids.append(move.id)
2360
2361             if move.picking_id:
2362                 picking_ids.append(move.picking_id.id)
2363             if move.move_dest_id.id and (move.state != 'done'):
2364                 self.write(cr, uid, [move.id], {'move_history_ids': [(4, move.move_dest_id.id)]})
2365                 #cr.execute('insert into stock_move_history_ids (parent_id,child_id) values (%s,%s)', (move.id, move.move_dest_id.id))
2366                 if move.move_dest_id.state in ('waiting', 'confirmed'):
2367                     self.force_assign(cr, uid, [move.move_dest_id.id], context=context)
2368                     if move.move_dest_id.picking_id:
2369                         wf_service.trg_write(uid, 'stock.picking', move.move_dest_id.picking_id.id, cr)
2370                     if move.move_dest_id.auto_validate:
2371                         self.action_done(cr, uid, [move.move_dest_id.id], context=context)
2372
2373             self._create_product_valuation_moves(cr, uid, move, context=context)
2374             if move.state not in ('confirmed','done','assigned'):
2375                 todo.append(move.id)
2376
2377         if todo:
2378             self.action_confirm(cr, uid, todo, context=context)
2379
2380         self.write(cr, uid, move_ids, {'state': 'done', 'date': time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
2381         for id in move_ids:
2382              wf_service.trg_trigger(uid, 'stock.move', id, cr)
2383
2384         for pick_id in picking_ids:
2385             wf_service.trg_write(uid, 'stock.picking', pick_id, cr)
2386
2387         return True
2388
2389     def _create_account_move_line(self, cr, uid, move, src_account_id, dest_account_id, reference_amount, reference_currency_id, context=None):
2390         """
2391         Generate the account.move.line values to post to track the stock valuation difference due to the
2392         processing of the given stock move.
2393         """
2394         # prepare default values considering that the destination accounts have the reference_currency_id as their main currency
2395         partner_id = (move.picking_id.partner_id and move.picking_id.partner_id.id and move.picking_id.partner_id.id) or False
2396         debit_line_vals = {
2397                     'name': move.name,
2398                     'product_id': move.product_id and move.product_id.id or False,
2399                     'quantity': move.product_qty,
2400                     'ref': move.picking_id and move.picking_id.name or False,
2401                     'date': time.strftime('%Y-%m-%d'),
2402                     'partner_id': partner_id,
2403                     'debit': reference_amount,
2404                     'account_id': dest_account_id,
2405         }
2406         credit_line_vals = {
2407                     'name': move.name,
2408                     'product_id': move.product_id and move.product_id.id or False,
2409                     'quantity': move.product_qty,
2410                     'ref': move.picking_id and move.picking_id.name or False,
2411                     'date': time.strftime('%Y-%m-%d'),
2412                     'partner_id': partner_id,
2413                     'credit': reference_amount,
2414                     'account_id': src_account_id,
2415         }
2416
2417         # if we are posting to accounts in a different currency, provide correct values in both currencies correctly
2418         # when compatible with the optional secondary currency on the account.
2419         # Financial Accounts only accept amounts in secondary currencies if there's no secondary currency on the account
2420         # or if it's the same as that of the secondary amount being posted.
2421         account_obj = self.pool.get('account.account')
2422         src_acct, dest_acct = account_obj.browse(cr, uid, [src_account_id, dest_account_id], context=context)
2423         src_main_currency_id = src_acct.company_id.currency_id.id
2424         dest_main_currency_id = dest_acct.company_id.currency_id.id
2425         cur_obj = self.pool.get('res.currency')
2426         if reference_currency_id != src_main_currency_id:
2427             # fix credit line:
2428             credit_line_vals['credit'] = cur_obj.compute(cr, uid, reference_currency_id, src_main_currency_id, reference_amount, context=context)
2429             if (not src_acct.currency_id) or src_acct.currency_id.id == reference_currency_id:
2430                 credit_line_vals.update(currency_id=reference_currency_id, amount_currency=reference_amount)
2431         if reference_currency_id != dest_main_currency_id:
2432             # fix debit line:
2433             debit_line_vals['debit'] = cur_obj.compute(cr, uid, reference_currency_id, dest_main_currency_id, reference_amount, context=context)
2434             if (not dest_acct.currency_id) or dest_acct.currency_id.id == reference_currency_id:
2435                 debit_line_vals.update(currency_id=reference_currency_id, amount_currency=reference_amount)
2436
2437         return [(0, 0, debit_line_vals), (0, 0, credit_line_vals)]
2438
2439     def unlink(self, cr, uid, ids, context=None):
2440         if context is None:
2441             context = {}
2442         ctx = context.copy()
2443         for move in self.browse(cr, uid, ids, context=context):
2444             if move.state != 'draft' and not ctx.get('call_unlink',False):
2445                 raise osv.except_osv(_('User Error!'),
2446                         _('You can only delete draft moves.'))
2447         return super(stock_move, self).unlink(
2448             cr, uid, ids, context=ctx)
2449
2450     # _create_lot function is not used anywhere
2451     def _create_lot(self, cr, uid, ids, product_id, prefix=False):
2452         """ Creates production lot
2453         @return: Production lot id
2454         """
2455         prodlot_obj = self.pool.get('stock.production.lot')
2456         prodlot_id = prodlot_obj.create(cr, uid, {'prefix': prefix, 'product_id': product_id})
2457         return prodlot_id
2458
2459     def action_scrap(self, cr, uid, ids, quantity, location_id, context=None):
2460         """ Move the scrap/damaged product into scrap location
2461         @param cr: the database cursor
2462         @param uid: the user id
2463         @param ids: ids of stock move object to be scrapped
2464         @param quantity : specify scrap qty
2465         @param location_id : specify scrap location
2466         @param context: context arguments
2467         @return: Scraped lines
2468         """
2469         #quantity should in MOVE UOM
2470         if quantity <= 0:
2471             raise osv.except_osv(_('Warning!'), _('Please provide a positive quantity to scrap.'))
2472         res = []
2473         for move in self.browse(cr, uid, ids, context=context):
2474             move_qty = move.product_qty
2475             uos_qty = quantity / move_qty * move.product_uos_qty
2476             default_val = {
2477                 'product_qty': quantity,
2478                 'product_uos_qty': uos_qty,
2479                 'state': move.state,
2480                 'scrapped' : True,
2481                 'location_dest_id': location_id,
2482                 'tracking_id': move.tracking_id.id,
2483                 'prodlot_id': move.prodlot_id.id,
2484             }
2485             if move.location_id.usage <> 'internal':
2486                 default_val.update({'location_id': move.location_dest_id.id})
2487             new_move = self.copy(cr, uid, move.id, default_val)
2488
2489             res += [new_move]
2490             product_obj = self.pool.get('product.product')
2491             for product in product_obj.browse(cr, uid, [move.product_id.id], context=context):
2492                 if move.picking_id:
2493                     uom = product.uom_id.name if product.uom_id else ''
2494                     move.picking_id.scrap_send_note(quantity, uom, product.name, context=context)
2495
2496         self.action_done(cr, uid, res, context=context)
2497         return res
2498
2499     # action_split function is not used anywhere
2500     def action_split(self, cr, uid, ids, quantity, split_by_qty=1, prefix=False, with_lot=True, context=None):
2501         """ Split Stock Move lines into production lot which specified split by quantity.
2502         @param cr: the database cursor
2503         @param uid: the user id
2504         @param ids: ids of stock move object to be splited
2505         @param split_by_qty : specify split by qty
2506         @param prefix : specify prefix of production lot
2507         @param with_lot : if true, prodcution lot will assign for split line otherwise not.
2508         @param context: context arguments
2509         @return: Splited move lines
2510         """
2511
2512         if context is None:
2513             context = {}
2514         if quantity <= 0:
2515             raise osv.except_osv(_('Warning!'), _('Please provide proper quantity.'))
2516
2517         res = []
2518
2519         for move in self.browse(cr, uid, ids, context=context):
2520             if split_by_qty <= 0 or quantity == 0:
2521                 return res
2522
2523             uos_qty = split_by_qty / move.product_qty * move.product_uos_qty
2524
2525             quantity_rest = quantity % split_by_qty
2526             uos_qty_rest = split_by_qty / move.product_qty * move.product_uos_qty
2527
2528             update_val = {
2529                 'product_qty': split_by_qty,
2530                 'product_uos_qty': uos_qty,
2531             }
2532             for idx in range(int(quantity//split_by_qty)):
2533                 if not idx and move.product_qty<=quantity:
2534                     current_move = move.id
2535                 else:
2536                     current_move = self.copy(cr, uid, move.id, {'state': move.state})
2537                 res.append(current_move)
2538                 if with_lot:
2539                     update_val['prodlot_id'] = self._create_lot(cr, uid, [current_move], move.product_id.id)
2540
2541                 self.write(cr, uid, [current_move], update_val)
2542
2543
2544             if quantity_rest > 0:
2545                 idx = int(quantity//split_by_qty)
2546                 update_val['product_qty'] = quantity_rest
2547                 update_val['product_uos_qty'] = uos_qty_rest
2548                 if not idx and move.product_qty<=quantity:
2549                     current_move = move.id
2550                 else:
2551                     current_move = self.copy(cr, uid, move.id, {'state': move.state})
2552
2553                 res.append(current_move)
2554
2555
2556                 if with_lot:
2557                     update_val['prodlot_id'] = self._create_lot(cr, uid, [current_move], move.product_id.id)
2558
2559                 self.write(cr, uid, [current_move], update_val)
2560         return res
2561
2562     def action_consume(self, cr, uid, ids, quantity, location_id=False, context=None):
2563         """ Consumed product with specific quatity from specific source location
2564         @param cr: the database cursor
2565         @param uid: the user id
2566         @param ids: ids of stock move object to be consumed
2567         @param quantity : specify consume quantity
2568         @param location_id : specify source location
2569         @param context: context arguments
2570         @return: Consumed lines
2571         """
2572         #quantity should in MOVE UOM
2573         if context is None:
2574             context = {}
2575         if quantity <= 0:
2576             raise osv.except_osv(_('Warning!'), _('Please provide proper quantity.'))
2577         res = []
2578         for move in self.browse(cr, uid, ids, context=context):
2579             move_qty = move.product_qty
2580             if move_qty <= 0:
2581                 raise osv.except_osv(_('Error!'), _('Cannot consume a move with negative or zero quantity.'))
2582             quantity_rest = move.product_qty
2583             quantity_rest -= quantity
2584             uos_qty_rest = quantity_rest / move_qty * move.product_uos_qty
2585             if quantity_rest <= 0:
2586                 quantity_rest = 0
2587                 uos_qty_rest = 0
2588                 quantity = move.product_qty
2589
2590             uos_qty = quantity / move_qty * move.product_uos_qty
2591             if quantity_rest > 0:
2592                 default_val = {
2593                     'product_qty': quantity,
2594                     'product_uos_qty': uos_qty,
2595                     'state': move.state,
2596                     'location_id': location_id or move.location_id.id,
2597                 }
2598                 current_move = self.copy(cr, uid, move.id, default_val)
2599                 res += [current_move]
2600                 update_val = {}
2601                 update_val['product_qty'] = quantity_rest
2602                 update_val['product_uos_qty'] = uos_qty_rest
2603                 self.write(cr, uid, [move.id], update_val)
2604
2605             else:
2606                 quantity_rest = quantity
2607                 uos_qty_rest =  uos_qty
2608                 res += [move.id]
2609                 update_val = {
2610                         'product_qty' : quantity_rest,
2611                         'product_uos_qty' : uos_qty_rest,
2612                         'location_id': location_id or move.location_id.id,
2613                 }
2614                 self.write(cr, uid, [move.id], update_val)
2615
2616         self.action_done(cr, uid, res, context=context)
2617
2618         return res
2619
2620     # FIXME: needs refactoring, this code is partially duplicated in stock_picking.do_partial()!
2621     def do_partial(self, cr, uid, ids, partial_datas, context=None):
2622         """ Makes partial pickings and moves done.
2623         @param partial_datas: Dictionary containing details of partial picking
2624                           like partner_id, delivery_date, delivery
2625                           moves with product_id, product_qty, uom
2626         """
2627         res = {}
2628         picking_obj = self.pool.get('stock.picking')
2629         product_obj = self.pool.get('product.product')
2630         currency_obj = self.pool.get('res.currency')
2631         uom_obj = self.pool.get('product.uom')
2632         wf_service = netsvc.LocalService("workflow")
2633
2634         if context is None:
2635             context = {}
2636
2637         complete, too_many, too_few = [], [], []
2638         move_product_qty = {}
2639         prodlot_ids = {}
2640         for move in self.browse(cr, uid, ids, context=context):
2641             if move.state in ('done', 'cancel'):
2642                 continue
2643             partial_data = partial_datas.get('move%s'%(move.id), False)
2644             assert partial_data, _('Missing partial picking data for move #%s.') % (move.id)
2645             product_qty = partial_data.get('product_qty',0.0)
2646             move_product_qty[move.id] = product_qty
2647             product_uom = partial_data.get('product_uom',False)
2648             product_price = partial_data.get('product_price',0.0)
2649             product_currency = partial_data.get('product_currency',False)
2650             prodlot_ids[move.id] = partial_data.get('prodlot_id')
2651             if move.product_qty == product_qty:
2652                 complete.append(move)
2653             elif move.product_qty > product_qty:
2654                 too_few.append(move)
2655             else:
2656                 too_many.append(move)
2657
2658             # Average price computation
2659             if (move.picking_id.type == 'in') and (move.product_id.cost_method == 'average'):
2660                 product = product_obj.browse(cr, uid, move.product_id.id)
2661                 move_currency_id = move.company_id.currency_id.id
2662                 context['currency_id'] = move_currency_id
2663                 qty = uom_obj._compute_qty(cr, uid, product_uom, product_qty, product.uom_id.id)
2664                 if qty > 0:
2665                     new_price = currency_obj.compute(cr, uid, product_currency,
2666                             move_currency_id, product_price)
2667                     new_price = uom_obj._compute_price(cr, uid, product_uom, new_price,
2668                             product.uom_id.id)
2669                     if product.qty_available <= 0:
2670                         new_std_price = new_price
2671                     else:
2672                         # Get the standard price
2673                         amount_unit = product.price_get('standard_price', context=context)[product.id]
2674                         new_std_price = ((amount_unit * product.qty_available)\
2675                             + (new_price * qty))/(product.qty_available + qty)
2676
2677                     product_obj.write(cr, uid, [product.id],{'standard_price': new_std_price})
2678
2679                     # Record the values that were chosen in the wizard, so they can be
2680                     # used for inventory valuation if real-time valuation is enabled.
2681                     self.write(cr, uid, [move.id],
2682                                 {'price_unit': product_price,
2683                                  'price_currency_id': product_currency,
2684                                 })
2685
2686         for move in too_few:
2687             product_qty = move_product_qty[move.id]
2688             if product_qty != 0:
2689                 defaults = {
2690                             'product_qty' : product_qty,
2691                             'product_uos_qty': product_qty,
2692                             'picking_id' : move.picking_id.id,
2693                             'state': 'assigned',
2694                             'move_dest_id': False,
2695                             'price_unit': move.price_unit,
2696                             }
2697                 prodlot_id = prodlot_ids[move.id]
2698                 if prodlot_id:
2699                     defaults.update(prodlot_id=prodlot_id)
2700                 new_move = self.copy(cr, uid, move.id, defaults)
2701                 complete.append(self.browse(cr, uid, new_move))
2702             self.write(cr, uid, [move.id],
2703                     {
2704                         'product_qty' : move.product_qty - product_qty,
2705                         'product_uos_qty':move.product_qty - product_qty,
2706                     })
2707
2708
2709         for move in too_many:
2710             self.write(cr, uid, [move.id],
2711                     {
2712                         'product_qty': move.product_qty,
2713                         'product_uos_qty': move.product_qty,
2714                     })
2715             complete.append(move)
2716
2717         for move in complete:
2718             if prodlot_ids.get(move.id):
2719                 self.write(cr, uid, [move.id],{'prodlot_id': prodlot_ids.get(move.id)})
2720             self.action_done(cr, uid, [move.id], context=context)
2721             if  move.picking_id.id :
2722                 # TOCHECK : Done picking if all moves are done
2723                 cr.execute("""
2724                     SELECT move.id FROM stock_picking pick
2725                     RIGHT JOIN stock_move move ON move.picking_id = pick.id AND move.state = %s
2726                     WHERE pick.id = %s""",
2727                             ('done', move.picking_id.id))
2728                 res = cr.fetchall()
2729                 if len(res) == len(move.picking_id.move_lines):
2730                     picking_obj.action_move(cr, uid, [move.picking_id.id])
2731                     wf_service.trg_validate(uid, 'stock.picking', move.picking_id.id, 'button_done', cr)
2732
2733         return [move.id for move in complete]
2734
2735 stock_move()
2736
2737 class stock_inventory(osv.osv):
2738     _name = "stock.inventory"
2739     _description = "Inventory"
2740     _columns = {
2741         'name': fields.char('Inventory Reference', size=64, required=True, readonly=True, states={'draft': [('readonly', False)]}),
2742         'date': fields.datetime('Creation Date', required=True, readonly=True, states={'draft': [('readonly', False)]}),
2743         'date_done': fields.datetime('Date done'),
2744         'inventory_line_id': fields.one2many('stock.inventory.line', 'inventory_id', 'Inventories', states={'done': [('readonly', True)]}),
2745         'move_ids': fields.many2many('stock.move', 'stock_inventory_move_rel', 'inventory_id', 'move_id', 'Created Moves'),
2746         'state': fields.selection( (('draft', 'Draft'), ('cancel','Cancelled'), ('confirm','Confirmed'), ('done', 'Done')), 'Status', readonly=True, select=True),
2747         'company_id': fields.many2one('res.company', 'Company', required=True, select=True, readonly=True, states={'draft':[('readonly',False)]}),
2748
2749     }
2750     _defaults = {
2751         'date': lambda *a: time.strftime('%Y-%m-%d %H:%M:%S'),
2752         'state': 'draft',
2753         'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.inventory', context=c)
2754     }
2755
2756     def copy(self, cr, uid, id, default=None, context=None):
2757         if default is None:
2758             default = {}
2759         default = default.copy()
2760         default.update({'move_ids': [], 'date_done': False})
2761         return super(stock_inventory, self).copy(cr, uid, id, default, context=context)
2762
2763     def _inventory_line_hook(self, cr, uid, inventory_line, move_vals):
2764         """ Creates a stock move from an inventory line
2765         @param inventory_line:
2766         @param move_vals:
2767         @return:
2768         """
2769         return self.pool.get('stock.move').create(cr, uid, move_vals)
2770
2771     def action_done(self, cr, uid, ids, context=None):
2772         """ Finish the inventory
2773         @return: True
2774         """
2775         if context is None:
2776             context = {}
2777         move_obj = self.pool.get('stock.move')
2778         for inv in self.browse(cr, uid, ids, context=context):
2779             move_obj.action_done(cr, uid, [x.id for x in inv.move_ids], context=context)
2780             self.write(cr, uid, [inv.id], {'state':'done', 'date_done': time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
2781         return True
2782
2783     def action_confirm(self, cr, uid, ids, context=None):
2784         """ Confirm the inventory and writes its finished date
2785         @return: True
2786         """
2787         if context is None:
2788             context = {}
2789         # to perform the correct inventory corrections we need analyze stock location by
2790         # location, never recursively, so we use a special context
2791         product_context = dict(context, compute_child=False)
2792
2793         location_obj = self.pool.get('stock.location')
2794         for inv in self.browse(cr, uid, ids, context=context):
2795             move_ids = []
2796             for line in inv.inventory_line_id:
2797                 pid = line.product_id.id
2798                 product_context.update(uom=line.product_uom.id, date=inv.date, prodlot_id=line.prod_lot_id.id)
2799                 amount = location_obj._product_get(cr, uid, line.location_id.id, [pid], product_context)[pid]
2800                 change = line.product_qty - amount
2801                 lot_id = line.prod_lot_id.id
2802                 if change:
2803                     location_id = line.product_id.product_tmpl_id.property_stock_inventory.id
2804                     value = {
2805                         'name': 'INV:' + str(line.inventory_id.id) + ':' + line.inventory_id.name,
2806                         'product_id': line.product_id.id,
2807                         'product_uom': line.product_uom.id,
2808                         'prodlot_id': lot_id,
2809                         'date': inv.date,
2810                     }
2811
2812                     if change > 0:
2813                         value.update( {
2814                             'product_qty': change,
2815                             'location_id': location_id,
2816                             'location_dest_id': line.location_id.id,
2817                         })
2818                     else:
2819                         value.update( {
2820                             'product_qty': -change,
2821                             'location_id': line.location_id.id,
2822                             'location_dest_id': location_id,
2823                         })
2824                     move_ids.append(self._inventory_line_hook(cr, uid, line, value))
2825             self.write(cr, uid, [inv.id], {'state': 'confirm', 'move_ids': [(6, 0, move_ids)]})
2826             self.pool.get('stock.move').action_confirm(cr, uid, move_ids, context=context)
2827         return True
2828
2829     def action_cancel_draft(self, cr, uid, ids, context=None):
2830         """ Cancels the stock move and change inventory state to draft.
2831         @return: True
2832         """
2833         for inv in self.browse(cr, uid, ids, context=context):
2834             self.pool.get('stock.move').action_cancel(cr, uid, [x.id for x in inv.move_ids], context=context)
2835             self.write(cr, uid, [inv.id], {'state':'draft'}, context=context)
2836         return True
2837
2838     def action_cancel_inventory(self, cr, uid, ids, context=None):
2839         """ Cancels both stock move and inventory
2840         @return: True
2841         """
2842         move_obj = self.pool.get('stock.move')
2843         account_move_obj = self.pool.get('account.move')
2844         for inv in self.browse(cr, uid, ids, context=context):
2845             move_obj.action_cancel(cr, uid, [x.id for x in inv.move_ids], context=context)
2846             for move in inv.move_ids:
2847                  account_move_ids = account_move_obj.search(cr, uid, [('name', '=', move.name)])
2848                  if account_move_ids:
2849                      account_move_data_l = account_move_obj.read(cr, uid, account_move_ids, ['state'], context=context)
2850                      for account_move in account_move_data_l:
2851                          if account_move['state'] == 'posted':
2852                              raise osv.except_osv(_('User Error!'),
2853                                                   _('In order to cancel this inventory, you must first unpost related journal entries.'))
2854                          account_move_obj.unlink(cr, uid, [account_move['id']], context=context)
2855             self.write(cr, uid, [inv.id], {'state': 'cancel'}, context=context)
2856         return True
2857
2858 stock_inventory()
2859
2860 class stock_inventory_line(osv.osv):
2861     _name = "stock.inventory.line"
2862     _description = "Inventory Line"
2863     _rec_name = "inventory_id"
2864     _columns = {
2865         'inventory_id': fields.many2one('stock.inventory', 'Inventory', ondelete='cascade', select=True),
2866         'location_id': fields.many2one('stock.location', 'Location', required=True),
2867         'product_id': fields.many2one('product.product', 'Product', required=True, select=True),
2868         'product_uom': fields.many2one('product.uom', 'Product Unit of Measure', required=True),
2869         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product Unit of Measure')),
2870         'company_id': fields.related('inventory_id','company_id',type='many2one',relation='res.company',string='Company',store=True, select=True, readonly=True),
2871         'prod_lot_id': fields.many2one('stock.production.lot', 'Serial Number', domain="[('product_id','=',product_id)]"),
2872         'state': fields.related('inventory_id','state',type='char',string='Status',readonly=True),
2873     }
2874
2875     def _default_stock_location(self, cr, uid, context=None):
2876         stock_location = self.pool.get('ir.model.data').get_object(cr, uid, 'stock', 'stock_location_stock')
2877         return stock_location.id
2878
2879     _defaults = {
2880         'location_id': _default_stock_location
2881     }
2882
2883     def on_change_product_id(self, cr, uid, ids, location_id, product, uom=False, to_date=False):
2884         """ Changes UoM and name if product_id changes.
2885         @param location_id: Location id
2886         @param product: Changed product_id
2887         @param uom: UoM product
2888         @return:  Dictionary of changed values
2889         """
2890         if not product:
2891             return {'value': {'product_qty': 0.0, 'product_uom': False}}
2892         obj_product = self.pool.get('product.product').browse(cr, uid, product)
2893         uom = uom or obj_product.uom_id.id
2894         amount = self.pool.get('stock.location')._product_get(cr, uid, location_id, [product], {'uom': uom, 'to_date': to_date, 'compute_child': False})[product]
2895         result = {'product_qty': amount, 'product_uom': uom}
2896         return {'value': result}
2897
2898 stock_inventory_line()
2899
2900 #----------------------------------------------------------
2901 # Stock Warehouse
2902 #----------------------------------------------------------
2903 class stock_warehouse(osv.osv):
2904     _name = "stock.warehouse"
2905     _description = "Warehouse"
2906     _columns = {
2907         'name': fields.char('Name', size=128, required=True, select=True),
2908         'company_id': fields.many2one('res.company', 'Company', required=True, select=True),
2909         'partner_id': fields.many2one('res.partner', 'Owner Address'),
2910         'lot_input_id': fields.many2one('stock.location', 'Location Input', required=True, domain=[('usage','<>','view')]),
2911         'lot_stock_id': fields.many2one('stock.location', 'Location Stock', required=True, domain=[('usage','=','internal')]),
2912         'lot_output_id': fields.many2one('stock.location', 'Location Output', required=True, domain=[('usage','<>','view')]),
2913     }
2914
2915     def _default_lot_input_stock_id(self, cr, uid, context=None):
2916         lot_input_stock = self.pool.get('ir.model.data').get_object(cr, uid, 'stock', 'stock_location_stock')
2917         return lot_input_stock.id
2918
2919     def _default_lot_output_id(self, cr, uid, context=None):
2920         lot_output = self.pool.get('ir.model.data').get_object(cr, uid, 'stock', 'stock_location_output')
2921         return lot_output.id
2922
2923     _defaults = {
2924         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.inventory', context=c),
2925         'lot_input_id': _default_lot_input_stock_id,
2926         'lot_stock_id': _default_lot_input_stock_id,
2927         'lot_output_id': _default_lot_output_id,
2928     }
2929
2930 stock_warehouse()
2931
2932 #----------------------------------------------------------
2933 # "Empty" Classes that are used to vary from the original stock.picking  (that are dedicated to the internal pickings)
2934 #   in order to offer a different usability with different views, labels, available reports/wizards...
2935 #----------------------------------------------------------
2936 class stock_picking_in(osv.osv):
2937     _name = "stock.picking.in"
2938     _inherit = "stock.picking"
2939     _table = "stock_picking"
2940     _description = "Incomming Shipments"
2941
2942     def check_access_rights(self, cr, uid, operation, raise_exception=True):
2943         #override in order to redirect the check of acces rights on the stock.picking object
2944         return self.pool.get('stock.picking').check_access_rights(cr, uid, operation, raise_exception=raise_exception)
2945
2946     def check_access_rule(self, cr, uid, ids, operation, context=None):
2947         #override in order to redirect the check of acces rules on the stock.picking object
2948         return self.pool.get('stock.picking').check_access_rule(cr, uid, ids, operation, context=context)
2949
2950     def _workflow_trigger(self, cr, uid, ids, trigger, context=None):
2951         #override in order to trigger the workflow of stock.picking at the end of create, write and unlink operation
2952         #instead of it's own workflow (which is not existing)
2953         return self.pool.get('stock.picking')._workflow_trigger(cr, uid, ids, trigger, context=context)
2954
2955     def _workflow_signal(self, cr, uid, ids, signal, context=None):
2956         #override in order to fire the workflow signal on given stock.picking workflow instance
2957         #instead of it's own workflow (which is not existing)
2958         return self.pool.get('stock.picking')._workflow_signal(cr, uid, ids, signal, context=context)
2959
2960     _columns = {
2961         'backorder_id': fields.many2one('stock.picking.in', '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),
2962         'state': fields.selection(
2963             [('draft', 'Draft'),
2964             ('auto', 'Waiting Another Operation'),
2965             ('confirmed', 'Waiting Availability'),
2966             ('assigned', 'Ready to Receive'),
2967             ('done', 'Received'),
2968             ('cancel', 'Cancelled'),],
2969             'State', readonly=True, select=True,
2970             help="""* Draft: not confirmed yet and will not be scheduled until confirmed\n
2971                  * Waiting Another Operation: waiting for another move to proceed before it becomes automatically available (e.g. in Make-To-Order flows)\n
2972                  * Waiting Availability: still waiting for the availability of products\n
2973                  * Ready to Receive: products reserved, simply waiting for confirmation.\n
2974                  * Received: has been processed, can't be modified or cancelled anymore\n
2975                  * Cancelled: has been cancelled, can't be confirmed anymore"""),
2976     }
2977     _defaults = {
2978         'type': 'in',
2979     }
2980
2981 class stock_picking_out(osv.osv):
2982     _name = "stock.picking.out"
2983     _inherit = "stock.picking"
2984     _table = "stock_picking"
2985     _description = "Delivery Orders"
2986
2987     def check_access_rights(self, cr, uid, operation, raise_exception=True):
2988         #override in order to redirect the check of acces rights on the stock.picking object
2989         return self.pool.get('stock.picking').check_access_rights(cr, uid, operation, raise_exception=raise_exception)
2990
2991     def check_access_rule(self, cr, uid, ids, operation, context=None):
2992         #override in order to redirect the check of acces rules on the stock.picking object
2993         return self.pool.get('stock.picking').check_access_rule(cr, uid, ids, operation, context=context)
2994
2995     def _workflow_trigger(self, cr, uid, ids, trigger, context=None):
2996         #override in order to trigger the workflow of stock.picking at the end of create, write and unlink operation
2997         #instead of it's own workflow (which is not existing)
2998         return self.pool.get('stock.picking')._workflow_trigger(cr, uid, ids, trigger, context=context)
2999
3000     def _workflow_signal(self, cr, uid, ids, signal, context=None):
3001         #override in order to fire the workflow signal on given stock.picking workflow instance
3002         #instead of it's own workflow (which is not existing)
3003         return self.pool.get('stock.picking')._workflow_signal(cr, uid, ids, signal, context=context)
3004
3005     _columns = {
3006         'backorder_id': fields.many2one('stock.picking.out', '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),
3007         'state': fields.selection(
3008             [('draft', 'Draft'),
3009             ('auto', 'Waiting Another Operation'),
3010             ('confirmed', 'Waiting Availability'),
3011             ('assigned', 'Ready to Deliver'),
3012             ('done', 'Delivered'),
3013             ('cancel', 'Cancelled'),],
3014             'State', readonly=True, select=True,
3015             help="""* Draft: not confirmed yet and will not be scheduled until confirmed\n
3016                  * Waiting Another Operation: waiting for another move to proceed before it becomes automatically available (e.g. in Make-To-Order flows)\n
3017                  * Waiting Availability: still waiting for the availability of products\n
3018                  * Ready to Deliver: products reserved, simply waiting for confirmation.\n
3019                  * Delivered: has been processed, can't be modified or cancelled anymore\n
3020                  * Cancelled: has been cancelled, can't be confirmed anymore"""),
3021     }
3022     _defaults = {
3023         'type': 'out',
3024     }
3025
3026 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: