371 lines
16 KiB
Python
371 lines
16 KiB
Python
#This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
#this repository contains the full copyright notices and license terms.
|
|
"Move"
|
|
from trytond.osv import fields, OSV
|
|
from decimal import Decimal
|
|
import datetime
|
|
|
|
STATES = {
|
|
'readonly': "(state in ('cancel', 'done'))",
|
|
}
|
|
|
|
|
|
class Move(OSV):
|
|
"Stock Move"
|
|
_name = 'stock.move'
|
|
_description = __doc__
|
|
_rec_name = "product"
|
|
product = fields.Many2One("product.product", "Product", required=True,
|
|
select=1, states=STATES,
|
|
on_change=['product', 'type', 'currency', 'uom', 'company'],
|
|
domain=[('type', '!=', 'service')])
|
|
uom = fields.Many2One("product.uom", "Uom", required=True, states=STATES,
|
|
domain="[('category', '=', " \
|
|
"(product, 'product.default_uom.category'))]")
|
|
unit_digits = fields.Function('get_unit_digits', type='integer',
|
|
string='Unit Digits', on_change_with=['uom'])
|
|
quantity = fields.Float("Quantity", required=True,
|
|
digits="(16, unit_digits)", states=STATES)
|
|
from_location = fields.Many2One("stock.location", "From Location", select=1,
|
|
required=True, states=STATES,
|
|
domain=[('type', '!=', 'warehouse')],
|
|
on_change=['from_location', 'to_location'])
|
|
to_location = fields.Many2One("stock.location", "To Location", select=1,
|
|
required=True, states=STATES,
|
|
domain=[('type', '!=', 'warehouse')],
|
|
on_change=['from_location', 'to_location'])
|
|
packing_in = fields.Many2One('stock.packing.in', 'Supplier Packing',
|
|
readonly=True, select=1)
|
|
packing_out = fields.Many2One('stock.packing.out', 'Customer Packing',
|
|
readonly=True, select=1)
|
|
packing_internal = fields.Many2One('stock.packing.internal',
|
|
'Internal Packing', readonly=True, select=1)
|
|
planned_date = fields.Date("Planned Date", states=STATES)
|
|
effective_date = fields.Date("Effective Date", readonly=True)
|
|
state = fields.Selection([
|
|
('draft', 'Draft'),
|
|
('assigned', 'Assigned'),
|
|
('done', 'Done'),
|
|
('cancel', 'Cancel'),
|
|
], 'State', select=1, readonly=True)
|
|
company = fields.Many2One('company.company', 'Company', required=True,
|
|
states={
|
|
'readonly': "state != 'draft'",
|
|
})
|
|
unit_price = fields.Numeric('Unit Price', digits=(16, 4),
|
|
states={
|
|
'invisible': "type not in ('input', 'output')",
|
|
'required': "type in ('input', 'output')",
|
|
'readonly': "state not in ('draft',)",
|
|
})
|
|
currency = fields.Many2One('currency.currency', 'Currency',
|
|
states={
|
|
'invisible': "type not in ('input', 'output')",
|
|
'required': "type in ('input', 'output')",
|
|
'readonly': "state not in ('draft',)",
|
|
})
|
|
type = fields.Function('get_type', type='selection',
|
|
selection=[
|
|
('input', 'Input'),
|
|
('output', 'Output'),
|
|
('internal', 'Internal'),
|
|
], string='Type')
|
|
|
|
def __init__(self):
|
|
super(Move, self).__init__()
|
|
self._sql_constraints += [
|
|
('check_move_qty_pos',
|
|
'CHECK(quantity >= 0.0)', 'Move quantity must be positive'),
|
|
('check_from_to_locations',
|
|
'CHECK(from_location != to_location)',
|
|
'Source and destination location must be different'),
|
|
('check_packing_in_out',
|
|
'CHECK(NOT(packing_in IS NOT NULL ' \
|
|
'AND packing_out IS NOT NULL))',
|
|
'Move can not be in both Supplier and Customer Packing'),
|
|
]
|
|
self._constraints += [
|
|
('check_product_type', 'service_product'),
|
|
]
|
|
self._order[0] = ('id', 'DESC')
|
|
self._error_messages.update({
|
|
'set_state_draft': 'You can not set state to draft!',
|
|
'set_state_assigned': 'You can not set state to assigned!',
|
|
'set_state_done': 'You can not set state to done!',
|
|
'del_draft_cancel': 'You can only delete draft or cancelled moves!',
|
|
'service_product': 'You can not use service product for a move!',
|
|
})
|
|
|
|
def default_planned_date(self, cursor, user, context=None):
|
|
if context and context.get('planned_date'):
|
|
return context.get('planned_date')
|
|
|
|
def default_to_location(self, cursor, user, context=None):
|
|
if context and context.get('to_location'):
|
|
return context.get('to_location')
|
|
if context and context.get('warehouse') and context.get('type'):
|
|
wh_location = self.pool.get('stock.location').browse(
|
|
cursor, user, context['warehouse'], context=context)
|
|
field = {'inventory_in': 'storage_location',
|
|
'inventory_out': 'output_location',
|
|
'incoming': 'input_location',}.get(context['type'])
|
|
return field and wh_location[field].id or False
|
|
|
|
def default_from_location(self, cursor, user, context=None):
|
|
if context and context.get('from_location'):
|
|
return context.get('from_location')
|
|
if context and context.get('warehouse') and context.get('type'):
|
|
wh_location = self.pool.get('stock.location').browse(
|
|
cursor, user, context['warehouse'], context=context)
|
|
field = {'inventory_in': 'input_location',
|
|
'inventory_out': 'storage_location',
|
|
'outgoing': 'output_location',}.get(context['type'])
|
|
return field and wh_location[field].id or False
|
|
|
|
def default_state(self, cursor, user, context=None):
|
|
return 'draft'
|
|
|
|
def default_company(self, cursor, user, context=None):
|
|
company_obj = self.pool.get('company.company')
|
|
if context is None:
|
|
context = {}
|
|
if context.get('company'):
|
|
return company_obj.name_get(cursor, user, context['company'],
|
|
context=context)[0]
|
|
return False
|
|
|
|
def default_currency(self, cursor, user, context=None):
|
|
company_obj = self.pool.get('company.company')
|
|
currency_obj = self.pool.get('currency.currency')
|
|
if context is None:
|
|
context = {}
|
|
company = None
|
|
if context.get('company'):
|
|
company = company_obj.browse(cursor, user, context['company'],
|
|
context=context)
|
|
return currency_obj.name_get(cursor, user, company.currency.id,
|
|
context=context)[0]
|
|
return False
|
|
|
|
def get_type(self, cursor, user, ids, name, args, context=None):
|
|
res = {}
|
|
for move in self.browse(cursor, user, ids, context=context):
|
|
res[move.id] = 'internal'
|
|
if move.from_location.type == 'supplier':
|
|
res[move.id] = 'input'
|
|
if move.to_location.type == 'customer':
|
|
res[move.id] = 'output'
|
|
return res
|
|
|
|
def on_change_with_unit_digits(self, cursor, user, ids, vals,
|
|
context=None):
|
|
uom_obj = self.pool.get('product.uom')
|
|
if vals.get('unit'):
|
|
uom = uom_obj.browse(cursor, user, vals['unit'],
|
|
context=context)
|
|
return uom.digits
|
|
return 2
|
|
|
|
def get_unit_digits(self, cursor, user, ids, name, arg, context=None):
|
|
res = {}
|
|
for move in self.browse(cursor, user, ids, context=context):
|
|
res[move.id] = move.uom.digits
|
|
return res
|
|
|
|
def on_change_product(self, cursor, user, ids, vals, context=None):
|
|
product_obj = self.pool.get('product.product')
|
|
uom_obj = self.pool.get('product.uom')
|
|
currency_obj = self.pool.get('currency.currency')
|
|
company_obj = self.pool.get('company.company')
|
|
if context is None:
|
|
context = {}
|
|
res = {'unit_price': Decimal('0.0')}
|
|
if vals.get('product'):
|
|
product = product_obj.browse(cursor, user, vals['product'],
|
|
context=context)
|
|
res['uom'] = uom_obj.name_get(cursor, user,
|
|
product.default_uom.id, context=context)[0]
|
|
res['unit_digits'] = product.default_uom.digits
|
|
if vals.get('type', 'internal') == 'input':
|
|
unit_price = product.cost_price
|
|
if vals.get('uom') and vals['uom'] != product.default_uom.id:
|
|
uom = uom_obj.browse(cursor, user, vals['uom'],
|
|
context=context)
|
|
unit_price = uom_obj.compute_price(cursor, user,
|
|
product.default_uom, unit_price, uom,
|
|
context=context)
|
|
if vals.get('currency') and vals.get('company'):
|
|
currency = currency_obj.browse(cursor, user,
|
|
vals['currency'], context=context)
|
|
company = company_obj.browse(cursor, user,
|
|
vals['company'], context=context)
|
|
unit_price = currency_obj.compute(cursor, user,
|
|
company.currency, unit_price, currency,
|
|
context=context)
|
|
res['unit_price'] = unit_price
|
|
return res
|
|
|
|
def _on_change_location(self, cursor, user, ids, vals, context=None):
|
|
location_obj = self.pool.get('stock.location')
|
|
res = {'type': 'internal'}
|
|
if vals.get('from_location'):
|
|
location = location_obj.browse(cursor, user, vals['from_location'],
|
|
context=context)
|
|
if location.type == 'supplier':
|
|
res['type'] = 'input'
|
|
if vals.get('to_location'):
|
|
location = location_obj.browse(cursor, user, vals['to_location'],
|
|
context=context)
|
|
if location.type == 'customer':
|
|
res['type'] = 'output'
|
|
return res
|
|
|
|
def check_product_type(self, cursor, user, ids):
|
|
for move in self.browse(cursor, user, ids):
|
|
if move.product.type == 'service':
|
|
return False
|
|
return True
|
|
|
|
def on_change_from_location(self, cursor, user, ids, vals, context=None):
|
|
return self._on_change_location(cursor, user, ids, vals,
|
|
context=context)
|
|
|
|
def on_change_to_location(self, cursor, user, ids, vals, context=None):
|
|
return self._on_change_location(cursor, user, ids, vals,
|
|
context=context)
|
|
|
|
def search(self, cursor, user, args, offset=0, limit=None, order=None,
|
|
context=None, count=False, query_string=False):
|
|
location_obj = self.pool.get('stock.location')
|
|
|
|
args = args[:]
|
|
def process_args(args):
|
|
i = 0
|
|
while i < len(args):
|
|
if isinstance(args[i], list):
|
|
process_args(args[i])
|
|
if isinstance(args[i], tuple) \
|
|
and args[i][0] == 'to_location_warehouse':
|
|
location_id = False
|
|
if args[i][2]:
|
|
location = location_obj.browse(cursor, user,
|
|
args[i][2], context=context)
|
|
if location.type == 'warehouse':
|
|
location_id = location.input_location.id
|
|
args[i] = ('to_location', args[i][1], location_id)
|
|
i += 1
|
|
process_args(args)
|
|
return super(Move, self).search(cursor, user, args, offset=offset,
|
|
limit=limit, order=order, context=context, count=count,
|
|
query_string=query_string)
|
|
|
|
def _update_product_cost_price(self, cursor, user, product_id, quantity, uom,
|
|
unit_price, currency, company, context=None):
|
|
"""
|
|
Update the cost price on the given product
|
|
"""
|
|
uom_obj = self.pool.get('product.uom')
|
|
product_obj = self.pool.get('product.product')
|
|
location_obj = self.pool.get('stock.location')
|
|
currency_obj = self.pool.get('currency.currency')
|
|
company_obj = self.pool.get('company.company')
|
|
|
|
product = product_obj.browse(cursor, user, product_id, context=context)
|
|
if not product.cost_price_method == 'average':
|
|
return
|
|
|
|
sign = quantity > 0 and 1 or -1
|
|
quantity = abs(quantity)
|
|
|
|
if isinstance(uom, (int, long)):
|
|
uom = uom_obj.browse(cursor, user, uom, context=context)
|
|
if isinstance(company, (int, long)):
|
|
company = company_obj.browse(cursor, user, company, context=context)
|
|
|
|
ctx = context and context.copy() or {}
|
|
ctx['locations'] = location_obj.search(
|
|
cursor, user, [('type', '=', 'storage')], context=context)
|
|
ctx['stock_date_end'] = datetime.datetime.now()
|
|
product = product_obj.browse(cursor, user, product_id, context=ctx)
|
|
qty = uom_obj.compute_qty(
|
|
cursor, user, uom, quantity, product.default_uom, context=context)
|
|
|
|
qty = Decimal(str(qty))
|
|
product_qty = Decimal(str(product.quantity))
|
|
# convert wrt currency
|
|
unit_price = currency_obj.compute(
|
|
cursor, user, currency, unit_price, company.currency,
|
|
context=context)
|
|
# convert wrt to the uom
|
|
unit_price = uom_obj.compute_price(
|
|
cursor, user, uom, unit_price, product.default_uom, context=context)
|
|
new_cost_price = (
|
|
(product.cost_price * product_qty) + (unit_price * qty * sign)
|
|
) / (product_qty + qty * sign)
|
|
product_obj.write(
|
|
cursor, user, product.id, {'cost_price': new_cost_price},
|
|
context=context)
|
|
|
|
def create(self, cursor, user, vals, context=None):
|
|
location_obj = self.pool.get('stock.location')
|
|
|
|
if vals.get('state') == 'done':
|
|
if not vals.get('effective_date'):
|
|
vals['effective_date'] = datetime.datetime.now()
|
|
from_location = location_obj.browse(
|
|
cursor, user, vals['from_location'], context=context)
|
|
if from_location.type == 'supplier':
|
|
self._update_product_cost_price(
|
|
cursor, user, vals['product'], vals['quantity'],
|
|
vals['uom'], vals['unit_price'], vals['currency'],
|
|
vals['company'], context=context)
|
|
return super(Move, self).create(cursor, user, vals, context=context)
|
|
|
|
def write(self, cursor, user, ids, vals, context=None):
|
|
if context is None:
|
|
context = {}
|
|
|
|
if isinstance(ids, (int, long)):
|
|
ids = [ids]
|
|
|
|
if 'state' in vals:
|
|
for move in self.browse(cursor, user, ids, context=context):
|
|
if vals['state'] == 'cancel':
|
|
vals['effective_date'] = False
|
|
if move.type == 'input' and move.state != 'cancel':
|
|
self._update_product_cost_price(
|
|
cursor, user, move.product.id, -move.quantity,
|
|
move.uom, move.unit_price, move.currency,
|
|
move.company, context=context)
|
|
|
|
elif vals['state'] == 'draft':
|
|
if move.state in ('assigned', 'done'):
|
|
self.raise_user_error(cursor, 'set_state_draft',
|
|
context=context)
|
|
elif vals['state'] == 'assigned':
|
|
if move.state in ('cancel', 'done'):
|
|
self.raise_user_error(cursor, 'set_state_assigned',
|
|
context=context)
|
|
elif vals['state'] == 'done':
|
|
if move.state in ('cancel'):
|
|
self.raise_user_error(cursor, 'set_state_done',
|
|
context=context)
|
|
vals['effective_date'] = datetime.datetime.now()
|
|
|
|
if move.type == 'input' and move.state != 'done':
|
|
self._update_product_cost_price(
|
|
cursor, user, move.product.id, move.quantity,
|
|
move.uom, move.unit_price, move.currency,
|
|
move.company, context=context)
|
|
|
|
return super(Move, self).write(cursor, user, ids, vals, context=context)
|
|
|
|
def delete(self, cursor, user, ids, context=None):
|
|
for move in self.browse(cursor, user, ids, context=context):
|
|
if move.state not in ('draft', 'cancel'):
|
|
self.raise_user_error(cursor, 'del_draft_cancel',
|
|
context=context)
|
|
return super(Move, self).delete(cursor, user, ids, context=context)
|
|
|
|
Move()
|