Cédric Krier
5c12d36b7d
We add the flake8 configuration used so we ensure everyone uses the same. We remove the usage of __all__ for non public API. When possible, we rationalize the class name according to its __name__ and module. issue9082 review297061002
49 lines
1.7 KiB
Python
49 lines
1.7 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.i18n import gettext
|
|
from trytond.pool import PoolMeta, Pool
|
|
|
|
from trytond.modules.party.exceptions import EraseError
|
|
|
|
|
|
class Replace(metaclass=PoolMeta):
|
|
__name__ = 'party.replace'
|
|
|
|
@classmethod
|
|
def fields_to_replace(cls):
|
|
return super().fields_to_replace() + [
|
|
('stock.shipment.in', 'supplier'),
|
|
('stock.shipment.in.return', 'supplier'),
|
|
('stock.shipment.out', 'customer'),
|
|
('stock.shipment.out.return', 'customer'),
|
|
]
|
|
|
|
|
|
class Erase(metaclass=PoolMeta):
|
|
__name__ = 'party.erase'
|
|
|
|
def check_erase_company(self, party, company):
|
|
pool = Pool()
|
|
ShipmentIn = pool.get('stock.shipment.in')
|
|
ShipmentInReturn = pool.get('stock.shipment.in.return')
|
|
ShipmentOut = pool.get('stock.shipment.out')
|
|
ShipmentOutReturn = pool.get('stock.shipment.out.return')
|
|
super().check_erase_company(party, company)
|
|
|
|
for Shipment, field in [
|
|
(ShipmentIn, 'supplier'),
|
|
(ShipmentInReturn, 'supplier'),
|
|
(ShipmentOut, 'customer'),
|
|
(ShipmentOutReturn, 'customer'),
|
|
]:
|
|
shipments = Shipment.search([
|
|
(field, '=', party.id),
|
|
('company', '=', company.id),
|
|
('state', 'not in', ['done', 'cancel']),
|
|
])
|
|
if shipments:
|
|
raise EraseError(
|
|
gettext('stock.msg_erase_party_shipment',
|
|
party=party.rec_name,
|
|
company=company.rec_name))
|