Cédric Krier
381c49ea1f
Since locale folder has been created with the translations, Python3 tests failed because 2to3 converts the import of locale into a relative import because it detects the locale folder.
71 lines
2.2 KiB
Python
71 lines
2.2 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 __future__ import absolute_import
|
|
|
|
import locale
|
|
|
|
from zeep.exceptions import Fault
|
|
|
|
from trytond.model import ModelSQL, ModelView, MatchMixin, fields
|
|
from trytond.pool import PoolMeta
|
|
|
|
from .configuration import get_client, LOGIN_SERVICE
|
|
|
|
__all__ = ['CredentialDPD', 'Carrier']
|
|
|
|
|
|
class CredentialDPD(ModelSQL, ModelView, MatchMixin):
|
|
'DPD Credential'
|
|
__name__ = 'carrier.credential.dpd'
|
|
|
|
company = fields.Many2One('company.company', 'Company')
|
|
user_id = fields.Char('User ID', required=True)
|
|
password = fields.Char('Password', required=True)
|
|
server = fields.Selection([
|
|
('testing', 'Testing'),
|
|
('production', 'Production'),
|
|
], 'Server')
|
|
depot = fields.Char('Depot', readonly=True)
|
|
token = fields.Char('Token', readonly=True)
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(CredentialDPD, cls).__setup__()
|
|
cls._error_messages.update({
|
|
'dpd_webservice_error': ('DPD webservice call failed with the'
|
|
' following error message:\n\n%(message)s'),
|
|
})
|
|
|
|
@classmethod
|
|
def default_server(cls):
|
|
return 'testing'
|
|
|
|
def update_token(self):
|
|
auth_client = get_client(self.server, LOGIN_SERVICE)
|
|
lang = (self.company.party.lang.code
|
|
if self.company.party.lang else 'en')
|
|
lang = locale.normalize(lang)[:5]
|
|
try:
|
|
result = auth_client.service.getAuth(
|
|
delisId=self.user_id, password=self.password,
|
|
messageLanguage=lang)
|
|
except Fault as e:
|
|
error_message = e.detail[0].find('errorMessage')
|
|
self.raise_user_error('authentication_error', {
|
|
'message': error_message.text,
|
|
})
|
|
|
|
self.token = result.authToken
|
|
self.depot = result.depot
|
|
self.save()
|
|
|
|
|
|
class Carrier:
|
|
__metaclass__ = PoolMeta
|
|
__name__ = 'carrier'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Carrier, cls).__setup__()
|
|
cls.shipping_service.selection.append(('dpd', 'DPD'))
|