SunjiERP/odoo/custom/account/account_ledger/wizard/ledger_inquiry.py

# -*- coding: utf-8 -*-
from collections import defaultdict
from datetime import timedelta

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


BALANCE_DIRECTION = [('debit', '借'), ('credit', '贷'), ('balance', '平')]


def balance_debit_to_balance_direction(balance_debit):
    if balance_debit > 0:
        balance = balance_debit
        direction = 'debit'
    elif balance_debit < 0:
        balance = - balance_debit
        direction = 'credit'
    else:
        balance = 0
        direction = 'balance'

    return balance, direction


class LedgerInquiryWizard(models.TransientModel):
    """总账查询向导"""
    _name = 'ledger.inquiry.wizard'
    _inherit = 'inquiry.wizard.mixin'
    _description = '总账查询向导'

    # company_id = fields.Many2one('res.company', string='公司', default=lambda self: self.env.company.id)

    def confirm(self):
        self.ensure_one()
        # 获取总账窗口动作
        act_window = self.env['ir.actions.act_window'].for_xml_id('account_ledger', 'AccountLedgerActionList')
        act_window['domain'] = eval(act_window['domain'])
        act_window['context'] = eval(act_window['context'])
        act_window['display_name'] = '总账：'

        periods = self.env['fr.account.period'].search(
            [('date_start', '>=', self.period_from.date_start), ('date_start', '<=', self.period_to.date_start)])

        # 添加会计期间筛选
        if self.period_from and self.period_to:
            act_window['domain'] += [('period_id', 'in', periods.ids)]
            act_window['display_name'] = f'总账：{self.period_from.name} - {self.period_to.name}'

        # 添加会计科目筛选
        if len(self.account_ids) > 0:
            act_window['domain'] += [('account_id', 'in', self.account_ids.ids)]

        return act_window


    def confirm(self):
        self.ensure_one()
        # 获取要打印的科目
        if self.account_ids:
            account_ids = self.account_ids
        else:
            account_ids = self.env['account.account'].search([('state', '=', 'on_use')])

        periods = self.env['fr.account.period'].search(
            [('date_start', '>=', self.period_from.date_start), ('date_start', '<=', self.period_to.date_start)])

        date_start = self.period_from.date_start
        date = date_start - timedelta(days=1)
        balance_data = self._fetch_balance_data(self.period_from.company_id.id, date)
        period_data = self._fetch_period_data(self.period_from.company_id.id)

        # 生成临时总账报表记录
        vals_list = []
        for account in account_ids:
            descendant_ids = account.get_descendant_ids(leaf=True)

            # 上年结转/初始余额
            balance_start = sum([balance_data[descendant_id] for descendant_id in descendant_ids])
            balance_temp, direction_temp = balance_debit_to_balance_direction(balance_start)
            vals_list.append({
                'account_id': account.id,
                'name': '初始余额',
                'direction': direction_temp,
                'balance': balance_temp
            })

            # 期间总账
            balance = balance_start
            debit_fiscalyear = credit_fiscalyear = 0
            for period in periods.filtered(lambda x: x.state != 'unuse'):
                debit_period = credit_period = 0

                # 从下级科目获取
                for descendant_id in descendant_ids:
                    data_key = f"{descendant_id}-{period.id}"
                    debit_period += period_data[data_key][0]
                    credit_period += period_data[data_key][1]
                balance += debit_period - credit_period
                balance, direction = balance_debit_to_balance_direction(balance)

                # 本期合计
                vals_list.append({
                    'account_id': account.id,
                    'period_id': period.id,
                    'name': '本期合计',
                    'debit': debit_period,
                    'credit': credit_period,
                    'direction': direction,
                    'balance': balance
                })

                # 本年累计
                debit_fiscalyear += debit_period
                credit_fiscalyear += credit_period
                vals_list.append({
                    'account_id': account.id,
                    'period_id': period.id,
                    'name': '本年累计',
                    'debit': debit_fiscalyear,
                    'credit': credit_fiscalyear,
                })

        # 创建临时记录集
        report_recs = self.env['account.ledger.temp'].create(vals_list)

        # 获取总账窗口动作
        act_window = self.env['ir.actions.act_window']._for_xml_id('account_ledger.AccountLedgerTempActionList')
        act_window['domain'] = eval(act_window['domain'])
        act_window['context'] = eval(act_window['context'])
        act_window['display_name'] = '总账：'

        # 添加会计期间筛选
        if self.period_from and self.period_to:
            act_window['domain'] += [('id', 'in', report_recs.ids)]
            act_window['display_name'] = f'总账：{self.period_from.name} - {self.period_to.name}'

        # 添加会计科目筛选
        if len(self.account_ids) > 0:
            act_window['domain'] += [('account_id', 'in', self.account_ids.ids)]

        return act_window

    def _fetch_balance_data(self, company_id, date_end):
        # 期末余额数据读取
        self.env.cr.execute(f"""
            SELECT
                account_id,
                sum( balance ) AS balance
            FROM
                account_move_line
            WHERE
                company_id = {company_id} and date <= '{date_end}' and fr_state = 'posted'
            GROUP BY
                account_id
        """)
        balance_data = defaultdict(float)
        for res in self.env.cr.fetchall():
            balance_data[res[0]] = res[1]

        return balance_data

    def _fetch_period_data(self, company_id):
        # 期间借贷发生数据读取
        self.env.cr.execute(f"""
            SELECT
                account_id,
                fr_period_id,
                sum( debit ) AS debit,
                sum( credit ) AS credit
            FROM
                account_move_line
            WHERE
                company_id = {company_id} and fr_state = 'posted'
            GROUP BY
                account_id,
                fr_period_id
        """)

        period_data = defaultdict(lambda: (0, 0))

        for res in self.env.cr.fetchall():
            data_key = f"{res[0]}-{res[1]}"
            period_data[data_key] = (res[2], res[3])

        return period_data


class AccountLedgerTemp(models.TransientModel):
    _name = 'account.ledger.temp'
    _description = '总账'
    _order = 'account_id, id'

    # 基础字段
    name = fields.Char(string='摘要')
    debit = fields.Monetary(string='借方')
    credit = fields.Monetary(string='贷方')
    direction = fields.Selection(BALANCE_DIRECTION, string='方向')
    balance = fields.Monetary(string='余额')

    # 关系字段
    period_id = fields.Many2one('fr.account.period', string='会计期间')
    account_id = fields.Many2one('account.account', strint='会计科目')
    fiscalyear_id = fields.Many2one('fr.account.fiscalyear', string='会计年度')

    # 关联字段
    company_id = fields.Many2one(related='account_id.company_id', string='公司')
    currency_id = fields.Many2one(related='account_id.currency_id', string='币种')