from odoo import Command, fields, models
from odoo.addons.account.models.chart_template import template


class AccountChartTemplate(models.AbstractModel):
    _inherit = 'account.chart.template'

    @template('in')
    def _get_in_template_data(self):
        return {
            'property_account_receivable_id': 'p10040',
            'property_account_payable_id': 'p11211',
            'property_account_expense_categ_id': 'p2107',
            'property_account_income_categ_id': 'p20011',
            'code_digits': '6',
            'display_invoice_amount_total_words': True,
        }

    @template('in', 'res.company')
    def _get_in_res_company(self):
        return {
            self.env.company.id: {
                'account_fiscal_country_id': 'base.in',
                'bank_account_code_prefix': '1002',
                'cash_account_code_prefix': '1001',
                'transfer_account_code_prefix': '1008',
                'account_default_pos_receivable_account_id': 'p10041',
                'income_currency_exchange_account_id': 'p2013',
                'expense_currency_exchange_account_id': 'p2117',
                'account_journal_early_pay_discount_loss_account_id': 'p2132',
                'account_journal_early_pay_discount_gain_account_id': '2012',
                'account_opening_date': fields.Date.context_today(self).replace(month=4, day=1),
                'fiscalyear_last_month': '3',
                'account_sale_tax_id': 'sgst_sale_5',
                'account_purchase_tax_id': 'sgst_purchase_5',
                'deferred_expense_account_id': 'p10084',
                'deferred_revenue_account_id': 'p10085',
            },
        }

    @template('in', 'account.cash.rounding')
    def _get_in_account_cash_rounding(self):
        return {
            'l10n_in.cash_rounding_in_half_up': {
                'profit_account_id': 'p213202',
                'loss_account_id': 'p213201',
            }
        }

    @template('in', 'account.fiscal.position')
    def _get_in_account_fiscal_position(self):
        _ = self.env._
        company = self.env.company
        state_ids = [Command.set(company.state_id.ids)] if company.state_id else False
        intra_state_name = company.state_id and _("Within %s", company.state_id.name) or _("Intra State")
        country_in_id = self.env.ref('base.in').id
        state_specific = {
            'fiscal_position_in_intra_state': {
                'name': intra_state_name,
                'sequence': 1,
                'auto_apply': True,
                'state_ids': state_ids,
                'country_id': country_in_id,
            },
            'fiscal_position_in_inter_state': {
                'name': _("Inter State"),
                'sequence': 2,
                'auto_apply': True,
                'tax_ids': self._get_l10n_in_fiscal_tax_vals(),
                'country_id': country_in_id,
            },
        }
        if company.parent_id:
            return state_specific
        return {
            **state_specific,
            'fiscal_position_in_export_sez_in': {
                'name': _("Export"),
                'sequence': 3,
                'auto_apply': True,
                'note': _("SUPPLY MEANT FOR EXPORT/SUPPLY TO SEZ UNIT OR SEZ DEVELOPER FOR AUTHORISED OPERATIONS ON PAYMENT OF INTEGRATED TAX."),
                'tax_ids': (
                    self._get_l10n_in_fiscal_tax_vals(trailing_id='_sez_exp')
                    + self._get_l10n_in_zero_rated_with_igst_zero_tax_vals()
                ),
            },
            'fiscal_position_in_sez': {
                'name': _("Special Economic Zone (SEZ)"),
                'sequence': 4,
                'auto_apply': True,
                'state_ids': [Command.set(self.env.ref('l10n_in.state_in_oc').ids)],
                'country_id': country_in_id,
                'note': _("SUPPLY MEANT FOR EXPORT/SUPPLY TO SEZ UNIT OR SEZ DEVELOPER FOR AUTHORISED OPERATIONS ON PAYMENT OF INTEGRATED TAX."),
                'tax_ids': (
                    self._get_l10n_in_fiscal_tax_vals()
                    + self._get_l10n_in_zero_rated_with_igst_zero_tax_vals()
                ),
            },
            'fiscal_position_in_lut_sez': {
                'name': _("Export - LUT (WOP)"),
                'sequence': 5,
                'note': _("SUPPLY MEANT FOR EXPORT/SUPPLY TO SEZ UNIT OR SEZ DEVELOPER FOR AUTHORISED OPERATIONS UNDER BOND OR LETTER OF UNDERTAKING WITHOUT PAYMENT OF INTEGRATED TAX."),
                'tax_ids': (
                    self._get_l10n_in_fiscal_tax_vals(use_zero_rated_igst=True, trailing_id='_sez_exp_lut')
                    + self._get_l10n_in_zero_rated_with_igst_zero_tax_vals()
                )
            },
            'fiscal_position_in_lut_sez_1': {
                'name': _("SEZ - LUT (WOP)"),
                'sequence': 6,
                'state_ids': [Command.set(self.env.ref('l10n_in.state_in_oc').ids)],
                'country_id': country_in_id,
                'note': _("SUPPLY MEANT FOR EXPORT/SUPPLY TO SEZ UNIT OR SEZ DEVELOPER FOR AUTHORISED OPERATIONS UNDER BOND OR LETTER OF UNDERTAKING WITHOUT PAYMENT OF INTEGRATED TAX."),
                'tax_ids': (
                    self._get_l10n_in_fiscal_tax_vals(use_zero_rated_igst=True, trailing_id='_sez_lut')
                    + self._get_l10n_in_zero_rated_with_igst_zero_tax_vals()
                )
            },
        }

    def _get_l10n_in_fiscal_tax_vals(self, use_zero_rated_igst=False, trailing_id=False):
        return [Command.clear()] + [
            Command.create({
                'tax_src_id': f"sgst_{tax_type}_{rate}",
                'tax_dest_id': f"igst_{tax_type}_{0 if use_zero_rated_igst and tax_type == 'purchase' else rate}{(tax_type == 'sale' and trailing_id) or ''}",
            })
            for tax_type in ["sale", "purchase"]
            for rate in [1, 2, 5, 12, 18, 28]  # Available existing GST Rates
        ]

    def _get_l10n_in_zero_rated_with_igst_zero_tax_vals(self):
        return [
            Command.create({
                'tax_src_id': zero_tax,
                'tax_dest_id': "igst_sale_0"
            })
            for zero_tax in ["exempt_sale", "nil_rated_sale"]
        ]