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