flectra/addons/recurring_account/models/account_invoice.py

33 lines
994 B
Python

# Part of Flectra. See LICENSE file for full copyright and licensing
# details.
from flectra import api, models, fields
class AccountInvoice(models.Model):
_inherit = "account.invoice"
recurring_id = fields.Many2one('recurring', string='Recurring Reference')
rec_source_id = fields.Many2one('account.invoice',
string='Recurring Source')
@api.multi
def get_recurring(self):
result = self.env['recurring'].get_recurring(
'account.invoice', self.id)
return result
@api.multi
def get_recurring_documents(self):
result = self.env['recurring'].get_recurring_documents(
'account.invoice', 'account.action_invoice_tree1',
self.recurring_id)
return result
@api.multi
def action_invoice_cancel(self):
res = super(AccountInvoice, self).action_invoice_cancel()
if self.recurring_id:
self.recurring_id.set_done()
return res