addons/stock/stock.py

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