l10n_it_split_payment: fix check_balanced

Author: TennyMkt

l10n_it_split_payment/models/account_move.py

@@ -42,18 +42,19 @@ def _compute_amount(self):
         return res
 
     def write(self, vals):
+        if self.env.context.get("skip_split_payment_computation"):
+            return super(AccountMove, self).write(vals)
         res = super(AccountMove, self.with_context(check_move_validity=False)).write(
             vals
         )
-        if self.env.context.get("skip_split_payment_computation"):
-            return res
         if "fiscal_position_id" in vals:
             self._compute_amount()
         self.with_context(
             skip_split_payment_computation=True, check_move_validity=False
         ).compute_split_payment()
         container = {"records": self}
-        self._check_balanced(container)
+        with self._check_balanced(container):
+            pass
         return res
 
     def copy(self, default=None):

l10n_it_split_payment/models/account_move_line.py

@@ -17,7 +17,7 @@ class AccountMoveLine(models.Model):
     def _compute_is_split_payment(self):
         for line in self:
             line.is_split_payment = False
-            if line.account_id == line.company_id.sp_account_id:
+            if line.account_id and line.account_id == line.company_id.sp_account_id:
                 line.is_split_payment = True
 
     def _compute_all_tax(self):