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