good/warehouse/models/location.py

from odoo import models, fields, api
from odoo.exceptions import UserError


class Location(models.Model):
    _name = 'location'
    _description = '库位'
    _order = 'name'

    def _get_current_qty(self):
        # 获取当前 库位 的商品数量
        for location in self:
            domain = [
                ('warehouse_dest_id', '=', location.warehouse_id.id),
                ('location_id', '=', location.id),
                ('state', '=', 'done')
            ]
            if location.attribute_id:
                domain.append(('attribute_id', '=', location.attribute_id.id))
            if location.goods_id:
                lines = self.env['wh.move.line'].search(domain)
                current_qty = sum(line.qty_remaining for line in lines)
                if not current_qty:
                    location.write({'goods_id': False, 'attribute_id': False})
            else:
                current_qty = 0
            location.current_qty = current_qty
            location.write({'save_qty': current_qty})

    name = fields.Char('库位号',
                       required=True)
    warehouse_id = fields.Many2one('warehouse',
                                   string='仓库',
                                   required=True)
    goods_id = fields.Many2one('goods',
                               '商品')
    attribute_id = fields.Many2one('attribute', '属性', ondelete='restrict',
                                   help='商品的属性')
    current_qty = fields.Float('数量', compute='_get_current_qty')
    save_qty = fields.Float('在手数量')

    _sql_constraints = [
        ('wh_loc_uniq', 'unique(warehouse_id, name)', '同仓库库位不能重名')
    ]

    def change_location(self):
        for location in self:
            view = self.env.ref('warehouse.change_location_form')
            return {
                'name': '库位',
                'view_mode': 'form',
                'res_model': 'change.location',
                'view_id': False,
                'views': [(view.id, 'form')],
                'type': 'ir.actions.act_window',
                'target': 'new',
                'context': {'default_from_location': location.id},
            }


class ChangeLocation(models.TransientModel):
    _name = 'change.location'
    _description = '库位转移'

    from_location = fields.Many2one('location', string='源库位', required=True)
    to_location = fields.Many2one('location', string='目的库位', required=True)
    change_qty = fields.Float('转出数量')

    def confirm_change(self):
        for change in self:
            if change.change_qty <= 0.0:
                raise UserError('转出数量必须大于零')
            if change.from_location.id == change.to_location.id:
                raise UserError('转出库位 %s 与转入库位不能相同' %
                                change.from_location.name)
            if change.from_location.current_qty < change.change_qty:
                raise UserError('转出数量不能大于库位现有数量，库位 %s 现有数量  %s'
                                % (change.from_location.name, change.from_location.current_qty))
            # 转出库位与转入库位的 产品、产品属性要相同
            if (change.from_location.goods_id.id != change.to_location.goods_id.id and change.to_location.goods_id.id) or \
                    (change.from_location.attribute_id.id != change.to_location.attribute_id.id and change.to_location.attribute_id.id):
                raise UserError('请检查转出库位与转入库位的产品、产品属性是否都相同！')

            # 创建 内部移库单
            wh_internal = self.env['wh.internal'].with_context({'location': change.from_location.id}).create({
                'warehouse_id': change.from_location.warehouse_id.id,
                'warehouse_dest_id': change.to_location.warehouse_id.id,
            })
            self.env['wh.move.line'].with_context({'type': 'internal'}).create({
                'move_id': wh_internal.move_id.id,
                'goods_id': change.from_location.goods_id.id,
                'attribute_id': change.from_location.attribute_id.id,
                'goods_qty': change.change_qty,
                'location_id': change.to_location.id,
            })
            # 自动审核 内部移库单
            wh_internal.approve_order()

            # 返回 更新产品数量后的 库位列表
            return {
                'type': 'ir.actions.client',
                'tag': 'reload',
            }