1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
|
# 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.pool import PoolMeta, Pool
__all__ = ['PartyReplace', 'PartyErase']
class PartyReplace(metaclass=PoolMeta):
__name__ = 'party.replace'
@classmethod
def fields_to_replace(cls):
return super(PartyReplace, cls).fields_to_replace() + [
('purchase.product_supplier', 'party'),
('purchase.purchase', 'party'),
]
class PartyErase(metaclass=PoolMeta):
__name__ = 'party.erase'
@classmethod
def __setup__(cls):
super(PartyErase, cls).__setup__()
cls._error_messages.update({
'pending_purchase': (
'The party "%(party)s" can not be erased '
'because he has pending purchases '
'for the company "%(company)s".'),
})
def check_erase_company(self, party, company):
pool = Pool()
Purchase = pool.get('purchase.purchase')
super(PartyErase, self).check_erase_company(party, company)
purchases = Purchase.search([
('party', '=', party.id),
('state', 'not in', ['done', 'cancel']),
])
if purchases:
self.raise_user_error('pending_purchase', {
'party': party.rec_name,
'company': company.rec_name,
})
|