From 332c02dc0709f2361def7fab06af9878501d5bc0 Mon Sep 17 00:00:00 2001
From: Stefan Rijnhart <stefan@opener.am>
Date: Fri, 15 May 2020 15:51:34 +0200
Subject: [PATCH] [FIX] Onchange methods can be called on newIds that have no
 company yet

---
 account_payment_partner/models/account_invoice.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/account_payment_partner/models/account_invoice.py b/account_payment_partner/models/account_invoice.py
index 95622bcb4..d1b50b0cc 100644
--- a/account_payment_partner/models/account_invoice.py
+++ b/account_payment_partner/models/account_invoice.py
@@ -22,9 +22,12 @@ class AccountInvoice(models.Model):
     def _onchange_partner_id(self):
         res = super(AccountInvoice, self)._onchange_partner_id()
         if self.partner_id:
+            company_id = (self.company_id.id or
+                          self.env.context.get('force_company') or
+                          self.env.user.company_id.id)
             if self.type == 'in_invoice':
                 pay_mode = self.with_context(
-                    force_company=self.company_id.id
+                    force_company=company_id
                 ).partner_id.supplier_payment_mode_id
                 self.payment_mode_id = pay_mode
                 if (
@@ -45,7 +48,7 @@ class AccountInvoice(models.Model):
                 # needed for printing purposes and it can conflict with
                 # SEPA direct debit payments. Current report prints it.
                 self.payment_mode_id = self.with_context(
-                    force_company=self.company_id.id,
+                    force_company=company_id,
                 ).partner_id.customer_payment_mode_id
         else:
             self.payment_mode_id = False