230 lines
8.0 KiB
Python
230 lines
8.0 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.
|
|
|
|
import locale
|
|
from io import BytesIO
|
|
from lxml import etree
|
|
from zeep.exceptions import Fault
|
|
from PyPDF2 import PdfFileReader, PdfFileWriter
|
|
|
|
from trytond.i18n import gettext
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.model import fields
|
|
from trytond.model.exceptions import AccessError
|
|
from trytond.wizard import Wizard, StateAction, StateTransition
|
|
from trytond.transaction import Transaction
|
|
|
|
from trytond.modules.stock_package_shipping.exceptions import (
|
|
PackingValidationError)
|
|
|
|
from .configuration import get_client, SHIPMENT_SERVICE
|
|
from .exceptions import DPDError
|
|
|
|
__all__ = ['ShipmentOut', 'CreateShipping', 'CreateDPDShipping']
|
|
|
|
|
|
class ShipmentOut(metaclass=PoolMeta):
|
|
__name__ = 'stock.shipment.out'
|
|
|
|
def validate_packing_dpd(self):
|
|
warehouse_address = self.warehouse.address
|
|
if not warehouse_address:
|
|
raise PackingValidationError(
|
|
gettext('stock_package_shipping_dpd'
|
|
'.msg_warehouse_address_required',
|
|
shipment=self.rec_name,
|
|
warehouse=self.warehouse.rec_name))
|
|
|
|
|
|
class CreateShipping(metaclass=PoolMeta):
|
|
__name__ = 'stock.shipment.create_shipping'
|
|
|
|
dpd = StateAction(
|
|
'stock_package_shipping_dpd.act_create_shipping_dpd_wizard')
|
|
|
|
def transition_start(self):
|
|
pool = Pool()
|
|
ShipmentOut = pool.get('stock.shipment.out')
|
|
|
|
shipment = ShipmentOut(Transaction().context['active_id'])
|
|
next_state = super(CreateShipping, self).transition_start()
|
|
if shipment.carrier.shipping_service == 'dpd':
|
|
next_state = 'dpd'
|
|
return next_state
|
|
|
|
def do_dpd(self, action):
|
|
ctx = Transaction().context
|
|
return action, {
|
|
'model': ctx['active_model'],
|
|
'id': ctx['active_id'],
|
|
'ids': [ctx['active_id']],
|
|
}
|
|
|
|
|
|
class CreateDPDShipping(Wizard):
|
|
'Create DPD Shipping'
|
|
__name__ = 'stock.shipment.create_shipping.dpd'
|
|
|
|
start = StateTransition()
|
|
|
|
def transition_start(self):
|
|
pool = Pool()
|
|
ShipmentOut = pool.get('stock.shipment.out')
|
|
Package = pool.get('stock.package')
|
|
|
|
shipment = ShipmentOut(Transaction().context['active_id'])
|
|
if shipment.reference:
|
|
raise AccessError(
|
|
gettext('stock_package_shipping_dpd'
|
|
'.msg_shipment_has_reference_number',
|
|
shipment=shipment.rec_name))
|
|
|
|
credential = self.get_credential(shipment)
|
|
if not credential.depot or not credential.token:
|
|
credential.update_token()
|
|
|
|
shipping_client = get_client(credential.server, SHIPMENT_SERVICE)
|
|
print_options = self.get_print_options(shipment)
|
|
packages = shipment.root_packages
|
|
shipment_data = self.get_shipment_data(credential, shipment, packages)
|
|
|
|
count = 0
|
|
while count < 2:
|
|
lang = (credential.company.party.lang.code
|
|
if credential.company.party.lang else 'en')
|
|
lang = locale.normalize(lang)[:5]
|
|
authentication = {
|
|
'delisId': credential.user_id,
|
|
'authToken': credential.token,
|
|
'messageLanguage': lang,
|
|
}
|
|
try:
|
|
shipment_response = shipping_client.service.storeOrders(
|
|
print_options, shipment_data, _soapheaders={
|
|
'authentication': authentication,
|
|
})
|
|
break
|
|
except Fault as e:
|
|
tag = etree.QName(e.detail[0].tag)
|
|
if tag.localname == 'authenticationFault':
|
|
count += 1
|
|
credential.update_token()
|
|
else:
|
|
raise
|
|
else:
|
|
raise DPDError(
|
|
gettext('stock_package_shipping_dpd.msg_dpd_login_error',
|
|
credential=credential.rec_name))
|
|
|
|
response, = shipment_response.shipmentResponses
|
|
if response.faults:
|
|
message = '\n'.join(f.message for f in response.faults)
|
|
raise DPDError(
|
|
gettext('stock_package_shipping_dpd.msg_dpd_webservice_error',
|
|
message=message))
|
|
|
|
labels = []
|
|
labels_pdf = BytesIO(shipment_response.parcellabelsPDF)
|
|
reader = PdfFileReader(labels_pdf)
|
|
for page_num in range(reader.getNumPages()):
|
|
new_pdf = PdfFileWriter()
|
|
new_label = BytesIO()
|
|
new_pdf.addPage(reader.getPage(page_num))
|
|
new_pdf.write(new_label)
|
|
labels.append(new_label)
|
|
|
|
shipment.reference = response.mpsId
|
|
parcels = response.parcelInformation
|
|
for package, label, parcel in zip(packages, labels, parcels):
|
|
package.shipping_label = fields.Binary.cast(label.getvalue())
|
|
package.shipping_reference = parcel.parcelLabelNumber
|
|
Package.save(shipment.root_packages)
|
|
shipment.save()
|
|
|
|
return 'end'
|
|
|
|
def get_credential_pattern(self, shipment):
|
|
return {
|
|
'company': shipment.company.id,
|
|
}
|
|
|
|
def get_credential(self, shipment):
|
|
pool = Pool()
|
|
DPDCredential = pool.get('carrier.credential.dpd')
|
|
|
|
credential_pattern = self.get_credential_pattern(shipment)
|
|
for credential in DPDCredential.search([]):
|
|
if credential.match(credential_pattern):
|
|
return credential
|
|
|
|
def get_print_options(self, shipment):
|
|
return {
|
|
'printerLanguage': 'PDF',
|
|
'paperFormat': 'A6',
|
|
}
|
|
|
|
def shipping_party(self, party, address):
|
|
shipping_party = {
|
|
'name1': address.party_full_name[:50],
|
|
'name2': '',
|
|
'street': ' '.join((address.street or '').splitlines())[:35],
|
|
'country': address.country.code if address.country else '',
|
|
'zipCode': address.zip[:9],
|
|
'city': address.city[:50],
|
|
}
|
|
if party.full_name != address.party_full_name:
|
|
shipping_party['name2'] = party.full_name[:35]
|
|
|
|
phone = email = ''
|
|
for mechanism in party.contact_mechanisms:
|
|
if mechanism.type in {'phone', 'mobile'} and not phone:
|
|
phone = mechanism.value
|
|
if mechanism.type == 'email' and not email:
|
|
email = mechanism.value
|
|
if phone:
|
|
shipping_party['phone'] = phone[:30]
|
|
if email:
|
|
shipping_party['email'] = email[:50]
|
|
|
|
return shipping_party
|
|
|
|
def get_parcel(self, package):
|
|
pool = Pool()
|
|
UoM = pool.get('product.uom')
|
|
ModelData = pool.get('ir.model.data')
|
|
|
|
cm = UoM(ModelData.get_id('product', 'uom_centimeter'))
|
|
|
|
parcel = {
|
|
'customerReferenceNumber1': package.code,
|
|
'weight': int(package.total_weight * 10) * 10
|
|
}
|
|
|
|
length = UoM.compute_qty(
|
|
package.type.length_uom, package.type.length, cm)
|
|
width = UoM.compute_qty(
|
|
package.type.width_uom, package.type.width, cm)
|
|
height = UoM.compute_qty(
|
|
package.type.height_uom, package.type.height, cm)
|
|
if length < 1000 and width < 1000 and height < 1000:
|
|
parcel['volume'] = '%03i%03i%03i' % (length, width, height)
|
|
|
|
return parcel
|
|
|
|
def get_shipment_data(self, credential, shipment, packages):
|
|
return {
|
|
'generalShipmentData': {
|
|
'identificationNumber': shipment.number,
|
|
'sendingDepot': credential.depot,
|
|
'product': 'CL',
|
|
'sender': self.shipping_party(
|
|
shipment.company.party, shipment.warehouse.address),
|
|
'recipient': self.shipping_party(
|
|
shipment.customer, shipment.delivery_address),
|
|
},
|
|
'parcels': [self.get_parcel(p) for p in packages],
|
|
'productAndServiceData': {
|
|
'orderType': 'consignment',
|
|
}
|
|
}
|