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(
        "res.partner.contract.annex",
        string="Contract Annex",
        readonly=True,
    )
    # Extend default field for showing payment terms created by this module only.
    payment_term_id = fields.Many2one(
        "account.payment.term",
        domain=lambda self: [("id", "in", self._get_payment_terms())],
    )

    def _get_payment_terms(self):
        ref = self.env.ref
        terms = (
            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 list(terms)