trytond-csv_sale_shipment_cost/csv_import.py

29 lines
853 B
Python

# This file is part of csv_sale_shipment_cost module for Tryton.
# The COPYRIGHT file at the top level of this repository contains
# the full copyright notices and license terms.
from trytond.pool import Pool, PoolMeta
__all__ = ['CSVArchive']
class CSVArchive:
__metaclass__ = PoolMeta
__name__ = 'csv.archive'
@classmethod
def post_import(cls, profile, records):
'''
Sale post import add delivery - shipment cost line
'''
Sale = Pool().get('sale.sale')
super(CSVArchive, cls).post_import(profile, records)
if profile.model.model == 'sale.sale':
to_write = []
for record in records:
sale = Sale(record)
sale.set_shipment_cost()
to_write.append(sale)
if to_write:
Sale.save(to_write)