diff --git a/addons/membership/membership.py b/addons/membership/membership.py index f25ac7f047e..8921cab52d3 100644 --- a/addons/membership/membership.py +++ b/addons/membership/membership.py @@ -477,6 +477,15 @@ class Invoice(osv.osv): '''Invoice''' _inherit = 'account.invoice' + def action_cancel_draft(self, cr, uid, ids, context=None): + member_line_obj = self.pool.get('membership.membership_line') + for invoice in self.browse(cr, uid, ids, context=context): + mlines = member_line_obj.search(cr, uid, + [('account_invoice_line', 'in', + [l.id for l in invoice.invoice_line])]) + member_line_obj.write(cr, uid, mlines, {'date_cancel': False}, context=context) + return super(Invoice, self).action_cancel_draft(cr, uid, ids, context=context) + def action_cancel(self, cr, uid, ids, context=None): '''Create a 'date_cancel' on the membership_line object''' member_line_obj = self.pool.get('membership.membership_line')