1 files changed, 28 insertions, 0 deletions
diff --git a/addons/snailmail/wizard/snailmail_letter_format_error.py b/addons/snailmail/wizard/snailmail_letter_format_error.py
new file mode 100644
index 00000000..38168101
--- /dev/null
+++ b/addons/snailmail/wizard/snailmail_letter_format_error.py
@@ -0,0 +1,28 @@
+
+from odoo import api, fields, models
+
+class SnailmailLetterFormatError(models.TransientModel):
+    _name = 'snailmail.letter.format.error'
+    _description = 'Format Error Sending a Snailmail Letter'
+
+    message_id = fields.Many2one(
+        'mail.message',
+        default=lambda self: self.env.context.get('message_id', None),
+    )
+    snailmail_cover = fields.Boolean(
+        string='Add a Cover Page',
+        default=lambda self: self.env.company.snailmail_cover,
+    )
+
+    def update_resend_action(self):
+        self.env.company.write({'snailmail_cover': self.snailmail_cover})
+        letters_to_resend = self.env['snailmail.letter'].search([
+            ('error_code', '=', 'FORMAT_ERROR'),
+        ])
+        for letter in letters_to_resend:
+            letter.attachment_id.unlink()
+            letter.write({'cover': self.snailmail_cover})
+            letter.snailmail_print()
+
+    def cancel_letter_action(self):
+        self.message_id.cancel_letter()