addons/stock/stock.py

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