passerelle/passerelle/apps/atal/models.py

254 lines
9.5 KiB
Python

# passerelle - uniform access to multiple data sources and services
# Copyright (C) 2019 Entr'ouvert
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import base64
from django.db import models
from django.utils.six.moves import urllib
from django.utils.translation import ugettext_lazy as _
import lxml.etree
from zeep import helpers
from zeep.exceptions import Fault
from passerelle.base.models import BaseResource
from passerelle.utils.api import endpoint
from passerelle.utils.jsonresponse import APIError
from . import schemas
def process_response(demande_number):
if not demande_number.startswith('DIT'):
raise APIError(demande_number)
return {'data': {'demande_number': demande_number}}
class ATALConnector(BaseResource):
base_soap_url = models.URLField(
max_length=400, verbose_name=_('Base SOAP endpoint'),
help_text=_('URL of the base SOAP endpoint'))
category = _('Business Process Connectors')
class Meta:
verbose_name = _('ATAL connector')
def _soap_call(self, wsdl, method, **kwargs):
wsdl_url = urllib.parse.urljoin(self.base_soap_url, '%s?wsdl' % wsdl)
client = self.soap_client(wsdl_url=wsdl_url)
try:
return getattr(client.service, method)(**kwargs)
except Fault as e:
raise APIError(unicode(e))
def _basic_ref(self, wsdl, method):
soap_res = self._soap_call(wsdl=wsdl, method=method)
res = []
for elem in soap_res:
res.append({'id': elem.code, 'text': elem.libelle})
return {'data': res}
@endpoint(methods=['get'], perm='can_access', name='get-type-activite')
def get_type_activite(self, request):
return self._basic_ref('VilleAgileService', 'getTypeActivite')
@endpoint(methods=['get'], perm='can_access', name='get-type-de-voie')
def get_type_de_voie(self, request):
return self._basic_ref('VilleAgileService', 'getTypeDeVoie')
@endpoint(methods=['get'], perm='can_access', name='get-types-equipement')
def get_types_equipement(self, request):
soap_res = self._soap_call(wsdl='VilleAgileService', method='getTypesEquipement')
tree = lxml.etree.fromstring(soap_res.encode('utf-8')).getroottree()
types = tree.xpath('//types')[0]
res = []
for type_elem in types.getchildren():
res.append({'id': type_elem.get('id'), 'text': type_elem.get('label')})
return {'data': res}
@endpoint(
perm='can_access', name='insert-action-comment',
post={
'description': 'Insert action comment',
'request_body': {
'schema': {
'application/json': schemas.INSERT_ACTION_COMMENT
}
}
}
)
def insert_action_comment(self, request, post_data):
demande_number = self._soap_call(
wsdl='DemandeService', method='insertActionComment',
numeroDemande=post_data['numero_demande'],
commentaire=post_data['commentaire']
)
return process_response(demande_number)
@endpoint(
perm='can_access', name='insert-demande-complet-by-type',
post={
'description': 'Insert demande complet by type',
'request_body': {
'schema': {
'application/json': schemas.INSERT_DEMANDE_COMPLET_BY_TYPE
}
}
}
)
def insert_demande_complet_by_type(self, request, post_data):
data = {}
for recv, send in [
('type_demande', 'typeDemande'),
('code_service_demandeur', 'codeServiceDemandeur'),
('date_saisie', 'dateSaisie'),
('date_demande', 'dateDemande'),
('date_souhaite', 'dateSouhaitee'),
('date_butoir', 'dateButoir'),
('contact_civilite', 'contactCivilite'),
('contact_nom', 'contactNom'),
('contact_prenom', 'contactPrenom'),
('contact_tel', 'contactTelephone'),
('contact_mobile', 'contactMobile'),
('contact_email', 'contactCourriel'),
('contact_info_compl', 'contactInfoCompl'),
('demande_type_support', 'demandeTypeDeSupport'),
('contact_adresse', 'contactAdresse'),
('contact_adresse_compl', 'contactAdresseCompl'),
('contact_code_postal', 'contactCodePostal'),
('contact_ville', 'contactVille'),
('contact_organisme', 'contactOrganisme'),
('contact_titre', 'contactTitre'),
('code_equipement', 'codeEquipement'),
('code_mairie_equipement', 'codeMairieEquipement'),
('code_sig_equipement', 'codeSIGEquipement'),
('code_collectivite_equipement', 'codeCollectiviteEquipement'),
('code_quartier_equipement', 'codeQuartierEquipement'),
('code_type_equipement', 'codeTypeEquipement'),
('demande_lieu', 'demandeLieu'),
('coord_x', 'coordX'),
('coord_y', 'coordY'),
('demande_priorite', 'demandePriorite'),
('demande_objet', 'demandeObjet'),
('demande_description', 'demandeDescription'),
('demande_commentaire', 'demandeCommentaire'),
('demande_mots_cles', 'demandeMotsCles'),
('code_thematique', 'codeThematiqueATAL'),
('code_priorite', 'codePrioriteATAL'),
('demande_thematique', 'demandeThematique'),
('code_projet', 'codeProjetATAL'),
]:
if recv in post_data:
data[send] = post_data[recv]
demande_number = self._soap_call(
wsdl='DemandeService', method='insertDemandeCompletByType', **data
)
return process_response(demande_number)
@endpoint(
methods=['get'], perm='can_access', example_pattern='{demande_number}/',
pattern='^(?P<demande_number>\w+)/$', name='retrieve-details-demande',
parameters={
'demande_number': {
'description': _('Demande number'), 'example_value': 'DIT18050001'
}
}
)
def retrieve_details_demande(self, request, demande_number):
soap_res = self._soap_call(
wsdl='DemandeService', method='retrieveDetailsDemande',
demandeNumberParam=demande_number)
return {'data': helpers.serialize_object(soap_res)}
@endpoint(
methods=['get'], perm='can_access', example_pattern='{demande_number}/',
pattern='^(?P<demande_number>\w+)/$', name='retrieve-etat-travaux',
parameters={
'demande_number': {
'description': _('Demande number'), 'example_value': 'DIT18050001'
}
}
)
def retrieve_etat_travaux(self, request, demande_number):
soap_res = self._soap_call(
wsdl='DemandeService', method='retrieveEtatTravaux',
numero=demande_number)
return {'data': helpers.serialize_object(soap_res)}
@endpoint(
methods=['get'], perm='can_access', example_pattern='{demande_number}/',
pattern='^(?P<demande_number>\w+)/$', name='status',
parameters={
'demande_number': {
'description': _('Demande number'), 'example_value': 'DIT18050001'
}
}
)
def status(self, request, demande_number):
soap_res = self._soap_call(
wsdl='DemandeService', method='retrieveDetailsDemande',
demandeNumberParam=demande_number)
status = helpers.serialize_object(soap_res).get('etatDemande', {}).get('description')
if not status:
raise APIError('Could not get a status')
if status != 'PRISE EN COMPTE':
return {
'data': {
'status': status
}
}
soap_res = self._soap_call(
wsdl='DemandeService', method='retrieveEtatTravaux',
numero=demande_number
)
status = helpers.serialize_object(soap_res).get('libelle')
if not status:
raise APIError('Could not get a status')
return {
'data': {
'status': status
}
}
@endpoint(
perm='can_access',
post={
'description': 'Upload a file',
'request_body': {
'schema': {
'application/json': schemas.UPLOAD
}
}
}
)
def upload(self, request, post_data):
try:
content = base64.b64decode(post_data['file']['content'])
except TypeError:
raise APIError('Invalid base64 string')
data = {
'donneesFichier': content,
'numeroDemande': post_data['numero_demande'],
'nomFichier': post_data['nom_fichier']
}
self._soap_call(
wsdl='ChargementPiecesJointesService', method='upload',
**data
)
return {}