mirror of
https://github.com/NaN-tic/trytond-sale_processing2confirmed.git
synced 2023-12-14 03:32:57 +01:00
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
# This file is part of sale_confirmed2quotation module for Tryton.
|
|
# The COPYRIGHT file at the top level of this repository contains
|
|
# the full copyright notices and license terms.
|
|
from trytond.model import Workflow, ModelView
|
|
from trytond.pool import PoolMeta, Pool
|
|
from trytond.pyson import Eval
|
|
|
|
__all__ = ['Sale']
|
|
|
|
|
|
class Sale:
|
|
__metaclass__ = PoolMeta
|
|
__name__ = 'sale.sale'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Sale, cls).__setup__()
|
|
cls._transitions.add(
|
|
('processing', 'draft'),
|
|
)
|
|
cls._buttons['draft']['invisible'] = ~Eval('state').in_(
|
|
['cancel', 'quotation', 'processing'])
|
|
|
|
@classmethod
|
|
def draft(cls, sales):
|
|
pool = Pool()
|
|
Invoice = pool.get('account.invoice')
|
|
ShipmentOut = pool.get('stock.shipment.out')
|
|
ShipmentOutReturn = pool.get('stock.shipment.out.return')
|
|
|
|
# Check if production module is installed
|
|
production_installed = False
|
|
if hasattr(sales[0], 'productions'):
|
|
production_installed = True
|
|
Production = pool.get('production')
|
|
|
|
to_write = []
|
|
to_delete_invoices = []
|
|
to_delete_shipments = []
|
|
to_delete_shipments_return = []
|
|
to_delete_productions = []
|
|
for sale in sales:
|
|
if sale.state != 'processing':
|
|
continue
|
|
if sale.invoices:
|
|
to_delete_invoices += sale.invoices
|
|
|
|
if sale.shipments or sale.shipment_returns:
|
|
to_delete_shipments += sale.shipments
|
|
to_delete_shipments_return += sale.shipment_returns
|
|
|
|
if production_installed:
|
|
to_delete_productions += sale.productions
|
|
to_write.extend(([sale], {'state': 'draft'}))
|
|
|
|
if to_write:
|
|
cls.write(*to_write)
|
|
if to_delete_invoices:
|
|
Invoice.delete(to_delete_invoices)
|
|
if to_delete_shipments:
|
|
ShipmentOut.delete(to_delete_shipments)
|
|
if to_delete_shipments_return:
|
|
ShipmentOutReturn.delete(to_delete_shipments_return)
|
|
if production_installed and to_delete_productions:
|
|
Production.delete(to_delete_productions)
|
|
|
|
super(Sale, cls).draft(sales)
|