# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.

from odoo import models


class PurchaseOrder(models.Model):
    _inherit = "purchase.order"

    def _prepare_invoice(self):
        res = super()._prepare_invoice()
        if (
            self.partner_id.country_id
            and 'INTRASTAT' in self.partner_id.country_id.country_group_codes
        ):
            res["intrastat_country_id"] = self.partner_id.country_id.id
        return res