Cédric Krier
13736071c8
* Remove relation field actions:
- delete_all
- unlink_all
- set
* Rename relation field action unlink into remove
issue3457
review2271002
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
#This file is part of Tryton. The COPYRIGHT file at the top level
|
|
#of this repository contains the full copyright notices and license terms.
|
|
from trytond.transaction import Transaction
|
|
from trytond.pool import Pool, PoolMeta
|
|
|
|
__all__ = ['InvoiceLine']
|
|
__metaclass__ = PoolMeta
|
|
|
|
|
|
class InvoiceLine:
|
|
__name__ = 'account.invoice.line'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(InvoiceLine, cls).__setup__()
|
|
cls._error_messages.update({
|
|
'delete_purchase_invoice_line': ('You can not delete '
|
|
'invoice lines that comes from a purchase.'),
|
|
})
|
|
|
|
@classmethod
|
|
def write(cls, *args):
|
|
Purchase = Pool().get('purchase.purchase')
|
|
|
|
actions = iter(args)
|
|
for lines, values in zip(actions, actions):
|
|
if 'invoice' not in values:
|
|
continue
|
|
with Transaction().set_user(0, set_context=True):
|
|
purchases = Purchase.search([
|
|
('invoice_lines', 'in', [l.id for l in lines]),
|
|
])
|
|
if values['invoice']:
|
|
Purchase.write(purchases, {
|
|
'invoices': [('add', [values['invoice']])],
|
|
})
|
|
else:
|
|
for purchase in purchases:
|
|
invoice_ids = list(set([x.invoice.id for x
|
|
in purchase.invoice_lines
|
|
if x.invoice and x.id in lines])
|
|
- set([x.invoice.id for x
|
|
in purchase.invoice_lines
|
|
if x.invoice and x.id not in lines]))
|
|
Purchase.write([purchase], {
|
|
'invoices': [('remove', invoice_ids)],
|
|
})
|
|
|
|
super(InvoiceLine, cls).write(*args)
|
|
|
|
@classmethod
|
|
def delete(cls, lines):
|
|
PurchaseLine = Pool().get('purchase.line')
|
|
if any(l for l in lines
|
|
if isinstance(l.origin, PurchaseLine) and l.type == 'line'):
|
|
cls.raise_user_error('delete_purchase_invoice_line')
|
|
super(InvoiceLine, cls).delete(lines)
|