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