fix eletronic document message disappearing when acconunt invoice is …

…created

br_account/models/account_invoice.py

@@ -11,6 +11,23 @@
 class AccountInvoice(models.Model):
     _inherit = 'account.invoice'
 
+    @api.multi
+    def write(self, vals):
+        if self.state == 'draft':
+            fis_obs_ids = ({'fiscal_observation_ids': [
+                x.id for x in self.fiscal_position_id.fiscal_observation_ids]})
+            vals.update({'fiscal_observation_ids': (6, None, fis_obs_ids)})
+        return super(AccountInvoice, self).write(vals)
+
+    @api.model
+    def create(self, vals):
+        if "fiscal_position_id" in vals:
+            fpos_id = self.env['account.fiscal.position'].browse(
+                vals["fiscal_position_id"])
+            vals.update({'fiscal_observation_ids': [
+                (4, x.id, None) for x in fpos_id.fiscal_observation_ids]})
+        return super(AccountInvoice, self).create(vals)
+
     @api.one
     @api.depends('invoice_line_ids.price_subtotal',
                  'invoice_line_ids.price_total',