from odoo import api, fields, models, _

from ..utils import MODULE_NAME


class SaleOrder(models.Model):
    _inherit = "sale.order"

    # TODO: exists original field "commitment_date".
    delivery_time = fields.Integer(
        string="Delivery Time",
        default=45,
    )
    contract_annex_id = fields.Many2one(
        comodel_name="res.partner.contract.annex",
        string="Contract Annex",
        compute="get_contract_annex_id",
    )
    contract_annex_ids = fields.One2many(
        comodel_name="res.partner.contract.annex",
        inverse_name="order_id",
        string="Annex for this Sale order",
        help="Technical field for binding with contract annex\n"
        "In form this link showed in 'contract_annex_id' field.",
    )

    # Extend default field for showing payment terms created by this module only.
    payment_term_id = fields.Many2one(
        comodel_name="account.payment.term",
        domain=lambda self: [("id", "in", self._get_payment_terms())],
    )

    def _get_payment_terms(self):
        terms = [
            self.env.ref("{}.{}".format(MODULE_NAME, external_id)).id
            for external_id in (
                "payment_term_prepaid",
                "payment_term_postpayment",
                "payment_term_partial_2",
                "payment_term_partial_3",
            )
        ]
        return terms

    @api.onchange("contract_annex_ids")
    def get_contract_annex_id(self):
        if self.contract_annex_ids:
            self.contract_annex_id = self.contract_annex_ids[0].id
        else:
            self.contract_annex_id = False

    def action_print_form(self):
        self.ensure_one()
        view = self.env.ref(
            "{}.res_partner_wizard_print_document_view".format(MODULE_NAME)
        )
        return {
            "name": _("Print Form of Contract"),
            "type": "ir.actions.act_window",
            "res_model": "res.partner.contract.wizard",
            "view_mode": "form",
            "view_id": view.id,
            "target": "new",
            "context": {
                "self_id": self.id,
                "active_model": self._name,
                "company_form": self.partner_id.company_form
                if self.partner_id.is_company
                else "person",
                "attachment_model": "sale.order",
            },
        }

    def get_filename_by_document_template(self, document_template_id):
        self.ensure_one()
        return "{doc_type} {number} {from_} {date}".format(
            doc_type=_("Offer"),
            number=self.name,
            from_=_("from"),
            date=self.date_order.strftime("%d.%m.%Y"),
        )

    @staticmethod
    def _t(arg):
        """Uses in xml action (data/fields_default)
        Arguments:
            arg {str} -- String to translate
        """
        return _(arg)