addons/stock/stock.py

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