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
 802                 if move.product_qty != 0.0:
 803                     return False
 804                 else:
 805                     move.write({'state': 'done'})
 806         return True
 807
 808     def test_assigned(self, cr, uid, ids):
 809         """ Tests whether the move is in assigned state or not.
 810         @return: True or False
 811         """
 812         ok = True
 813         for pick in self.browse(cr, uid, ids):
 814             mt = pick.move_type
 815             for move in pick.move_lines:
 816                 if (move.state in ('confirmed', 'draft')) and (mt == 'one'):
 817                     return False
 818                 if (mt == 'direct') and (move.state == 'assigned') and (move.product_qty):
 819                     return True
 820                 ok = ok and (move.state in ('cancel', 'done', 'assigned'))
 821         return ok
 822
 823     def action_cancel(self, cr, uid, ids, context=None):
 824         """ Changes picking state to cancel.
 825         @return: True
 826         """
 827         for pick in self.browse(cr, uid, ids, context=context):
 828             ids2 = [move.id for move in pick.move_lines]
 829             self.pool.get('stock.move').action_cancel(cr, uid, ids2, context)
 830         self.write(cr, uid, ids, {'state': 'cancel', 'invoice_state': 'none'})
 831         self.log_picking(cr, uid, ids, context=context)
 832         return True
 833
 834     #
 835     # TODO: change and create a move if not parents
 836     #
 837     def action_done(self, cr, uid, ids, context=None):
 838         """ Changes picking state to done.
 839         @return: True
 840         """
 841         self.write(cr, uid, ids, {'state': 'done', 'date_done': time.strftime('%Y-%m-%d %H:%M:%S')})
 842         return True
 843
 844     def action_move(self, cr, uid, ids, context=None):
 845         """ Changes move state to assigned.
 846         @return: True
 847         """
 848         for pick in self.browse(cr, uid, ids, context=context):
 849             todo = []
 850             for move in pick.move_lines:
 851                 if move.state == 'assigned':
 852                     todo.append(move.id)
 853             if len(todo):
 854                 self.pool.get('stock.move').action_done(cr, uid, todo,
 855                         context=context)
 856         return True
 857
 858     def get_currency_id(self, cr, uid, picking):
 859         return False
 860
 861     def _get_payment_term(self, cr, uid, picking):
 862         """ Gets payment term from partner.
 863         @return: Payment term
 864         """
 865         partner = picking.address_id.partner_id
 866         return partner.property_payment_term and partner.property_payment_term.id or False
 867
 868     def _get_address_invoice(self, cr, uid, picking):
 869         """ Gets invoice address of a partner
 870         @return {'contact': address, 'invoice': address} for invoice
 871         """
 872         partner_obj = self.pool.get('res.partner')
 873         partner = (picking.purchase_id and picking.purchase_id.partner_id) or (picking.sale_id and picking.sale_id.partner_id) or picking.address_id.partner_id
 874
 875         return partner_obj.address_get(cr, uid, [partner.id],
 876                 ['contact', 'invoice'])
 877
 878     def _get_comment_invoice(self, cr, uid, picking):
 879         """
 880         @return: comment string for invoice
 881         """
 882         return picking.note or ''
 883
 884     def _get_price_unit_invoice(self, cr, uid, move_line, type, context=None):
 885         """ Gets price unit for invoice
 886         @param move_line: Stock move lines
 887         @param type: Type of invoice
 888         @return: The price unit for the move line
 889         """
 890         if context is None:
 891             context = {}
 892
 893         if type in ('in_invoice', 'in_refund'):
 894             # Take the user company and pricetype
 895             context['currency_id'] = move_line.company_id.currency_id.id
 896             amount_unit = move_line.product_id.price_get('standard_price', context)[move_line.product_id.id]
 897             return amount_unit
 898         else:
 899             return move_line.product_id.list_price
 900
 901     def _get_discount_invoice(self, cr, uid, move_line):
 902         '''Return the discount for the move line'''
 903         return 0.0
 904
 905     def _get_taxes_invoice(self, cr, uid, move_line, type):
 906         """ Gets taxes on invoice
 907         @param move_line: Stock move lines
 908         @param type: Type of invoice
 909         @return: Taxes Ids for the move line
 910         """
 911         if type in ('in_invoice', 'in_refund'):
 912             taxes = move_line.product_id.supplier_taxes_id
 913         else:
 914             taxes = move_line.product_id.taxes_id
 915
 916         if move_line.picking_id and move_line.picking_id.address_id and move_line.picking_id.address_id.partner_id:
 917             return self.pool.get('account.fiscal.position').map_tax(
 918                 cr,
 919                 uid,
 920                 move_line.picking_id.address_id.partner_id.property_account_position,
 921                 taxes
 922             )
 923         else:
 924             return map(lambda x: x.id, taxes)
 925
 926     def _get_account_analytic_invoice(self, cr, uid, picking, move_line):
 927         return False
 928
 929     def _invoice_line_hook(self, cr, uid, move_line, invoice_line_id):
 930         '''Call after the creation of the invoice line'''
 931         return
 932
 933     def _invoice_hook(self, cr, uid, picking, invoice_id):
 934         '''Call after the creation of the invoice'''
 935         return
 936
 937     def _get_invoice_type(self, pick):
 938         src_usage = dest_usage = None
 939         inv_type = None
 940         if pick.invoice_state == '2binvoiced':
 941             if pick.move_lines:
 942                 src_usage = pick.move_lines[0].location_id.usage
 943                 dest_usage = pick.move_lines[0].location_dest_id.usage
 944             if pick.type == 'out' and dest_usage == 'supplier':
 945                 inv_type = 'in_refund'
 946             elif pick.type == 'out' and dest_usage == 'customer':
 947                 inv_type = 'out_invoice'
 948             elif pick.type == 'in' and src_usage == 'supplier':
 949                 inv_type = 'in_invoice'
 950             elif pick.type == 'in' and src_usage == 'customer':
 951                 inv_type = 'out_refund'
 952             else:
 953                 inv_type = 'out_invoice'
 954         return inv_type
 955
 956     def action_invoice_create(self, cr, uid, ids, journal_id=False,
 957             group=False, type='out_invoice', context=None):
 958         """ Creates invoice based on the invoice state selected for picking.
 959         @param journal_id: Id of journal
 960         @param group: Whether to create a group invoice or not
 961         @param type: Type invoice to be created
 962         @return: Ids of created invoices for the pickings
 963         """
 964         if context is None:
 965             context = {}
 966
 967         invoice_obj = self.pool.get('account.invoice')
 968         invoice_line_obj = self.pool.get('account.invoice.line')
 969         invoices_group = {}
 970         res = {}
 971         inv_type = type
 972         for picking in self.browse(cr, uid, ids, context=context):
 973             if picking.invoice_state != '2binvoiced':
 974                 continue
 975             payment_term_id = False
 976             partner = (picking.purchase_id and picking.purchase_id.partner_id) or (picking.sale_id and picking.sale_id.partner_id) or (picking.address_id and picking.address_id.partner_id)
 977             if not partner:
 978                 raise osv.except_osv(_('Error, no partner !'),
 979                     _('Please put a partner on the picking list if you want to generate invoice.'))
 980
 981             if not inv_type:
 982                 inv_type = self._get_invoice_type(picking)
 983
 984             if inv_type in ('out_invoice', 'out_refund'):
 985                 account_id = partner.property_account_receivable.id
 986                 payment_term_id = self._get_payment_term(cr, uid, picking)
 987             else:
 988                 account_id = partner.property_account_payable.id
 989
 990             address_contact_id, address_invoice_id = \
 991                     self._get_address_invoice(cr, uid, picking).values()
 992
 993             comment = self._get_comment_invoice(cr, uid, picking)
 994             if group and partner.id in invoices_group:
 995                 invoice_id = invoices_group[partner.id]
 996                 invoice = invoice_obj.browse(cr, uid, invoice_id)
 997                 invoice_vals = {
 998                     'name': (invoice.name or '') + ', ' + (picking.name or ''),
 999                     'origin': (invoice.origin or '') + ', ' + (picking.name or '') + (picking.origin and (':' + picking.origin) or ''),
1000                     'comment': (comment and (invoice.comment and invoice.comment+"\n"+comment or comment)) or (invoice.comment and invoice.comment or ''),
1001                     'date_invoice':context.get('date_inv',False),
1002                     'user_id':uid
1003                 }
1004                 invoice_obj.write(cr, uid, [invoice_id], invoice_vals, context=context)
1005             else:
1006                 invoice_vals = {
1007                     'name': picking.name,
1008                     'origin': (picking.name or '') + (picking.origin and (':' + picking.origin) or ''),
1009                     'type': inv_type,
1010                     'account_id': account_id,
1011                     'partner_id': partner.id,
1012                     'address_invoice_id': address_invoice_id,
1013                     'address_contact_id': address_contact_id,
1014                     'comment': comment,
1015                     'payment_term': payment_term_id,
1016                     'fiscal_position': partner.property_account_position.id,
1017                     'date_invoice': context.get('date_inv',False),
1018                     'company_id': picking.company_id.id,
1019                     'user_id':uid
1020                 }
1021                 cur_id = self.get_currency_id(cr, uid, picking)
1022                 if cur_id:
1023                     invoice_vals['currency_id'] = cur_id
1024                 if journal_id:
1025                     invoice_vals['journal_id'] = journal_id
1026                 invoice_id = invoice_obj.create(cr, uid, invoice_vals,
1027                         context=context)
1028                 invoices_group[partner.id] = invoice_id
1029             res[picking.id] = invoice_id
1030             for move_line in picking.move_lines:
1031                 if move_line.state == 'cancel':
1032                     continue
1033                 origin = move_line.picking_id.name or ''
1034                 if move_line.picking_id.origin:
1035                     origin += ':' + move_line.picking_id.origin
1036                 if group:
1037                     name = (picking.name or '') + '-' + move_line.name
1038                 else:
1039                     name = move_line.name
1040
1041                 if inv_type in ('out_invoice', 'out_refund'):
1042                     account_id = move_line.product_id.product_tmpl_id.\
1043                             property_account_income.id
1044                     if not account_id:
1045                         account_id = move_line.product_id.categ_id.\
1046                                 property_account_income_categ.id
1047                 else:
1048                     account_id = move_line.product_id.product_tmpl_id.\
1049                             property_account_expense.id
1050                     if not account_id:
1051                         account_id = move_line.product_id.categ_id.\
1052                                 property_account_expense_categ.id
1053
1054                 price_unit = self._get_price_unit_invoice(cr, uid,
1055                         move_line, inv_type)
1056                 discount = self._get_discount_invoice(cr, uid, move_line)
1057                 tax_ids = self._get_taxes_invoice(cr, uid, move_line, inv_type)
1058                 account_analytic_id = self._get_account_analytic_invoice(cr, uid, picking, move_line)
1059
1060                 #set UoS if it's a sale and the picking doesn't have one
1061                 uos_id = move_line.product_uos and move_line.product_uos.id or False
1062                 if not uos_id and inv_type in ('out_invoice', 'out_refund'):
1063                     uos_id = move_line.product_uom.id
1064
1065                 account_id = self.pool.get('account.fiscal.position').map_account(cr, uid, partner.property_account_position, account_id)
1066                 invoice_line_id = invoice_line_obj.create(cr, uid, {
1067                     'name': name,
1068                     'origin': origin,
1069                     'invoice_id': invoice_id,
1070                     'uos_id': uos_id,
1071                     'product_id': move_line.product_id.id,
1072                     'account_id': account_id,
1073                     'price_unit': price_unit,
1074                     'discount': discount,
1075                     'quantity': move_line.product_uos_qty or move_line.product_qty,
1076                     'invoice_line_tax_id': [(6, 0, tax_ids)],
1077                     'account_analytic_id': account_analytic_id,
1078                 }, context=context)
1079                 self._invoice_line_hook(cr, uid, move_line, invoice_line_id)
1080
1081             invoice_obj.button_compute(cr, uid, [invoice_id], context=context,
1082                     set_total=(inv_type in ('in_invoice', 'in_refund')))
1083             self.write(cr, uid, [picking.id], {
1084                 'invoice_state': 'invoiced',
1085                 }, context=context)
1086             self._invoice_hook(cr, uid, picking, invoice_id)
1087         self.write(cr, uid, res.keys(), {
1088             'invoice_state': 'invoiced',
1089             }, context=context)
1090         return res
1091
1092     def test_done(self, cr, uid, ids, context=None):
1093         """ Test whether the move lines are done or not.
1094         @return: True or False
1095         """
1096         ok = False
1097         for pick in self.browse(cr, uid, ids, context=context):
1098             if not pick.move_lines:
1099                 return True
1100             for move in pick.move_lines:
1101                 if move.state not in ('cancel','done'):
1102                     return False
1103                 if move.state=='done':
1104                     ok = True
1105         return ok
1106
1107     def test_cancel(self, cr, uid, ids, context=None):
1108         """ Test whether the move lines are canceled or not.
1109         @return: True or False
1110         """
1111         for pick in self.browse(cr, uid, ids, context=context):
1112             for move in pick.move_lines:
1113                 if move.state not in ('cancel',):
1114                     return False
1115         return True
1116
1117     def allow_cancel(self, cr, uid, ids, context=None):
1118         for pick in self.browse(cr, uid, ids, context=context):
1119             if not pick.move_lines:
1120                 return True
1121             for move in pick.move_lines:
1122                 if move.state == 'done':
1123                     raise osv.except_osv(_('Error'), _('You cannot cancel picking because stock move is in done state !'))
1124         return True
1125     def unlink(self, cr, uid, ids, context=None):
1126         move_obj = self.pool.get('stock.move')
1127         if context is None:
1128             context = {}
1129         for pick in self.browse(cr, uid, ids, context=context):
1130             if pick.state in ['done','cancel']:
1131                 raise osv.except_osv(_('Error'), _('You cannot remove the picking which is in %s state !')%(pick.state,))
1132             elif pick.state in ['confirmed','assigned', 'draft']:
1133                 ids2 = [move.id for move in pick.move_lines]
1134                 ctx = context.copy()
1135                 ctx.update({'call_unlink':True})
1136                 if pick.state != 'draft':
1137                     #Cancelling the move in order to affect Virtual stock of product
1138                     move_obj.action_cancel(cr, uid, ids2, ctx)
1139                 #Removing the move
1140                 move_obj.unlink(cr, uid, ids2, ctx)
1141
1142         return super(stock_picking, self).unlink(cr, uid, ids, context=context)
1143
1144     # FIXME: needs refactoring, this code is partially duplicated in stock_move.do_partial()!
1145     def do_partial(self, cr, uid, ids, partial_datas, context=None):
1146         """ Makes partial picking and moves done.
1147         @param partial_datas : Dictionary containing details of partial picking
1148                           like partner_id, address_id, delivery_date,
1149                           delivery moves with product_id, product_qty, uom
1150         @return: Dictionary of values
1151         """
1152         if context is None:
1153             context = {}
1154         else:
1155             context = dict(context)
1156         res = {}
1157         move_obj = self.pool.get('stock.move')
1158         product_obj = self.pool.get('product.product')
1159         currency_obj = self.pool.get('res.currency')
1160         uom_obj = self.pool.get('product.uom')
1161         sequence_obj = self.pool.get('ir.sequence')
1162         wf_service = netsvc.LocalService("workflow")
1163         for pick in self.browse(cr, uid, ids, context=context):
1164             new_picking = None
1165             complete, too_many, too_few = [], [], []
1166             move_product_qty = {}
1167             prodlot_ids = {}
1168             for move in pick.move_lines:
1169                 if move.state in ('done', 'cancel'):
1170                     continue
1171                 partial_data = partial_datas.get('move%s'%(move.id), False)
1172                 assert partial_data, _('Missing partial picking data for move #%s') % (move.id)
1173                 product_qty = partial_data.get('product_qty',0.0)
1174                 move_product_qty[move.id] = product_qty
1175                 product_uom = partial_data.get('product_uom',False)
1176                 product_price = partial_data.get('product_price',0.0)
1177                 product_currency = partial_data.get('product_currency',False)
1178                 prodlot_id = partial_data.get('prodlot_id')
1179                 prodlot_ids[move.id] = prodlot_id
1180                 if move.product_qty == product_qty:
1181                     complete.append(move)
1182                 elif move.product_qty > product_qty:
1183                     too_few.append(move)
1184                 else:
1185                     too_many.append(move)
1186
1187                 # Average price computation
1188                 if (pick.type == 'in') and (move.product_id.cost_method == 'average'):
1189                     product = product_obj.browse(cr, uid, move.product_id.id)
1190                     move_currency_id = move.company_id.currency_id.id
1191                     context['currency_id'] = move_currency_id
1192                     qty = uom_obj._compute_qty(cr, uid, product_uom, product_qty, product.uom_id.id)
1193                     if qty > 0:
1194                         new_price = currency_obj.compute(cr, uid, product_currency,
1195                                 move_currency_id, product_price)
1196                         new_price = uom_obj._compute_price(cr, uid, product_uom, new_price,
1197                                 product.uom_id.id)
1198                         if product.qty_available <= 0:
1199                             new_std_price = new_price
1200                         else:
1201                             # Get the standard price
1202                             amount_unit = product.price_get('standard_price', context)[product.id]
1203                             new_std_price = ((amount_unit * product.qty_available)\
1204                                 + (new_price * qty))/(product.qty_available + qty)
1205
1206                         # Write the field according to price type field
1207                         product_obj.write(cr, uid, [product.id], {'standard_price': new_std_price})
1208
1209                         # Record the values that were chosen in the wizard, so they can be
1210                         # used for inventory valuation if real-time valuation is enabled.
1211                         move_obj.write(cr, uid, [move.id],
1212                                 {'price_unit': product_price,
1213                                  'price_currency_id': product_currency})
1214
1215
1216             for move in too_few:
1217                 product_qty = move_product_qty[move.id]
1218
1219                 if not new_picking:
1220                     new_picking = self.copy(cr, uid, pick.id,
1221                             {
1222                                 'name': sequence_obj.get(cr, uid, 'stock.picking.%s'%(pick.type)),
1223                                 'move_lines' : [],
1224                                 'state':'draft',
1225                             })
1226                 if product_qty != 0:
1227                     defaults = {
1228                             'product_qty' : product_qty,
1229                             'product_uos_qty': product_qty, #TODO: put correct uos_qty
1230                             'picking_id' : new_picking,
1231                             'state': 'assigned',
1232                             'move_dest_id': False,
1233                             'price_unit': move.price_unit,
1234                     }
1235                     prodlot_id = prodlot_ids[move.id]
1236                     if prodlot_id:
1237                         defaults.update(prodlot_id=prodlot_id)
1238                     move_obj.copy(cr, uid, move.id, defaults)
1239
1240                 move_obj.write(cr, uid, [move.id],
1241                         {
1242                             'product_qty' : move.product_qty - product_qty,
1243                             'product_uos_qty':move.product_qty - product_qty, #TODO: put correct uos_qty
1244                         })
1245
1246             if new_picking:
1247                 move_obj.write(cr, uid, [c.id for c in complete], {'picking_id': new_picking})
1248                 for move in complete:
1249                     if prodlot_ids.get(move.id):
1250                         move_obj.write(cr, uid, move.id, {'prodlot_id': prodlot_ids[move.id]})
1251             for move in too_many:
1252                 product_qty = move_product_qty[move.id]
1253                 defaults = {
1254                     'product_qty' : product_qty,
1255                     'product_uos_qty': product_qty, #TODO: put correct uos_qty
1256                 }
1257                 prodlot_id = prodlot_ids.get(move.id)
1258                 if prodlot_ids.get(move.id):
1259                     defaults.update(prodlot_id=prodlot_id)
1260                 if new_picking:
1261                     defaults.update(picking_id=new_picking)
1262                 move_obj.write(cr, uid, [move.id], defaults)
1263
1264
1265             # At first we confirm the new picking (if necessary)
1266             if new_picking:
1267                 wf_service.trg_validate(uid, 'stock.picking', new_picking, 'button_confirm', cr)
1268                 # Then we finish the good picking
1269                 self.write(cr, uid, [pick.id], {'backorder_id': new_picking})
1270                 self.action_move(cr, uid, [new_picking])
1271                 wf_service.trg_validate(uid, 'stock.picking', new_picking, 'button_done', cr)
1272                 wf_service.trg_write(uid, 'stock.picking', pick.id, cr)
1273                 delivered_pack_id = new_picking
1274             else:
1275                 self.action_move(cr, uid, [pick.id])
1276                 wf_service.trg_validate(uid, 'stock.picking', pick.id, 'button_done', cr)
1277                 delivered_pack_id = pick.id
1278
1279             delivered_pack = self.browse(cr, uid, delivered_pack_id, context=context)
1280             res[pick.id] = {'delivered_picking': delivered_pack.id or False}
1281
1282         return res
1283
1284     def log_picking(self, cr, uid, ids, context=None):
1285         """ This function will create log messages for picking.
1286         @param cr: the database cursor
1287         @param uid: the current user's ID for security checks,
1288         @param ids: List of Picking Ids
1289         @param context: A standard dictionary for contextual values
1290         """
1291         if context is None:
1292             context = {}
1293         data_obj = self.pool.get('ir.model.data')
1294         for pick in self.browse(cr, uid, ids, context=context):
1295             msg=''
1296             if pick.auto_picking:
1297                 continue
1298             type_list = {
1299                 'out':_("Delivery Order"),
1300                 'in':_('Reception'),
1301                 'internal': _('Internal picking'),
1302             }
1303             view_list = {
1304                 'out': 'view_picking_out_form',
1305                 'in': 'view_picking_in_form',
1306                 'internal': 'view_picking_form',
1307             }
1308             message = type_list.get(pick.type, _('Document')) + " '" + (pick.name or '?') + "' "
1309             if pick.min_date:
1310                 msg= _(' for the ')+ datetime.strptime(pick.min_date, '%Y-%m-%d %H:%M:%S').strftime('%m/%d/%Y')
1311             state_list = {
1312                 'confirmed': _("is scheduled") + msg +'.',
1313                 'assigned': _('is ready to process.'),
1314                 'cancel': _('is cancelled.'),
1315                 'done': _('is done.'),
1316                 'draft':_('is in draft state.'),
1317             }
1318             res = data_obj.get_object_reference(cr, uid, 'stock', view_list.get(pick.type, 'view_picking_form'))
1319             context.update({'view_id': res and res[1] or False})
1320             message += state_list[pick.state]
1321             self.log(cr, uid, pick.id, message, context=context)
1322         return True
1323
1324 stock_picking()
1325
1326 class stock_production_lot(osv.osv):
1327
1328     def name_get(self, cr, uid, ids, context=None):
1329         if not ids:
1330             return []
1331         reads = self.read(cr, uid, ids, ['name', 'prefix', 'ref'], context)
1332         res = []
1333         for record in reads:
1334             name = record['name']
1335             prefix = record['prefix']
1336             if prefix:
1337                 name = prefix + '/' + name
1338             if record['ref']:
1339                 name = '%s [%s]' % (name, record['ref'])
1340             res.append((record['id'], name))
1341         return res
1342
1343     _name = 'stock.production.lot'
1344     _description = 'Production lot'
1345
1346     def _get_stock(self, cr, uid, ids, field_name, arg, context=None):
1347         """ Gets stock of products for locations
1348         @return: Dictionary of values
1349         """
1350         if context is None:
1351             context = {}
1352         if 'location_id' not in context:
1353             locations = self.pool.get('stock.location').search(cr, uid, [('usage', '=', 'internal')], context=context)
1354         else:
1355             locations = context['location_id'] and [context['location_id']] or []
1356
1357         if isinstance(ids, (int, long)):
1358             ids = [ids]
1359
1360         res = {}.fromkeys(ids, 0.0)
1361         if locations:
1362             cr.execute('''select
1363                     prodlot_id,
1364                     sum(qty)
1365                 from
1366                     stock_report_prodlots
1367                 where
1368                     location_id IN %s and prodlot_id IN %s group by prodlot_id''',(tuple(locations),tuple(ids),))
1369             res.update(dict(cr.fetchall()))
1370
1371         return res
1372
1373     def _stock_search(self, cr, uid, obj, name, args, context=None):
1374         """ Searches Ids of products
1375         @return: Ids of locations
1376         """
1377         locations = self.pool.get('stock.location').search(cr, uid, [('usage', '=', 'internal')])
1378         cr.execute('''select
1379                 prodlot_id,
1380                 sum(qty)
1381             from
1382                 stock_report_prodlots
1383             where
1384                 location_id IN %s group by prodlot_id
1385             having  sum(qty) '''+ str(args[0][1]) + str(args[0][2]),(tuple(locations),))
1386         res = cr.fetchall()
1387         ids = [('id', 'in', map(lambda x: x[0], res))]
1388         return ids
1389
1390     _columns = {
1391         'name': fields.char('Production Lot', size=64, required=True, help="Unique production lot, will be displayed as: PREFIX/SERIAL [INT_REF]"),
1392         'ref': fields.char('Internal Reference', size=256, help="Internal reference number in case it differs from the manufacturer's serial number"),
1393         'prefix': fields.char('Prefix', size=64, help="Optional prefix to prepend when displaying this serial number: PREFIX/SERIAL [INT_REF]"),
1394         'product_id': fields.many2one('product.product', 'Product', required=True, domain=[('type', '<>', 'service')]),
1395         'date': fields.datetime('Creation Date', required=True),
1396         'stock_available': fields.function(_get_stock, fnct_search=_stock_search, method=True, type="float", string="Available", select=True,
1397             help="Current quantity of products with this Production Lot Number available in company warehouses",
1398             digits_compute=dp.get_precision('Product UoM')),
1399         'revisions': fields.one2many('stock.production.lot.revision', 'lot_id', 'Revisions'),
1400         'company_id': fields.many2one('res.company', 'Company', select=True),
1401         'move_ids': fields.one2many('stock.move', 'prodlot_id', 'Moves for this production lot', readonly=True),
1402     }
1403     _defaults = {
1404         'date':  time.strftime('%Y-%m-%d %H:%M:%S'),
1405         'name': lambda x, y, z, c: x.pool.get('ir.sequence').get(y, z, 'stock.lot.serial'),
1406         'product_id': lambda x, y, z, c: c.get('product_id', False),
1407     }
1408     _sql_constraints = [
1409         ('name_ref_uniq', 'unique (name, ref)', 'The combination of serial number and internal reference must be unique !'),
1410     ]
1411     def action_traceability(self, cr, uid, ids, context=None):
1412         """ It traces the information of a product
1413         @param self: The object pointer.
1414         @param cr: A database cursor
1415         @param uid: ID of the user currently logged in
1416         @param ids: List of IDs selected
1417         @param context: A standard dictionary
1418         @return: A dictionary of values
1419         """
1420         value=self.pool.get('action.traceability').action_traceability(cr,uid,ids,context)
1421         return value
1422 stock_production_lot()
1423
1424 class stock_production_lot_revision(osv.osv):
1425     _name = 'stock.production.lot.revision'
1426     _description = 'Production lot revisions'
1427
1428     _columns = {
1429         'name': fields.char('Revision Name', size=64, required=True),
1430         'description': fields.text('Description'),
1431         'date': fields.date('Revision Date'),
1432         'indice': fields.char('Revision Number', size=16),
1433         'author_id': fields.many2one('res.users', 'Author'),
1434         'lot_id': fields.many2one('stock.production.lot', 'Production lot', select=True, ondelete='cascade'),
1435         'company_id': fields.related('lot_id','company_id',type='many2one',relation='res.company',string='Company',store=True),
1436     }
1437
1438     _defaults = {
1439         'author_id': lambda x, y, z, c: z,
1440         'date': time.strftime('%Y-%m-%d'),
1441     }
1442
1443 stock_production_lot_revision()
1444
1445 # ----------------------------------------------------
1446 # Move
1447 # ----------------------------------------------------
1448
1449 #
1450 # Fields:
1451 #   location_dest_id is only used for predicting futur stocks
1452 #
1453 class stock_move(osv.osv):
1454
1455     def _getSSCC(self, cr, uid, context=None):
1456         cr.execute('select id from stock_tracking where create_uid=%s order by id desc limit 1', (uid,))
1457         res = cr.fetchone()
1458         return (res and res[0]) or False
1459     _name = "stock.move"
1460     _description = "Stock Move"
1461     _order = 'date_expected desc, id'
1462     _log_create = False
1463
1464     def name_get(self, cr, uid, ids, context=None):
1465         res = []
1466         for line in self.browse(cr, uid, ids, context=context):
1467             res.append((line.id, (line.product_id.code or '/')+': '+line.location_id.name+' > '+line.location_dest_id.name))
1468         return res
1469
1470     def _check_tracking(self, cr, uid, ids, context=None):
1471         """ Checks if production lot is assigned to stock move or not.
1472         @return: True or False
1473         """
1474         for move in self.browse(cr, uid, ids, context=context):
1475             if not move.prodlot_id and \
1476                (move.state == 'done' and \
1477                ( \
1478                    (move.product_id.track_production and move.location_id.usage == 'production') or \
1479                    (move.product_id.track_production and move.location_dest_id.usage == 'production') or \
1480                    (move.product_id.track_incoming and move.location_id.usage == 'supplier') or \
1481                    (move.product_id.track_outgoing and move.location_dest_id.usage == 'customer') \
1482                )):
1483                 return False
1484         return True
1485
1486     def _check_product_lot(self, cr, uid, ids, context=None):
1487         """ Checks whether move is done or not and production lot is assigned to that move.
1488         @return: True or False
1489         """
1490         for move in self.browse(cr, uid, ids, context=context):
1491             if move.prodlot_id and move.state == 'done' and (move.prodlot_id.product_id.id != move.product_id.id):
1492                 return False
1493         return True
1494
1495     _columns = {
1496         'name': fields.char('Name', size=64, required=True, select=True),
1497         'priority': fields.selection([('0', 'Not urgent'), ('1', 'Urgent')], 'Priority'),
1498         'create_date': fields.datetime('Creation Date', readonly=True),
1499         'date': fields.datetime('Date', required=True, help="Move date: scheduled date until move is done, then date of actual move processing", readonly=True),
1500         'date_expected': fields.datetime('Scheduled Date', states={'done': [('readonly', True)]},required=True, help="Scheduled date for the processing of this move"),
1501         'product_id': fields.many2one('product.product', 'Product', required=True, select=True, domain=[('type','<>','service')],states={'done': [('readonly', True)]}),
1502
1503         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product UoM'), required=True,states={'done': [('readonly', True)]}),
1504         'product_uom': fields.many2one('product.uom', 'Unit of Measure', required=True,states={'done': [('readonly', True)]}),
1505         'product_uos_qty': fields.float('Quantity (UOS)', digits_compute=dp.get_precision('Product UoM')),
1506         'product_uos': fields.many2one('product.uom', 'Product UOS'),
1507         'product_packaging': fields.many2one('product.packaging', 'Packaging', help="It specifies attributes of packaging like type, quantity of packaging,etc."),
1508
1509         '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."),
1510         '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."),
1511         'address_id': fields.many2one('res.partner.address', 'Destination Address', help="Optional address where goods are to be delivered, specifically used for allotment"),
1512
1513         '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),
1514         'tracking_id': fields.many2one('stock.tracking', 'Pack', select=True, states={'done': [('readonly', True)]}, help="Logistical shipping unit: pallet, box, pack ..."),
1515
1516         'auto_validate': fields.boolean('Auto Validate'),
1517
1518         'move_dest_id': fields.many2one('stock.move', 'Destination Move', help="Optional: next stock move when chaining them", select=True),
1519         'move_history_ids': fields.many2many('stock.move', 'stock_move_history_ids', 'parent_id', 'child_id', 'Move History (child moves)'),
1520         'move_history_ids2': fields.many2many('stock.move', 'stock_move_history_ids', 'child_id', 'parent_id', 'Move History (parent moves)'),
1521         'picking_id': fields.many2one('stock.picking', 'Reference', select=True,states={'done': [('readonly', True)]}),
1522         'note': fields.text('Notes'),
1523         'state': fields.selection([('draft', 'Draft'), ('waiting', 'Waiting'), ('confirmed', 'Confirmed'), ('assigned', 'Available'), ('done', 'Done'), ('cancel', 'Cancelled')], 'State', readonly=True, select=True,
1524                                   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\'.\
1525                                   \nThe state is \'Waiting\' if the move is waiting for another one.'),
1526         '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)"),
1527         '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)"),
1528         'company_id': fields.many2one('res.company', 'Company', required=True, select=True),
1529         'partner_id': fields.related('picking_id','address_id','partner_id',type='many2one', relation="res.partner", string="Partner", store=True, select=True),
1530         'backorder_id': fields.related('picking_id','backorder_id',type='many2one', relation="stock.picking", string="Back Order", select=True),
1531         'origin': fields.related('picking_id','origin',type='char', size=64, relation="stock.picking", string="Origin", store=True),
1532
1533         # used for colors in tree views:
1534         'scrapped': fields.related('location_dest_id','scrap_location',type='boolean',relation='stock.location',string='Scrapped', readonly=True),
1535     }
1536     _constraints = [
1537         (_check_tracking,
1538             'You must assign a production lot for this product',
1539             ['prodlot_id']),
1540         (_check_product_lot,
1541             'You try to assign a lot which is not from the same product',
1542             ['prodlot_id'])]
1543
1544     def _default_location_destination(self, cr, uid, context=None):
1545         """ Gets default address of partner for destination location
1546         @return: Address id or False
1547         """
1548         if context is None:
1549             context = {}
1550         if context.get('move_line', []):
1551             if context['move_line'][0]:
1552                 if isinstance(context['move_line'][0], (tuple, list)):
1553                     return context['move_line'][0][2] and context['move_line'][0][2].get('location_dest_id',False)
1554                 else:
1555                     move_list = self.pool.get('stock.move').read(cr, uid, context['move_line'][0], ['location_dest_id'])
1556                     return move_list and move_list['location_dest_id'][0] or False
1557         if context.get('address_out_id', False):
1558             property_out = self.pool.get('res.partner.address').browse(cr, uid, context['address_out_id'], context).partner_id.property_stock_customer
1559             return property_out and property_out.id or False
1560         return False
1561
1562     def _default_location_source(self, cr, uid, context=None):
1563         """ Gets default address of partner for source location
1564         @return: Address id or False
1565         """
1566         if context is None:
1567             context = {}
1568         if context.get('move_line', []):
1569             try:
1570                 return context['move_line'][0][2]['location_id']
1571             except:
1572                 pass
1573         if context.get('address_in_id', False):
1574             return self.pool.get('res.partner.address').browse(cr, uid, context['address_in_id'], context).partner_id.property_stock_supplier.id
1575         return False
1576
1577     _defaults = {
1578         'location_id': _default_location_source,
1579         'location_dest_id': _default_location_destination,
1580         'state': 'draft',
1581         'priority': '1',
1582         'product_qty': 1.0,
1583         'scrapped' :  False,
1584         'date': time.strftime('%Y-%m-%d %H:%M:%S'),
1585         'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.move', context=c),
1586         'date_expected': time.strftime('%Y-%m-%d %H:%M:%S'),
1587     }
1588
1589     def write(self, cr, uid, ids, vals, context=None):
1590         if uid != 1:
1591             frozen_fields = set(['product_qty', 'product_uom', 'product_uos_qty', 'product_uos', 'location_id', 'location_dest_id', 'product_id'])
1592             for move in self.browse(cr, uid, ids, context=context):
1593                 if move.state == 'done':
1594                     if frozen_fields.intersection(vals):
1595                         raise osv.except_osv(_('Operation forbidden'),
1596                                              _('Quantities, UoMs, Products and Locations cannot be modified on stock moves that have already been processed (except by the Administrator)'))
1597         return  super(stock_move, self).write(cr, uid, ids, vals, context=context)
1598
1599     def copy(self, cr, uid, id, default=None, context=None):
1600         if default is None:
1601             default = {}
1602         default = default.copy()
1603         return super(stock_move, self).copy(cr, uid, id, default, context=context)
1604
1605     def _auto_init(self, cursor, context=None):
1606         res = super(stock_move, self)._auto_init(cursor, context=context)
1607         cursor.execute('SELECT indexname \
1608                 FROM pg_indexes \
1609                 WHERE indexname = \'stock_move_location_id_location_dest_id_product_id_state\'')
1610         if not cursor.fetchone():
1611             cursor.execute('CREATE INDEX stock_move_location_id_location_dest_id_product_id_state \
1612                     ON stock_move (location_id, location_dest_id, product_id, state)')
1613         return res
1614
1615     def onchange_lot_id(self, cr, uid, ids, prodlot_id=False, product_qty=False,
1616                         loc_id=False, product_id=False, uom_id=False, context=None):
1617         """ On change of production lot gives a warning message.
1618         @param prodlot_id: Changed production lot id
1619         @param product_qty: Quantity of product
1620         @param loc_id: Location id
1621         @param product_id: Product id
1622         @return: Warning message
1623         """
1624         if not prodlot_id or not loc_id:
1625             return {}
1626         ctx = context and context.copy() or {}
1627         ctx['location_id'] = loc_id
1628         ctx.update({'raise-exception': True})
1629         uom_obj = self.pool.get('product.uom')
1630         product_obj = self.pool.get('product.product')
1631         product_uom = product_obj.browse(cr, uid, product_id, context=ctx).uom_id
1632         prodlot = self.pool.get('stock.production.lot').browse(cr, uid, prodlot_id, context=ctx)
1633         location = self.pool.get('stock.location').browse(cr, uid, loc_id, context=ctx)
1634         uom = uom_obj.browse(cr, uid, uom_id, context=ctx)
1635         amount_actual = uom_obj._compute_qty_obj(cr, uid, product_uom, prodlot.stock_available, uom, context=ctx)
1636         warning = {}
1637         if (location.usage == 'internal') and (product_qty > (amount_actual or 0.0)):
1638             warning = {
1639                 'title': _('Insufficient Stock in Lot !'),
1640                 'message': _('You are moving %.2f %s products but only %.2f %s available in this lot.') % (product_qty, uom.name, amount_actual, uom.name)
1641             }
1642         return {'warning': warning}
1643
1644     def onchange_quantity(self, cr, uid, ids, product_id, product_qty,
1645                           product_uom, product_uos):
1646         """ On change of product quantity finds UoM and UoS quantities
1647         @param product_id: Product id
1648         @param product_qty: Changed Quantity of product
1649         @param product_uom: Unit of measure of product
1650         @param product_uos: Unit of sale of product
1651         @return: Dictionary of values
1652         """
1653         result = {
1654                   'product_uos_qty': 0.00
1655           }
1656
1657         if (not product_id) or (product_qty <=0.0):
1658             return {'value': result}
1659
1660         product_obj = self.pool.get('product.product')
1661         uos_coeff = product_obj.read(cr, uid, product_id, ['uos_coeff'])
1662
1663         if product_uos and product_uom and (product_uom != product_uos):
1664             result['product_uos_qty'] = product_qty * uos_coeff['uos_coeff']
1665         else:
1666             result['product_uos_qty'] = product_qty
1667
1668         return {'value': result}
1669
1670     def onchange_uos_quantity(self, cr, uid, ids, product_id, product_uos_qty,
1671                           product_uos, product_uom):
1672         """ On change of product quantity finds UoM and UoS quantities
1673         @param product_id: Product id
1674         @param product_uos_qty: Changed UoS Quantity of product
1675         @param product_uom: Unit of measure of product
1676         @param product_uos: Unit of sale of product
1677         @return: Dictionary of values
1678         """
1679         result = {
1680                   'product_qty': 0.00
1681           }
1682
1683         if (not product_id) or (product_uos_qty <=0.0):
1684             return {'value': result}
1685
1686         product_obj = self.pool.get('product.product')
1687         uos_coeff = product_obj.read(cr, uid, product_id, ['uos_coeff'])
1688
1689         if product_uos and product_uom and (product_uom != product_uos):
1690             result['product_qty'] = product_uos_qty / uos_coeff['uos_coeff']
1691         else:
1692             result['product_qty'] = product_uos_qty
1693
1694         return {'value': result}
1695
1696     def onchange_product_id(self, cr, uid, ids, prod_id=False, loc_id=False,
1697                             loc_dest_id=False, address_id=False):
1698         """ On change of product id, if finds UoM, UoS, quantity and UoS quantity.
1699         @param prod_id: Changed Product id
1700         @param loc_id: Source location id
1701         @param loc_id: Destination location id
1702         @param address_id: Address id of partner
1703         @return: Dictionary of values
1704         """
1705         if not prod_id:
1706             return {}
1707         lang = False
1708         if address_id:
1709             addr_rec = self.pool.get('res.partner.address').browse(cr, uid, address_id)
1710             if addr_rec:
1711                 lang = addr_rec.partner_id and addr_rec.partner_id.lang or False
1712         ctx = {'lang': lang}
1713
1714         product = self.pool.get('product.product').browse(cr, uid, [prod_id], context=ctx)[0]
1715         uos_id  = product.uos_id and product.uos_id.id or False
1716         result = {
1717             'product_uom': product.uom_id.id,
1718             'product_uos': uos_id,
1719             'product_qty': 1.00,
1720             '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']
1721         }
1722         if not ids:
1723             result['name'] = product.partner_ref
1724         if loc_id:
1725             result['location_id'] = loc_id
1726         if loc_dest_id:
1727             result['location_dest_id'] = loc_dest_id
1728         return {'value': result}
1729
1730     def _chain_compute(self, cr, uid, moves, context=None):
1731         """ Finds whether the location has chained location type or not.
1732         @param moves: Stock moves
1733         @return: Dictionary containing destination location with chained location type.
1734         """
1735         result = {}
1736         for m in moves:
1737             dest = self.pool.get('stock.location').chained_location_get(
1738                 cr,
1739                 uid,
1740                 m.location_dest_id,
1741                 m.picking_id and m.picking_id.address_id and m.picking_id.address_id.partner_id,
1742                 m.product_id,
1743                 context
1744             )
1745             if dest:
1746                 if dest[1] == 'transparent':
1747                     newdate = (datetime.strptime(m.date, '%Y-%m-%d %H:%M:%S') + relativedelta(days=dest[2] or 0)).strftime('%Y-%m-%d')
1748                     self.write(cr, uid, [m.id], {
1749                         'date': newdate,
1750                         'location_dest_id': dest[0].id})
1751                     if m.picking_id and (dest[3] or dest[5]):
1752                         self.pool.get('stock.picking').write(cr, uid, [m.picking_id.id], {
1753                             'stock_journal_id': dest[3] or m.picking_id.stock_journal_id.id,
1754                             'type': dest[5] or m.picking_id.type
1755                         }, context=context)
1756                     m.location_dest_id = dest[0]
1757                     res2 = self._chain_compute(cr, uid, [m], context=context)
1758                     for pick_id in res2.keys():
1759                         result.setdefault(pick_id, [])
1760                         result[pick_id] += res2[pick_id]
1761                 else:
1762                     result.setdefault(m.picking_id, [])
1763                     result[m.picking_id].append( (m, dest) )
1764         return result
1765     def _create_chained_picking(self, cr, uid, pick_name,picking,ptype,move, context=None):
1766         res_obj = self.pool.get('res.company')
1767         picking_obj = self.pool.get('stock.picking')
1768         pick_id= picking_obj.create(cr, uid, {
1769                                 'name': pick_name,
1770                                 'origin': str(picking.origin or ''),
1771                                 'type': ptype,
1772                                 'note': picking.note,
1773                                 'move_type': picking.move_type,
1774                                 'auto_picking': move[0][1][1] == 'auto',
1775                                 'stock_journal_id': move[0][1][3],
1776                                 'company_id': move[0][1][4] or res_obj._company_default_get(cr, uid, 'stock.company', context=context),
1777                                 'address_id': picking.address_id.id,
1778                                 'invoice_state': 'none',
1779                                 'date': picking.date,
1780                             })
1781         return pick_id
1782     def action_confirm(self, cr, uid, ids, context=None):
1783         """ Confirms stock move.
1784         @return: List of ids.
1785         """
1786         moves = self.browse(cr, uid, ids, context=context)
1787         self.write(cr, uid, ids, {'state': 'confirmed'})
1788         res_obj = self.pool.get('res.company')
1789         location_obj = self.pool.get('stock.location')
1790         move_obj = self.pool.get('stock.move')
1791         wf_service = netsvc.LocalService("workflow")
1792
1793         def create_chained_picking(self, cr, uid, moves, context=None):
1794             new_moves = []
1795             if context is None:
1796                 context = {}
1797             for picking, todo in self._chain_compute(cr, uid, moves, context=context).items():
1798                 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])
1799                 pick_name = picking.name or ''
1800                 if picking:
1801                     pickid = self._create_chained_picking(cr, uid, pick_name,picking,ptype,todo,context)
1802                 else:
1803                     pickid = False
1804                 for move, (loc, dummy, delay, dummy, company_id, ptype) in todo:
1805                     new_id = move_obj.copy(cr, uid, move.id, {
1806                         'location_id': move.location_dest_id.id,
1807                         'location_dest_id': loc.id,
1808                         'date_moved': time.strftime('%Y-%m-%d'),
1809                         'picking_id': pickid,
1810                         'state': 'waiting',
1811                         'company_id': company_id or res_obj._company_default_get(cr, uid, 'stock.company', context=context)  ,
1812                         'move_history_ids': [],
1813                         'date': (datetime.strptime(move.date, '%Y-%m-%d %H:%M:%S') + relativedelta(days=delay or 0)).strftime('%Y-%m-%d'),
1814                         'move_history_ids2': []}
1815                     )
1816                     move_obj.write(cr, uid, [move.id], {
1817                         'move_dest_id': new_id,
1818                         'move_history_ids': [(4, new_id)]
1819                     })
1820                     new_moves.append(self.browse(cr, uid, [new_id])[0])
1821                 if pickid:
1822                     wf_service.trg_validate(uid, 'stock.picking', pickid, 'button_confirm', cr)
1823             if new_moves:
1824                 create_chained_picking(self, cr, uid, new_moves, context)
1825         create_chained_picking(self, cr, uid, moves, context)
1826         return []
1827
1828     def action_assign(self, cr, uid, ids, *args):
1829         """ Changes state to confirmed or waiting.
1830         @return: List of values
1831         """
1832         todo = []
1833         for move in self.browse(cr, uid, ids):
1834             if move.state in ('confirmed', 'waiting'):
1835                 todo.append(move.id)
1836         res = self.check_assign(cr, uid, todo)
1837         return res
1838
1839     def force_assign(self, cr, uid, ids, context=None):
1840         """ Changes the state to assigned.
1841         @return: True
1842         """
1843         self.write(cr, uid, ids, {'state': 'assigned'})
1844         return True
1845
1846     def cancel_assign(self, cr, uid, ids, context=None):
1847         """ Changes the state to confirmed.
1848         @return: True
1849         """
1850         self.write(cr, uid, ids, {'state': 'confirmed'})
1851         return True
1852
1853     #
1854     # Duplicate stock.move
1855     #
1856     def check_assign(self, cr, uid, ids, context=None):
1857         """ Checks the product type and accordingly writes the state.
1858         @return: No. of moves done
1859         """
1860         done = []
1861         count = 0
1862         pickings = {}
1863         if context is None:
1864             context = {}
1865         for move in self.browse(cr, uid, ids, context=context):
1866             if move.product_id.type == 'consu' or move.location_id.usage == 'supplier':
1867                 if move.state in ('confirmed', 'waiting'):
1868                     done.append(move.id)
1869                 pickings[move.picking_id.id] = 1
1870                 continue
1871             if move.state in ('confirmed', 'waiting'):
1872                 # Important: we must pass lock=True to _product_reserve() to avoid race conditions and double reservations
1873                 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)
1874                 if res:
1875                     #_product_available_test depends on the next status for correct functioning
1876                     #the test does not work correctly if the same product occurs multiple times
1877                     #in the same order. This is e.g. the case when using the button 'split in two' of
1878                     #the stock outgoing form
1879                     self.write(cr, uid, [move.id], {'state':'assigned'})
1880                     done.append(move.id)
1881                     pickings[move.picking_id.id] = 1
1882                     r = res.pop(0)
1883                     cr.execute('update stock_move set location_id=%s, product_qty=%s where id=%s', (r[1], r[0], move.id))
1884
1885                     while res:
1886                         r = res.pop(0)
1887                         move_id = self.copy(cr, uid, move.id, {'product_qty': r[0], 'location_id': r[1]})
1888                         done.append(move_id)
1889         if done:
1890             count += len(done)
1891             self.write(cr, uid, done, {'state': 'assigned'})
1892
1893         if count:
1894             for pick_id in pickings:
1895                 wf_service = netsvc.LocalService("workflow")
1896                 wf_service.trg_write(uid, 'stock.picking', pick_id, cr)
1897         return count
1898
1899     def setlast_tracking(self, cr, uid, ids, context=None):
1900         tracking_obj = self.pool.get('stock.tracking')
1901         picking = self.browse(cr, uid, ids, context=context)[0].picking_id
1902         if picking:
1903             last_track = [line.tracking_id.id for line in picking.move_lines if line.tracking_id]
1904             if not last_track:
1905                 last_track = tracking_obj.create(cr, uid, {}, context=context)
1906             else:
1907                 last_track.sort()
1908                 last_track = last_track[-1]
1909             self.write(cr, uid, ids, {'tracking_id': last_track})
1910         return True
1911
1912     #
1913     # Cancel move => cancel others move and pickings
1914     #
1915     def action_cancel(self, cr, uid, ids, context=None):
1916         """ Cancels the moves and if all moves are cancelled it cancels the picking.
1917         @return: True
1918         """
1919         if not len(ids):
1920             return True
1921         if context is None:
1922             context = {}
1923         pickings = {}
1924         for move in self.browse(cr, uid, ids, context=context):
1925             if move.state in ('confirmed', 'waiting', 'assigned', 'draft'):
1926                 if move.picking_id:
1927                     pickings[move.picking_id.id] = True
1928             if move.move_dest_id and move.move_dest_id.state == 'waiting':
1929                 self.write(cr, uid, [move.move_dest_id.id], {'state': 'assigned'})
1930                 if context.get('call_unlink',False) and move.move_dest_id.picking_id:
1931                     wf_service = netsvc.LocalService("workflow")
1932                     wf_service.trg_write(uid, 'stock.picking', move.move_dest_id.picking_id.id, cr)
1933         self.write(cr, uid, ids, {'state': 'cancel', 'move_dest_id': False})
1934         if not context.get('call_unlink',False):
1935             for pick in self.pool.get('stock.picking').browse(cr, uid, pickings.keys()):
1936                 if all(move.state == 'cancel' for move in pick.move_lines):
1937                     self.pool.get('stock.picking').write(cr, uid, [pick.id], {'state': 'cancel'})
1938
1939         wf_service = netsvc.LocalService("workflow")
1940         for id in ids:
1941             wf_service.trg_trigger(uid, 'stock.move', id, cr)
1942         return True
1943
1944     def _get_accounting_data_for_valuation(self, cr, uid, move, context=None):
1945         """
1946         Return the accounts and journal to use to post Journal Entries for the real-time
1947         valuation of the move.
1948
1949         :param context: context dictionary that can explicitly mention the company to consider via the 'force_company' key
1950         :raise: osv.except_osv() is any mandatory account or journal is not defined.
1951         """
1952         product_obj=self.pool.get('product.product')
1953         accounts = product_obj.get_product_accounts(cr, uid, move.product_id.id, context)
1954         acc_src = accounts['stock_account_input']
1955         acc_dest = accounts['stock_account_output']
1956         acc_variation = accounts.get('property_stock_variation', False)
1957         journal_id = accounts['stock_journal']
1958
1959         if acc_dest == acc_variation:
1960             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.'))
1961
1962         if acc_src == acc_variation:
1963             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.'))
1964
1965         if not acc_src:
1966             raise osv.except_osv(_('Error!'),  _('There is no stock input account defined for this product or its category: "%s" (id: %d)') % \
1967                                     (move.product_id.name, move.product_id.id,))
1968         if not acc_dest:
1969             raise osv.except_osv(_('Error!'),  _('There is no stock output account defined for this product or its category: "%s" (id: %d)') % \
1970                                     (move.product_id.name, move.product_id.id,))
1971         if not journal_id:
1972             raise osv.except_osv(_('Error!'), _('There is no journal defined on the product category: "%s" (id: %d)') % \
1973                                     (move.product_id.categ_id.name, move.product_id.categ_id.id,))
1974         if not acc_variation:
1975             raise osv.except_osv(_('Error!'), _('There is no inventory variation account defined on the product category: "%s" (id: %d)') % \
1976                                     (move.product_id.categ_id.name, move.product_id.categ_id.id,))
1977
1978         return journal_id, acc_src, acc_dest, acc_variation
1979
1980     def _get_reference_accounting_values_for_valuation(self, cr, uid, move, context=None):
1981         """
1982         Return the reference amount and reference currency representing the inventory valuation for this move.
1983         These reference values should possibly be converted before being posted in Journals to adapt to the primary
1984         and secondary currencies of the relevant accounts.
1985         """
1986         product_uom_obj = self.pool.get('product.uom')
1987
1988         # by default the reference currency is that of the move's company
1989         reference_currency_id = move.company_id.currency_id.id
1990
1991         default_uom = move.product_id.uom_id.id
1992         qty = product_uom_obj._compute_qty(cr, uid, move.product_uom.id, move.product_qty, default_uom)
1993
1994         # if product is set to average price and a specific value was entered in the picking wizard,
1995         # we use it
1996         if move.product_id.cost_method == 'average' and move.price_unit:
1997             reference_amount = qty * move.price_unit
1998             reference_currency_id = move.price_currency_id.id or reference_currency_id
1999
2000         # Otherwise we default to the company's valuation price type, considering that the values of the
2001         # valuation field are expressed in the default currency of the move's company.
2002         else:
2003             if context is None:
2004                 context = {}
2005             currency_ctx = dict(context, currency_id = move.company_id.currency_id.id)
2006             amount_unit = move.product_id.price_get('standard_price', currency_ctx)[move.product_id.id]
2007             reference_amount = amount_unit * qty or 1.0
2008
2009         return reference_amount, reference_currency_id
2010
2011
2012     def _create_product_valuation_moves(self, cr, uid, move, context=None):
2013         """
2014         Generate the appropriate accounting moves if the product being moves is subject
2015         to real_time valuation tracking, and the source or destination location is
2016         a transit location or is outside of the company.
2017         """
2018         if move.product_id.valuation == 'real_time': # FIXME: product valuation should perhaps be a property?
2019             if context is None:
2020                 context = {}
2021             src_company_ctx = dict(context,force_company=move.location_id.company_id.id)
2022             dest_company_ctx = dict(context,force_company=move.location_dest_id.company_id.id)
2023             account_moves = []
2024             # Outgoing moves (or cross-company output part)
2025             if move.location_id.company_id \
2026                 and (move.location_id.usage == 'internal' and move.location_dest_id.usage != 'internal'\
2027                      or move.location_id.company_id != move.location_dest_id.company_id):
2028                 journal_id, acc_src, acc_dest, acc_variation = self._get_accounting_data_for_valuation(cr, uid, move, src_company_ctx)
2029                 reference_amount, reference_currency_id = self._get_reference_accounting_values_for_valuation(cr, uid, move, src_company_ctx)
2030                 account_moves += [(journal_id, self._create_account_move_line(cr, uid, move, acc_variation, acc_dest, reference_amount, reference_currency_id, context))]
2031
2032             # Incoming moves (or cross-company input part)
2033             if move.location_dest_id.company_id \
2034                 and (move.location_id.usage != 'internal' and move.location_dest_id.usage == 'internal'\
2035                      or move.location_id.company_id != move.location_dest_id.company_id):
2036                 journal_id, acc_src, acc_dest, acc_variation = self._get_accounting_data_for_valuation(cr, uid, move, dest_company_ctx)
2037                 reference_amount, reference_currency_id = self._get_reference_accounting_values_for_valuation(cr, uid, move, src_company_ctx)
2038                 account_moves += [(journal_id, self._create_account_move_line(cr, uid, move, acc_src, acc_variation, reference_amount, reference_currency_id, context))]
2039
2040             move_obj = self.pool.get('account.move')
2041             for j_id, move_lines in account_moves:
2042                 move_obj.create(cr, uid,
2043                         {'name': move.name,
2044                          'journal_id': j_id,
2045                          'line_id': move_lines,
2046                          'ref': move.picking_id and move.picking_id.name})
2047
2048
2049     def action_done(self, cr, uid, ids, context=None):
2050         """ Makes the move done and if all moves are done, it will finish the picking.
2051         @return:
2052         """
2053         partial_datas=''
2054         picking_ids = []
2055         move_ids = []
2056         partial_obj=self.pool.get('stock.partial.picking')
2057         wf_service = netsvc.LocalService("workflow")
2058         partial_id=partial_obj.search(cr,uid,[])
2059         if partial_id:
2060             partial_datas = partial_obj.read(cr, uid, partial_id, context=context)[0]
2061         if context is None:
2062             context = {}
2063
2064         todo = []
2065         for move in self.browse(cr, uid, ids, context=context):
2066             if move.state=="draft":
2067                 todo.append(move.id)
2068         if todo:
2069             self.action_confirm(cr, uid, todo, context=context)
2070
2071         for move in self.browse(cr, uid, ids, context=context):
2072             if move.state in ['done','cancel']:
2073                 continue
2074             move_ids.append(move.id)
2075
2076             if move.picking_id:
2077                 picking_ids.append(move.picking_id.id)
2078             if move.move_dest_id.id and (move.state != 'done'):
2079                 self.write(cr, uid, [move.id], {'move_history_ids': [(4, move.move_dest_id.id)]})
2080                 #cr.execute('insert into stock_move_history_ids (parent_id,child_id) values (%s,%s)', (move.id, move.move_dest_id.id))
2081                 if move.move_dest_id.state in ('waiting', 'confirmed'):
2082                     self.write(cr, uid, [move.move_dest_id.id], {'state': 'assigned'})
2083                     if move.move_dest_id.picking_id:
2084                         wf_service.trg_write(uid, 'stock.picking', move.move_dest_id.picking_id.id, cr)
2085                     if move.move_dest_id.auto_validate:
2086                         self.action_done(cr, uid, [move.move_dest_id.id], context=context)
2087
2088             self._create_product_valuation_moves(cr, uid, move, context=context)
2089             prodlot_id = partial_datas and partial_datas.get('move%s_prodlot_id' % (move.id), False)
2090             if prodlot_id:
2091                 self.write(cr, uid, [move.id], {'prodlot_id': prodlot_id}, context=context)
2092
2093         self.write(cr, uid, move_ids, {'state': 'done', 'date_planned': time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
2094         for id in move_ids:
2095              wf_service.trg_trigger(uid, 'stock.move', id, cr)
2096
2097         for pick_id in picking_ids:
2098             wf_service.trg_write(uid, 'stock.picking', pick_id, cr)
2099
2100         return True
2101
2102     def _create_account_move_line(self, cr, uid, move, src_account_id, dest_account_id, reference_amount, reference_currency_id, context=None):
2103         """
2104         Generate the account.move.line values to post to track the stock valuation difference due to the
2105         processing of the given stock move.
2106         """
2107         # prepare default values considering that the destination accounts have the reference_currency_id as their main currency
2108         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
2109         debit_line_vals = {
2110                     'name': move.name,
2111                     'product_id': move.product_id and move.product_id.id or False,
2112                     'quantity': move.product_qty,
2113                     'ref': move.picking_id and move.picking_id.name or False,
2114                     'date': time.strftime('%Y-%m-%d'),
2115                     'partner_id': partner_id,
2116                     'debit': reference_amount,
2117                     'account_id': dest_account_id,
2118         }
2119         credit_line_vals = {
2120                     'name': move.name,
2121                     'product_id': move.product_id and move.product_id.id or False,
2122                     'quantity': move.product_qty,
2123                     'ref': move.picking_id and move.picking_id.name or False,
2124                     'date': time.strftime('%Y-%m-%d'),
2125                     'partner_id': partner_id,
2126                     'credit': reference_amount,
2127                     'account_id': src_account_id,
2128         }
2129
2130         # if we are posting to accounts in a different currency, provide correct values in both currencies correctly
2131         # when compatible with the optional secondary currency on the account.
2132         # Financial Accounts only accept amounts in secondary currencies if there's no secondary currency on the account
2133         # or if it's the same as that of the secondary amount being posted.
2134         account_obj = self.pool.get('account.account')
2135         src_acct, dest_acct = account_obj.browse(cr, uid, [src_account_id, dest_account_id], context=context)
2136         src_main_currency_id = src_acct.company_id.currency_id.id
2137         dest_main_currency_id = dest_acct.company_id.currency_id.id
2138         cur_obj = self.pool.get('res.currency')
2139         if reference_currency_id != src_main_currency_id:
2140             # fix credit line:
2141             credit_line_vals['credit'] = cur_obj.compute(cr, uid, reference_currency_id, src_main_currency_id, reference_amount, context=context)
2142             if (not src_acct.currency_id) or src_acct.currency_id.id == reference_currency_id:
2143                 credit_line_vals.update(currency_id=reference_currency_id, amount_currency=reference_amount)
2144         if reference_currency_id != dest_main_currency_id:
2145             # fix debit line:
2146             debit_line_vals['debit'] = cur_obj.compute(cr, uid, reference_currency_id, dest_main_currency_id, reference_amount, context=context)
2147             if (not dest_acct.currency_id) or dest_acct.currency_id.id == reference_currency_id:
2148                 debit_line_vals.update(currency_id=reference_currency_id, amount_currency=reference_amount)
2149
2150         return [(0, 0, debit_line_vals), (0, 0, credit_line_vals)]
2151
2152     def unlink(self, cr, uid, ids, context=None):
2153         if context is None:
2154             context = {}
2155         ctx = context.copy()
2156         for move in self.browse(cr, uid, ids, context=context):
2157             if move.state != 'draft' and not ctx.get('call_unlink',False):
2158                 raise osv.except_osv(_('UserError'),
2159                         _('You can only delete draft moves.'))
2160         return super(stock_move, self).unlink(
2161             cr, uid, ids, context=ctx)
2162
2163     def _create_lot(self, cr, uid, ids, product_id, prefix=False):
2164         """ Creates production lot
2165         @return: Production lot id
2166         """
2167         prodlot_obj = self.pool.get('stock.production.lot')
2168         prodlot_id = prodlot_obj.create(cr, uid, {'prefix': prefix, 'product_id': product_id})
2169         return prodlot_id
2170
2171     def action_scrap(self, cr, uid, ids, quantity, location_id, context=None):
2172         """ Move the scrap/damaged product into scrap location
2173         @param cr: the database cursor
2174         @param uid: the user id
2175         @param ids: ids of stock move object to be scrapped
2176         @param quantity : specify scrap qty
2177         @param location_id : specify scrap location
2178         @param context: context arguments
2179         @return: Scraped lines
2180         """
2181         if quantity <= 0:
2182             raise osv.except_osv(_('Warning!'), _('Please provide a positive quantity to scrap!'))
2183         res = []
2184         for move in self.browse(cr, uid, ids, context=context):
2185             move_qty = move.product_qty
2186             uos_qty = quantity / move_qty * move.product_uos_qty
2187             default_val = {
2188                 'product_qty': quantity,
2189                 'product_uos_qty': uos_qty,
2190                 'state': move.state,
2191                 'scrapped' : True,
2192                 'location_dest_id': location_id,
2193                 'tracking_id': move.tracking_id.id,
2194                 'prodlot_id': move.prodlot_id.id,
2195             }
2196             new_move = self.copy(cr, uid, move.id, default_val)
2197
2198             res += [new_move]
2199             product_obj = self.pool.get('product.product')
2200             for (id, name) in product_obj.name_get(cr, uid, [move.product_id.id]):
2201                 self.log(cr, uid, move.id, "%s x %s %s" % (move.product_qty, name, _("were scrapped")))
2202
2203         self.action_done(cr, uid, res)
2204         return res
2205
2206     def action_split(self, cr, uid, ids, quantity, split_by_qty=1, prefix=False, with_lot=True, context=None):
2207         """ Split Stock Move lines into production lot which specified split by quantity.
2208         @param cr: the database cursor
2209         @param uid: the user id
2210         @param ids: ids of stock move object to be splited
2211         @param split_by_qty : specify split by qty
2212         @param prefix : specify prefix of production lot
2213         @param with_lot : if true, prodcution lot will assign for split line otherwise not.
2214         @param context: context arguments
2215         @return: Splited move lines
2216         """
2217
2218         if context is None:
2219             context = {}
2220         if quantity <= 0:
2221             raise osv.except_osv(_('Warning!'), _('Please provide Proper Quantity !'))
2222
2223         res = []
2224
2225         for move in self.browse(cr, uid, ids, context=context):
2226             if split_by_qty <= 0 or quantity == 0:
2227                 return res
2228
2229             uos_qty = split_by_qty / move.product_qty * move.product_uos_qty
2230
2231             quantity_rest = quantity % split_by_qty
2232             uos_qty_rest = split_by_qty / move.product_qty * move.product_uos_qty
2233
2234             update_val = {
2235                 'product_qty': split_by_qty,
2236                 'product_uos_qty': uos_qty,
2237             }
2238             for idx in range(int(quantity//split_by_qty)):
2239                 if not idx and move.product_qty<=quantity:
2240                     current_move = move.id
2241                 else:
2242                     current_move = self.copy(cr, uid, move.id, {'state': move.state})
2243                 res.append(current_move)
2244                 if with_lot:
2245                     update_val['prodlot_id'] = self._create_lot(cr, uid, [current_move], move.product_id.id)
2246
2247                 self.write(cr, uid, [current_move], update_val)
2248
2249
2250             if quantity_rest > 0:
2251                 idx = int(quantity//split_by_qty)
2252                 update_val['product_qty'] = quantity_rest
2253                 update_val['product_uos_qty'] = uos_qty_rest
2254                 if not idx and move.product_qty<=quantity:
2255                     current_move = move.id
2256                 else:
2257                     current_move = self.copy(cr, uid, move.id, {'state': move.state})
2258
2259                 res.append(current_move)
2260
2261
2262                 if with_lot:
2263                     update_val['prodlot_id'] = self._create_lot(cr, uid, [current_move], move.product_id.id)
2264
2265                 self.write(cr, uid, [current_move], update_val)
2266         return res
2267
2268     def action_consume(self, cr, uid, ids, quantity, location_id=False, context=None):
2269         """ Consumed product with specific quatity from specific source location
2270         @param cr: the database cursor
2271         @param uid: the user id
2272         @param ids: ids of stock move object to be consumed
2273         @param quantity : specify consume quantity
2274         @param location_id : specify source location
2275         @param context: context arguments
2276         @return: Consumed lines
2277         """
2278         if context is None:
2279             context = {}
2280         if quantity <= 0:
2281             raise osv.except_osv(_('Warning!'), _('Please provide Proper Quantity !'))
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             quantity_rest -= quantity
2289             uos_qty_rest = quantity_rest / move_qty * move.product_uos_qty
2290             if quantity_rest <= 0:
2291                 quantity_rest = 0
2292                 uos_qty_rest = 0
2293                 quantity = move.product_qty
2294
2295             uos_qty = quantity / move_qty * move.product_uos_qty
2296
2297             if quantity_rest > 0:
2298                 default_val = {
2299                     'product_qty': quantity,
2300                     'product_uos_qty': uos_qty,
2301                     'state': move.state,
2302                     'location_id': location_id or move.location_id.id,
2303                 }
2304                 if (not move.prodlot_id.id) and (move.product_id.track_production and location_id):
2305                     # IF product has checked track for production lot, move lines will be split by 1
2306                     res += self.action_split(cr, uid, [move.id], quantity, split_by_qty=1, context=context)
2307                 else:
2308                     current_move = self.copy(cr, uid, move.id, default_val)
2309                     res += [current_move]
2310                 update_val = {}
2311                 update_val['product_qty'] = quantity_rest
2312                 update_val['product_uos_qty'] = uos_qty_rest
2313                 self.write(cr, uid, [move.id], update_val)
2314
2315             else:
2316                 quantity_rest = quantity
2317                 uos_qty_rest =  uos_qty
2318                 if (not move.prodlot_id.id) and (move.product_id.track_production and location_id):
2319                     res += self.action_split(cr, uid, [move.id], quantity_rest, split_by_qty=1, context=context)
2320                 else:
2321                     res += [move.id]
2322                     update_val = {
2323                         'product_qty' : quantity_rest,
2324                         'product_uos_qty' : uos_qty_rest,
2325                         'location_id': location_id or move.location_id.id
2326                     }
2327                     self.write(cr, uid, [move.id], update_val)
2328
2329             product_obj = self.pool.get('product.product')
2330             for new_move in self.browse(cr, uid, res, context=context):
2331                 for (id, name) in product_obj.name_get(cr, uid, [new_move.product_id.id]):
2332                     message = _('Product ') + " '" + name + "' "+ _("is consumed with") + " '" + str(new_move.product_qty) + "' "+ _("quantity.")
2333                     self.log(cr, uid, new_move.id, message)
2334         self.action_done(cr, uid, res)
2335
2336         return res
2337
2338     # FIXME: needs refactoring, this code is partially duplicated in stock_picking.do_partial()!
2339     def do_partial(self, cr, uid, ids, partial_datas, context=None):
2340         """ Makes partial pickings and moves done.
2341         @param partial_datas: Dictionary containing details of partial picking
2342                           like partner_id, address_id, delivery_date, delivery
2343                           moves with product_id, product_qty, uom
2344         """
2345         res = {}
2346         picking_obj = self.pool.get('stock.picking')
2347         product_obj = self.pool.get('product.product')
2348         currency_obj = self.pool.get('res.currency')
2349         uom_obj = self.pool.get('product.uom')
2350         wf_service = netsvc.LocalService("workflow")
2351
2352         if context is None:
2353             context = {}
2354
2355         complete, too_many, too_few = [], [], []
2356         move_product_qty = {}
2357         prodlot_ids = {}
2358         for move in self.browse(cr, uid, ids, context=context):
2359             if move.state in ('done', 'cancel'):
2360                 continue
2361             partial_data = partial_datas.get('move%s'%(move.id), False)
2362             assert partial_data, _('Missing partial picking data for move #%s') % (move.id)
2363             product_qty = partial_data.get('product_qty',0.0)
2364             move_product_qty[move.id] = product_qty
2365             product_uom = partial_data.get('product_uom',False)
2366             product_price = partial_data.get('product_price',0.0)
2367             product_currency = partial_data.get('product_currency',False)
2368             prodlot_ids[move.id] = partial_data.get('prodlot_id')
2369             if move.product_qty == product_qty:
2370                 complete.append(move)
2371             elif move.product_qty > product_qty:
2372                 too_few.append(move)
2373             else:
2374                 too_many.append(move)
2375
2376             # Average price computation
2377             if (move.picking_id.type == 'in') and (move.product_id.cost_method == 'average'):
2378                 product = product_obj.browse(cr, uid, move.product_id.id)
2379                 move_currency_id = move.company_id.currency_id.id
2380                 context['currency_id'] = move_currency_id
2381                 qty = uom_obj._compute_qty(cr, uid, product_uom, product_qty, product.uom_id.id)
2382                 if qty > 0:
2383                     new_price = currency_obj.compute(cr, uid, product_currency,
2384                             move_currency_id, product_price)
2385                     new_price = uom_obj._compute_price(cr, uid, product_uom, new_price,
2386                             product.uom_id.id)
2387                     if product.qty_available <= 0:
2388                         new_std_price = new_price
2389                     else:
2390                         # Get the standard price
2391                         amount_unit = product.price_get('standard_price', context)[product.id]
2392                         new_std_price = ((amount_unit * product.qty_available)\
2393                             + (new_price * qty))/(product.qty_available + qty)
2394
2395                     product_obj.write(cr, uid, [product.id],{'standard_price': new_std_price})
2396
2397                     # Record the values that were chosen in the wizard, so they can be
2398                     # used for inventory valuation if real-time valuation is enabled.
2399                     self.write(cr, uid, [move.id],
2400                                 {'price_unit': product_price,
2401                                  'price_currency_id': product_currency,
2402                                 })
2403
2404         for move in too_few:
2405             product_qty = move_product_qty[move.id]
2406             if product_qty != 0:
2407                 defaults = {
2408                             'product_qty' : product_qty,
2409                             'product_uos_qty': product_qty,
2410                             'picking_id' : move.picking_id.id,
2411                             'state': 'assigned',
2412                             'move_dest_id': False,
2413                             'price_unit': move.price_unit,
2414                             }
2415                 prodlot_id = prodlot_ids[move.id]
2416                 if prodlot_id:
2417                     defaults.update(prodlot_id=prodlot_id)
2418                 new_move = self.copy(cr, uid, move.id, defaults)
2419                 complete.append(self.browse(cr, uid, new_move))
2420             self.write(cr, uid, move.id,
2421                     {
2422                         'product_qty' : move.product_qty - product_qty,
2423                         'product_uos_qty':move.product_qty - product_qty,
2424                     })
2425
2426
2427         for move in too_many:
2428             self.write(cr, uid, move.id,
2429                     {
2430                         'product_qty': move.product_qty,
2431                         'product_uos_qty': move.product_qty,
2432                     })
2433             complete.append(move)
2434
2435         for move in complete:
2436             if prodlot_ids.get(move.id):
2437                 self.write(cr, uid, [move.id],{'prodlot_id': prodlot_ids.get(move.id)})
2438             self.action_done(cr, uid, [move.id], context=context)
2439             if  move.picking_id.id :
2440                 # TOCHECK : Done picking if all moves are done
2441                 cr.execute("""
2442                     SELECT move.id FROM stock_picking pick
2443                     RIGHT JOIN stock_move move ON move.picking_id = pick.id AND move.state = %s
2444                     WHERE pick.id = %s""",
2445                             ('done', move.picking_id.id))
2446                 res = cr.fetchall()
2447                 if len(res) == len(move.picking_id.move_lines):
2448                     picking_obj.action_move(cr, uid, [move.picking_id.id])
2449                     wf_service.trg_validate(uid, 'stock.picking', move.picking_id.id, 'button_done', cr)
2450
2451         return [move.id for move in complete]
2452
2453 stock_move()
2454
2455 class stock_inventory(osv.osv):
2456     _name = "stock.inventory"
2457     _description = "Inventory"
2458     _columns = {
2459         'name': fields.char('Inventory Reference', size=64, required=True, readonly=True, states={'draft': [('readonly', False)]}),
2460         'date': fields.datetime('Creation Date', required=True, readonly=True, states={'draft': [('readonly', False)]}),
2461         'date_done': fields.datetime('Date done'),
2462         'inventory_line_id': fields.one2many('stock.inventory.line', 'inventory_id', 'Inventories', states={'done': [('readonly', True)]}),
2463         'move_ids': fields.many2many('stock.move', 'stock_inventory_move_rel', 'inventory_id', 'move_id', 'Created Moves'),
2464         'state': fields.selection( (('draft', 'Draft'), ('done', 'Done'), ('confirm','Confirmed'),('cancel','Cancelled')), 'State', readonly=True, select=True),
2465         'company_id': fields.many2one('res.company', 'Company', required=True, select=True, readonly=True, states={'draft':[('readonly',False)]}),
2466
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         if context is None:
2487             context = {}
2488         move_obj = self.pool.get('stock.move')
2489         for inv in self.browse(cr, uid, ids, context=context):
2490             move_obj.action_done(cr, uid, [x.id for x in inv.move_ids], context=context)
2491             self.write(cr, uid, [inv.id], {'state':'done', 'date_done': time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
2492         return True
2493
2494     def action_confirm(self, cr, uid, ids, context=None):
2495         """ Confirm the inventory and writes its finished date
2496         @return: True
2497         """
2498         if context is None:
2499             context = {}
2500         # to perform the correct inventory corrections we need analyze stock location by
2501         # location, never recursively, so we use a special context
2502         product_context = dict(context, compute_child=False)
2503
2504         location_obj = self.pool.get('stock.location')
2505         for inv in self.browse(cr, uid, ids, context=context):
2506             move_ids = []
2507             for line in inv.inventory_line_id:
2508                 pid = line.product_id.id
2509                 product_context.update(uom=line.product_uom.id,date=inv.date)
2510                 amount = location_obj._product_get(cr, uid, line.location_id.id, [pid], product_context)[pid]
2511
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                     }
2523                     if change > 0:
2524                         value.update( {
2525                             'product_qty': change,
2526                             'location_id': location_id,
2527                             'location_dest_id': line.location_id.id,
2528                         })
2529                     else:
2530                         value.update( {
2531                             'product_qty': -change,
2532                             'location_id': line.location_id.id,
2533                             'location_dest_id': location_id,
2534                         })
2535                     if lot_id:
2536                         value.update({
2537                             'prodlot_id': lot_id,
2538                             'product_qty': line.product_qty
2539                         })
2540                     move_ids.append(self._inventory_line_hook(cr, uid, line, value))
2541             message = _('Inventory') + " '" + inv.name + "' "+ _("is done.")
2542             self.log(cr, uid, inv.id, message)
2543             self.write(cr, uid, [inv.id], {'state': 'confirm', 'move_ids': [(6, 0, move_ids)]})
2544         return True
2545
2546     def action_cancel(self, cr, uid, ids, context=None):
2547         """ Cancels the stock move and change inventory state to draft.
2548         @return: True
2549         """
2550         for inv in self.browse(cr, uid, ids, context=context):
2551             self.pool.get('stock.move').action_cancel(cr, uid, [x.id for x in inv.move_ids], context)
2552             self.write(cr, uid, [inv.id], {'state': 'draft'})
2553         return True
2554
2555     def action_cancel_inventary(self, cr, uid, ids, context=None):
2556         """ Cancels both stock move and inventory
2557         @return: True
2558         """
2559         for inv in self.browse(cr, uid, ids, context=context):
2560             self.pool.get('stock.move').action_cancel(cr, uid, [x.id for x in inv.move_ids], context)
2561             self.write(cr, uid, [inv.id], {'state':'cancel'})
2562         return True
2563
2564 stock_inventory()
2565
2566 class stock_inventory_line(osv.osv):
2567     _name = "stock.inventory.line"
2568     _description = "Inventory Line"
2569     _columns = {
2570         'inventory_id': fields.many2one('stock.inventory', 'Inventory', ondelete='cascade', select=True),
2571         'location_id': fields.many2one('stock.location', 'Location', required=True),
2572         'product_id': fields.many2one('product.product', 'Product', required=True, select=True),
2573         'product_uom': fields.many2one('product.uom', 'Product UOM', required=True),
2574         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product UoM')),
2575         'company_id': fields.related('inventory_id','company_id',type='many2one',relation='res.company',string='Company',store=True, select=True),
2576         'prod_lot_id': fields.many2one('stock.production.lot', 'Production Lot', domain="[('product_id','=',product_id)]"),
2577         'state': fields.related('inventory_id','state',type='char',string='State',readonly=True),
2578     }
2579
2580     def on_change_product_id(self, cr, uid, ids, location_id, product, uom=False, to_date=False):
2581         """ Changes UoM and name if product_id changes.
2582         @param location_id: Location id
2583         @param product: Changed product_id
2584         @param uom: UoM product
2585         @return:  Dictionary of changed values
2586         """
2587         if not product:
2588             return {}
2589         if not uom:
2590             prod = self.pool.get('product.product').browse(cr, uid, [product], {'uom': uom})[0]
2591             uom = prod.uom_id.id
2592         amount = self.pool.get('stock.location')._product_get(cr, uid, location_id, [product], {'uom': uom, 'to_date': to_date})[product]
2593         result = {'product_qty': amount, 'product_uom': uom}
2594         return {'value': result}
2595
2596 stock_inventory_line()
2597
2598 #----------------------------------------------------------
2599 # Stock Warehouse
2600 #----------------------------------------------------------
2601 class stock_warehouse(osv.osv):
2602     _name = "stock.warehouse"
2603     _description = "Warehouse"
2604     _columns = {
2605         'name': fields.char('Name', size=128, required=True, select=True),
2606         'company_id': fields.many2one('res.company', 'Company', required=True, select=True),
2607         'partner_address_id': fields.many2one('res.partner.address', 'Owner Address'),
2608         'lot_input_id': fields.many2one('stock.location', 'Location Input', required=True, domain=[('usage','<>','view')]),
2609         'lot_stock_id': fields.many2one('stock.location', 'Location Stock', required=True, domain=[('usage','<>','view')]),
2610         'lot_output_id': fields.many2one('stock.location', 'Location Output', required=True, domain=[('usage','<>','view')]),
2611     }
2612     _defaults = {
2613         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'stock.inventory', context=c),
2614     }
2615
2616 stock_warehouse()
2617
2618 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: