good/warehouse/report/stock_transceive.py

from odoo import models, fields, api
from datetime import datetime, timedelta


class ReportStockTransceive(models.TransientModel):
    _name = 'report.stock.transceive'
    _description = '商品收发明细表'

    goods = fields.Char('商品')
    code = fields.Char('编码')
    attribute = fields.Char('属性')
    id_lists = fields.Text('库存调拨id列表')
    uom = fields.Char('单位')
    warehouse = fields.Char('仓库')
    goods_qty_begain = fields.Float(
        '期初数量', digits='Quantity')
    cost_begain = fields.Float(
        '期初成本', digits='Amount')
    goods_qty_end = fields.Float(
        '期末数量', digits='Quantity')
    cost_end = fields.Float(
        '期末成本', digits='Amount')
    goods_qty_out = fields.Float(
        '出库数量', digits='Quantity')
    cost_out = fields.Float(
        '出库成本', digits='Amount')
    goods_qty_in = fields.Float(
        '入库数量', digits='Quantity')
    cost_in = fields.Float(
        '入库成本', digits='Amount')

    def select_sql(self, sql_type='out'):
        return '''
        SELECT goods.name as goods,
                goods.code as code,
                att.name as attribute,
                array_agg(line.id) as id_lists,
                uom.name as uom,
                wh.name as warehouse,
                sum(case when
                    line.date < '{date_start}' THEN line.goods_qty ELSE 0 END)
                    as goods_qty_begain,
                sum(case when
                    line.date < '{date_start}' THEN line.cost ELSE 0 END)
                    as cost_begain,
                sum(case when
                    line.date <= '{date_end}' THEN line.goods_qty ELSE 0 END)
                    as goods_qty_end,
                sum(case when
                    line.date <= '{date_end}' THEN line.cost ELSE 0 END)
                    as cost_end,
                sum(case when
                    line.date <= '{date_end}' AND line.date >= '{date_start}'
                  THEN
                    line.goods_qty ELSE 0 END)
                    as goods_qty,
                sum(case when
                    line.date <= '{date_end}' AND line.date >= '{date_start}'
                  THEN
                    line.cost ELSE 0 END)
                    as cost
        '''

    def from_sql(self, sql_type='out'):
        return '''
        FROM wh_move_line line
            LEFT JOIN goods goods ON line.goods_id = goods.id
            LEFT JOIN attribute att ON line.attribute_id = att.id
            LEFT JOIN uom uom ON line.uom_id = uom.id
            LEFT JOIN warehouse wh ON line.%s = wh.id
        ''' % (sql_type == 'out' and 'warehouse_id' or 'warehouse_dest_id')

    def where_sql(self, sql_type='out'):
        extra = ''
        if self.env.context.get('warehouse_id'):
            extra += 'AND wh.id = {warehouse_id}'
        if self.env.context.get('goods_id'):
            extra += 'AND goods.id = {goods_id}'
        return '''
        WHERE line.state = 'done'
          AND wh.type = 'stock'
          AND wh.active = true
          AND line.date <= '{date_end}'
          %s
        ''' % extra

    def group_sql(self, sql_type='out'):
        return '''
        GROUP BY goods.id, goods.code, att.name, uom.name, wh.name
        '''

    def order_sql(self, sql_type='out'):
        return '''
        ORDER BY goods.id, wh.name
        '''

    def get_record_key(self, record, sql_type='out'):
        return (
            record.get('goods'),
            record.get('uom'),
            record.get('warehouse'),
            record.get('attribute'),
            record.get('code')
        )

    def unzip_record_key(self, key):
        return {
            'goods': key[0],
            'uom': key[1],
            'warehouse': key[2],
            'attribute': key[3],
            'code': key[4],
        }

    def get_default_value_by_record(self, record, sql_type='out'):
        return {
            'id': record.get('id'),
        }

    def update_record_value(self, value, record, sql_type='out'):
        tag = sql_type == 'out' and -1 or 1

        value.update({
            'goods_qty_begain': value.get('goods_qty_begain', 0) +
                    (tag * record.get('goods_qty_begain', 0)),
            'cost_begain': value.get('cost_begain', 0) +
                    (tag * record.get('cost_begain', 0)),
            'goods_qty_end': value.get('goods_qty_end', 0) +
                    (tag * record.get('goods_qty_end', 0)),
            'cost_end': value.get('cost_end', 0) +
                    (tag * record.get('cost_end', 0)),

            'goods_qty_out': value.get('goods_qty_out', 0) +
                    (sql_type == 'out' and record.get('goods_qty', 0) or 0),
            'cost_out': value.get('cost_out', 0) +
                    (sql_type == 'out' and record.get('cost', 0) or 0),
            'goods_qty_in': value.get('goods_qty_in', 0) +
                    (sql_type == 'in' and record.get('goods_qty', 0) or 0),
            'cost_in': value.get('cost_in', 0) +
                    (sql_type == 'in' and record.get('cost', 0) or 0),
            'id_lists': value.get('id_lists', []) + record.get('id_lists', []),
        })

    def compute_history_stock_by_collect(self, res, records, sql_type='out'):
        for record in records:
            record_key = self.get_record_key(record, sql_type=sql_type)
            if not res.get(record_key):
                res[record_key] = self.get_default_value_by_record(
                    record, sql_type=sql_type)

            self.update_record_value(
                res[record_key], record, sql_type=sql_type)

    def execute_sql(self, sql_type='out', wizard_id=False):
        context = {
            'date_start': wizard_id.date_start or '',
            'date_end': wizard_id.date_end or '',
            'warehouse_id': wizard_id.warehouse_id and wizard_id.warehouse_id[0].id or '',
            'goods_id': wizard_id.goods_id and wizard_id.goods_id[0].id or '',
        }
        for key, value in list(context.items()):
            if key == "date_end":
                continue
            if key == "date_start":
                continue
            if isinstance(context[key], str):
                context[key] = value.encode('utf-8')
        self.env.context = dict(self.env.context, **context)
        search_sql = (self.select_sql(sql_type) + self.from_sql(sql_type) + self.where_sql(
            sql_type) + self.group_sql(sql_type) + self.order_sql(
            sql_type)).format(**context)

        self.env.cr.execute(search_sql)

        return self.env.cr.dictfetchall()

    def collect_data_by_sql(self, wizard_id=False):
        out_collection = self.execute_sql(sql_type='out', wizard_id=wizard_id)
        in_collection = self.execute_sql(sql_type='in', wizard_id=wizard_id)

        res = {}
        self.compute_history_stock_by_collect(res, in_collection, sql_type='in')
        self.compute_history_stock_by_collect(res, out_collection, sql_type='out')

        result = []
        for key, value in res.items():
            value.update(self.unzip_record_key(key))
            result.append(value)

        return result
    
    def find_source_move_line(self):
        # 查看库存调拨明细
        move_line_ids = []
        # 获得'report.stock.transceive'记录集

        date_start = self.env.context.get('date_start')
        date_end = self.env.context.get('date_end')
        domain_dict = [
            ('date', '>=', date_start),
            ('date', '<=', date_end),
            ('id', 'in', [int(mid) for mid in self.id_lists[1:-1].split(',')]),
        ]

        view = self.env.ref('warehouse.wh_move_line_list')
        return {
            'name': ('库存调拨' + str(date_start) +
                     '~' + str(date_end) +
                     '~' + self.goods),
            'view_mode': 'list',
            'views': [(view.id, 'list')],
            'res_model': 'wh.move.line',
            'type': 'ir.actions.act_window',
            'domain': domain_dict
        }