from odoo import api, fields, models

from ..utils import MODULE_NAME


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

    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
    payment_term_id = fields.Many2one(
        "account.payment.term",
        domain=lambda self: [("id", "in", self._get_payment_terms())],
    )

    @api.multi
    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)