initial commit 2

1 files changed, 214 insertions, 0 deletions

diff --git a/addons/stock/models/stock_package_level.py b/addons/stock/models/stock_package_level.py
new file mode 100644
index 00000000..008c9c80
--- /dev/null
+++ b/addons/stock/models/stock_package_level.py
@@ -0,0 +1,214 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from itertools import groupby
+from operator import itemgetter
+from collections import defaultdict
+
+from odoo import _, api, fields, models
+
+
+class StockPackageLevel(models.Model):
+    _name = 'stock.package_level'
+    _description = 'Stock Package Level'
+    _check_company_auto = True
+
+    package_id = fields.Many2one(
+        'stock.quant.package', 'Package', required=True, check_company=True,
+        domain="[('location_id', 'child_of', parent.location_id), '|', ('company_id', '=', False), ('company_id', '=', company_id)]")
+    picking_id = fields.Many2one('stock.picking', 'Picking', check_company=True)
+    move_ids = fields.One2many('stock.move', 'package_level_id')
+    move_line_ids = fields.One2many('stock.move.line', 'package_level_id')
+    location_id = fields.Many2one('stock.location', 'From', compute='_compute_location_id', check_company=True)
+    location_dest_id = fields.Many2one(
+        'stock.location', 'To', check_company=True,
+        domain="[('id', 'child_of', parent.location_dest_id), '|', ('company_id', '=', False), ('company_id', '=', company_id)]")
+    is_done = fields.Boolean('Done', compute='_compute_is_done', inverse='_set_is_done')
+    state = fields.Selection([
+        ('draft', 'Draft'),
+        ('confirmed', 'Confirmed'),
+        ('assigned', 'Reserved'),
+        ('new', 'New'),
+        ('done', 'Done'),
+        ('cancel', 'Cancelled'),
+    ],string='State', compute='_compute_state')
+    is_fresh_package = fields.Boolean(compute='_compute_fresh_pack')
+
+    picking_type_code = fields.Selection(related='picking_id.picking_type_code')
+    show_lots_m2o = fields.Boolean(compute='_compute_show_lot')
+    show_lots_text = fields.Boolean(compute='_compute_show_lot')
+    company_id = fields.Many2one('res.company', 'Company', required=True, index=True)
+
+    @api.depends('move_line_ids', 'move_line_ids.qty_done')
+    def _compute_is_done(self):
+        for package_level in self:
+            # If it is an existing package
+            if package_level.is_fresh_package:
+                package_level.is_done = True
+            else:
+                package_level.is_done = package_level._check_move_lines_map_quant_package(package_level.package_id)
+
+    def _set_is_done(self):
+        for package_level in self:
+            if package_level.is_done:
+                if not package_level.is_fresh_package:
+                    ml_update_dict = defaultdict(float)
+                    for quant in package_level.package_id.quant_ids:
+                        corresponding_ml = package_level.move_line_ids.filtered(lambda ml: ml.product_id == quant.product_id and ml.lot_id == quant.lot_id)
+                        if corresponding_ml:
+                            ml_update_dict[corresponding_ml[0]] += quant.quantity
+                        else:
+                            corresponding_move = package_level.move_ids.filtered(lambda m: m.product_id == quant.product_id)[:1]
+                            self.env['stock.move.line'].create({
+                                'location_id': package_level.location_id.id,
+                                'location_dest_id': package_level.location_dest_id.id,
+                                'picking_id': package_level.picking_id.id,
+                                'product_id': quant.product_id.id,
+                                'qty_done': quant.quantity,
+                                'product_uom_id': quant.product_id.uom_id.id,
+                                'lot_id': quant.lot_id.id,
+                                'package_id': package_level.package_id.id,
+                                'result_package_id': package_level.package_id.id,
+                                'package_level_id': package_level.id,
+                                'move_id': corresponding_move.id,
+                                'owner_id': quant.owner_id.id,
+                            })
+                    for rec, quant in ml_update_dict.items():
+                        rec.qty_done = quant
+            else:
+                package_level.move_line_ids.filtered(lambda ml: ml.product_qty == 0).unlink()
+                package_level.move_line_ids.filtered(lambda ml: ml.product_qty != 0).write({'qty_done': 0})
+
+    @api.depends('move_line_ids', 'move_line_ids.package_id', 'move_line_ids.result_package_id')
+    def _compute_fresh_pack(self):
+        for package_level in self:
+            if not package_level.move_line_ids or all(ml.package_id and ml.package_id == ml.result_package_id for ml in package_level.move_line_ids):
+                package_level.is_fresh_package = False
+            else:
+                package_level.is_fresh_package = True
+
+    @api.depends('move_ids', 'move_ids.state', 'move_line_ids', 'move_line_ids.state')
+    def _compute_state(self):
+        for package_level in self:
+            if not package_level.move_ids and not package_level.move_line_ids:
+                package_level.state = 'draft'
+            elif not package_level.move_line_ids and package_level.move_ids.filtered(lambda m: m.state not in ('done', 'cancel')):
+                package_level.state = 'confirmed'
+            elif package_level.move_line_ids and not package_level.move_line_ids.filtered(lambda ml: ml.state == 'done'):
+                if package_level.is_fresh_package:
+                    package_level.state = 'new'
+                elif package_level._check_move_lines_map_quant_package(package_level.package_id, 'product_uom_qty'):
+                    package_level.state = 'assigned'
+                else:
+                    package_level.state = 'confirmed'
+            elif package_level.move_line_ids.filtered(lambda ml: ml.state =='done'):
+                package_level.state = 'done'
+            elif package_level.move_line_ids.filtered(lambda ml: ml.state == 'cancel') or package_level.move_ids.filtered(lambda m: m.state == 'cancel'):
+                package_level.state = 'cancel'
+            else:
+                package_level.state = 'draft'
+
+    def _compute_show_lot(self):
+        for package_level in self:
+            if any(ml.product_id.tracking != 'none' for ml in package_level.move_line_ids):
+                if package_level.picking_id.picking_type_id.use_existing_lots or package_level.state == 'done':
+                    package_level.show_lots_m2o = True
+                    package_level.show_lots_text = False
+                else:
+                    if self.picking_id.picking_type_id.use_create_lots and package_level.state != 'done':
+                        package_level.show_lots_m2o = False
+                        package_level.show_lots_text = True
+                    else:
+                        package_level.show_lots_m2o = False
+                        package_level.show_lots_text = False
+            else:
+                package_level.show_lots_m2o = False
+                package_level.show_lots_text = False
+
+    def _generate_moves(self):
+        for package_level in self:
+            if package_level.package_id:
+                for quant in package_level.package_id.quant_ids:
+                    self.env['stock.move'].create({
+                        'picking_id': package_level.picking_id.id,
+                        'name': quant.product_id.display_name,
+                        'product_id': quant.product_id.id,
+                        'product_uom_qty': quant.quantity,
+                        'product_uom': quant.product_id.uom_id.id,
+                        'location_id': package_level.location_id.id,
+                        'location_dest_id': package_level.location_dest_id.id,
+                        'package_level_id': package_level.id,
+                        'company_id': package_level.company_id.id,
+                    })
+
+    @api.model
+    def create(self, vals):
+        result = super(StockPackageLevel, self).create(vals)
+        if vals.get('location_dest_id'):
+            result.mapped('move_line_ids').write({'location_dest_id': vals['location_dest_id']})
+            result.mapped('move_ids').write({'location_dest_id': vals['location_dest_id']})
+        return result
+
+    def write(self, vals):
+        result = super(StockPackageLevel, self).write(vals)
+        if vals.get('location_dest_id'):
+            self.mapped('move_line_ids').write({'location_dest_id': vals['location_dest_id']})
+            self.mapped('move_ids').write({'location_dest_id': vals['location_dest_id']})
+        return result
+
+    def unlink(self):
+        self.mapped('move_ids').write({'package_level_id': False})
+        self.mapped('move_line_ids').write({'result_package_id': False})
+        return super(StockPackageLevel, self).unlink()
+
+    def _check_move_lines_map_quant_package(self, package, field='qty_done'):
+        """ should compare in good uom """
+        all_in = True
+        pack_move_lines = self.move_line_ids
+        keys = ['product_id', 'lot_id']
+
+        def sorted_key(object):
+            object.ensure_one()
+            return [object.product_id.id, object.lot_id.id]
+
+        grouped_quants = {}
+        for k, g in groupby(sorted(package.quant_ids, key=sorted_key), key=itemgetter(*keys)):
+            grouped_quants[k] = sum(self.env['stock.quant'].concat(*list(g)).mapped('quantity'))
+
+        grouped_ops = {}
+        for k, g in groupby(sorted(pack_move_lines, key=sorted_key), key=itemgetter(*keys)):
+            grouped_ops[k] = sum(self.env['stock.move.line'].concat(*list(g)).mapped(field))
+        if any(grouped_quants.get(key, 0) - grouped_ops.get(key, 0) != 0 for key in grouped_quants) \
+                or any(grouped_ops.get(key, 0) - grouped_quants.get(key, 0) != 0 for key in grouped_ops):
+            all_in = False
+        return all_in
+
+    @api.depends('package_id', 'state', 'is_fresh_package', 'move_ids', 'move_line_ids')
+    def _compute_location_id(self):
+        for pl in self:
+            if pl.state == 'new' or pl.is_fresh_package:
+                pl.location_id = False
+            elif pl.package_id:
+                pl.location_id = pl.package_id.location_id
+            elif pl.state == 'confirmed' and pl.move_ids:
+                pl.location_id = pl.move_ids[0].location_id
+            elif pl.state in ('assigned', 'done') and pl.move_line_ids:
+                pl.location_id = pl.move_line_ids[0].location_id
+            else:
+                pl.location_id = pl.picking_id.location_id
+
+    def action_show_package_details(self):
+        self.ensure_one()
+        view = self.env.ref('stock.package_level_form_edit_view', raise_if_not_found=False) or self.env.ref('stock.package_level_form_view')
+
+        return {
+            'name': _('Package Content'),
+            'type': 'ir.actions.act_window',
+            'view_mode': 'form',
+            'res_model': 'stock.package_level',
+            'views': [(view.id, 'form')],
+            'view_id': view.id,
+            'target': 'new',
+            'res_id': self.id,
+            'flags': {'mode': 'readonly'},
+        }