use local wsdl files to improve client load

olist · Aug 29, 2017 · 0f1c3e0 · 0f1c3e0
1 parent 093fe24
commit 0f1c3e0
Show file tree

Hide file tree

Showing 9 changed files with 5,263 additions and 7 deletions.
diff --git a/correios/client.py b/correios/client.py
@@ -21,7 +21,7 @@
 from correios import xml_utils, DATADIR
 from correios.exceptions import PostingListSerializerError, TrackingCodesLimitExceededError
 from correios.models.data import EXTRA_SERVICE_MP, EXTRA_SERVICE_AR
-from correios.utils import to_decimal, to_integer
+from correios.utils import to_decimal, to_integer, get_wsdl_path
 from .models.address import ZipAddress, ZipCode
 from .models.posting import (NotFoundTrackingEvent, TrackingCode, PostingList, ShippingLabel,
                              TrackingEvent, EventStatus, Package, Freight, FreightError)
@@ -311,11 +311,11 @@ class Correios:
 
     # 'environment': ('url', 'ssl_verification')
     sigep_urls = {
-        'production': ("https://apps.correios.com.br/SigepMasterJPA/AtendeClienteService/AtendeCliente?wsdl", True),
-        'test': ("https://apphom.correios.com.br/SigepMasterJPA/AtendeClienteService/AtendeCliente?wsdl", False),
+        'production': (get_wsdl_path('AtendeCliente-production.wsdl'), True),
+        'test': (get_wsdl_path('AtendeCliente-test.wsdl'), False),
     }
-    websro_url = "https://webservice.correios.com.br/service/rastro/Rastro.wsdl"
-    freight_url = "http://ws.correios.com.br/calculador/CalcPrecoPrazo.asmx?WSDL"
+    websro_url = get_wsdl_path('Rastro.wsdl')
+    freight_url = get_wsdl_path('CalcPrecoPrazo.asmx')
 
     def __init__(self, username, password, timeout=8, environment="production"):
         self.username = username

diff --git a/correios/utils.py b/correios/utils.py
@@ -3,6 +3,7 @@
 from itertools import chain
 from typing import Container, Iterable, Sized, Union
 
+import os
 import re
 
 
@@ -81,3 +82,9 @@ def to_decimal(value: Union[Decimal, str, float], precision=2):
 
     quantize = Decimal("0." + "0" * precision)
     return value.quantize(quantize)
+
+
+def get_wsdl_path(filename) -> str:
+    return os.path.abspath(
+        'correios/wsdls/{}'.format(filename)
+    )
diff --git a/correios/wsdls/AtendeCliente-production.wsdl b/correios/wsdls/AtendeCliente-production.wsdl
diff --git a/correios/wsdls/AtendeCliente-test.wsdl b/correios/wsdls/AtendeCliente-test.wsdl
diff --git a/correios/wsdls/CalcPrecoPrazo.asmx b/correios/wsdls/CalcPrecoPrazo.asmx
diff --git a/correios/wsdls/Rastro.wsdl b/correios/wsdls/Rastro.wsdl
@@ -0,0 +1,64 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<definitions name="rastro" targetNamespace="http://resource.webservice.correios.com.br/" xmlns:wsp1_2="http://schemas.xmlsoap.org/ws/2004/09/policy" xmlns="http://schemas.xmlsoap.org/wsdl/" xmlns:wsp="http://www.w3.org/ns/ws-policy" xmlns:wsam="http://www.w3.org/2007/05/addressing/metadata" xmlns:xsd="http://www.w3.org/2001/XMLSchema" xmlns:wsu="http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-utility-1.0.xsd" xmlns:tns="http://resource.webservice.correios.com.br/" xmlns:soap="http://schemas.xmlsoap.org/wsdl/soap/">
+  <types>
+    <xsd:schema>
+      <xsd:import namespace="http://resource.webservice.correios.com.br/" schemaLocation="Rastro_schema1.xsd"/>
+    </xsd:schema>
+  </types>
+  <message name="buscaEventosLista">
+    <part name="parameters" element="tns:buscaEventosLista">
+    </part>
+  </message>
+  <message name="buscaEventosResponse">
+    <part name="parameters" element="tns:buscaEventosResponse">
+    </part>
+  </message>
+  <message name="buscaEventosListaResponse">
+    <part name="parameters" element="tns:buscaEventosListaResponse">
+    </part>
+  </message>
+  <message name="buscaEventos">
+    <part name="parameters" element="tns:buscaEventos">
+    </part>
+  </message>
+  <portType name="Service">
+    <operation name="buscaEventos">
+      <input message="tns:buscaEventos" wsam:Action="buscaEventos">
+    </input>
+      <output message="tns:buscaEventosResponse" wsam:Action="http://resource.webservice.correios.com.br/Service/buscaEventosResponse">
+    </output>
+    </operation>
+    <operation name="buscaEventosLista">
+      <input message="tns:buscaEventosLista" wsam:Action="buscaEventosLista">
+    </input>
+      <output message="tns:buscaEventosListaResponse" wsam:Action="http://resource.webservice.correios.com.br/Service/buscaEventosListaResponse">
+    </output>
+    </operation>
+  </portType>
+  <binding name="ServicePortBinding" type="tns:Service">
+    <soap:binding style="document" transport="http://schemas.xmlsoap.org/soap/http"/>
+    <operation name="buscaEventos">
+      <soap:operation soapAction="buscaEventos"/>
+      <input>
+        <soap:body use="literal"/>
+      </input>
+      <output>
+        <soap:body use="literal"/>
+      </output>
+    </operation>
+    <operation name="buscaEventosLista">
+      <soap:operation soapAction="buscaEventosLista"/>
+      <input>
+        <soap:body use="literal"/>
+      </input>
+      <output>
+        <soap:body use="literal"/>
+      </output>
+    </operation>
+  </binding>
+  <service name="rastro">
+    <port name="ServicePort" binding="tns:ServicePortBinding">
+      <soap:address location="http://webservice.correios.com.br:80/service/rastro"/>
+    </port>
+  </service>
+</definitions>
diff --git a/correios/wsdls/Rastro_schema1.xsd b/correios/wsdls/Rastro_schema1.xsd
@@ -0,0 +1,125 @@
+<?xml version="1.0" encoding="UTF-8"?><!--Generated by the JavaTM Architecture for XML Binding(JAXB) Reference Implementation, (build IBM 2.2.3-11/25/2013 12:35 PM(foreman)-) 
+See <a href="http://java.sun.com/xml/jaxb">http://java.sun.com/xml/jaxb</a> --><xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:tns="http://resource.webservice.correios.com.br/" version="1.0" targetNamespace="http://resource.webservice.correios.com.br/">
+
+  <xs:element name="buscaEventos" type="tns:buscaEventos"></xs:element>
+
+  <xs:element name="buscaEventosLista" type="tns:buscaEventosLista"></xs:element>
+
+  <xs:element name="buscaEventosListaResponse" type="tns:buscaEventosListaResponse"></xs:element>
+
+  <xs:element name="buscaEventosResponse" type="tns:buscaEventosResponse"></xs:element>
+
+  <xs:element name="destino" type="tns:destinos"></xs:element>
+
+  <xs:element name="endereco" type="tns:enderecoMobile"></xs:element>
+
+  <xs:element name="evento" type="tns:eventos"></xs:element>
+
+  <xs:element name="objeto" type="tns:objeto"></xs:element>
+
+  <xs:element name="rastro" type="tns:sroxml"></xs:element>
+
+  <xs:complexType name="buscaEventosLista">
+    <xs:sequence>
+      <xs:element name="usuario" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="senha" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="tipo" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="resultado" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="lingua" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="objetos" type="xs:string" minOccurs="0" maxOccurs="unbounded"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="buscaEventosListaResponse">
+    <xs:sequence>
+      <xs:element name="return" type="tns:sroxml" minOccurs="0"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="sroxml">
+    <xs:sequence>
+      <xs:element name="versao" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="qtd" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="TipoPesquisa" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="TipoResultado" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element ref="tns:objeto" minOccurs="0" maxOccurs="unbounded"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="objeto">
+    <xs:sequence>
+      <xs:element name="numero" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="sigla" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="nome" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="categoria" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="erro" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element ref="tns:evento" minOccurs="0" maxOccurs="unbounded"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="eventos">
+    <xs:sequence>
+      <xs:element name="tipo" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="status" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="data" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="hora" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="descricao" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="detalhe" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="recebedor" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="documento" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="comentario" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="local" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="codigo" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="cidade" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="uf" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="sto" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="amazoncode" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="amazontimezone" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element ref="tns:destino" minOccurs="0" maxOccurs="unbounded"></xs:element>
+      <xs:element ref="tns:endereco" minOccurs="0"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="destinos">
+    <xs:sequence>
+      <xs:element name="local" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="codigo" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="cidade" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="bairro" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="uf" type="xs:string" minOccurs="0"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="enderecoMobile">
+    <xs:sequence>
+      <xs:element name="codigo" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="cep" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="logradouro" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="complemento" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="numero" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="localidade" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="uf" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="bairro" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="latitude" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="longitude" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="celular" type="xs:string" minOccurs="0"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="buscaEventos">
+    <xs:sequence>
+      <xs:element name="usuario" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="senha" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="tipo" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="resultado" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="lingua" type="xs:string" minOccurs="0"></xs:element>
+      <xs:element name="objetos" type="xs:string" minOccurs="0"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+
+  <xs:complexType name="buscaEventosResponse">
+    <xs:sequence>
+      <xs:element name="return" type="tns:sroxml" minOccurs="0"></xs:element>
+    </xs:sequence>
+  </xs:complexType>
+</xs:schema>
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -24,6 +24,7 @@
 from correios.models.posting import (NotFoundTrackingEvent, PostingList, ShippingLabel,
                                      TrackingCode, Package)
 from correios.models.user import PostingCard, Service, ExtraService
+from correios.utils import get_wsdl_path
 from .vcr import vcr
 
 try:
@@ -36,7 +37,7 @@
 @vcr.use_cassette
 def test_basic_client():
     client = correios.Correios(username="sigep", password="XXXXXX", environment=correios.Correios.TEST)
-    assert client.sigep_url == "https://apphom.correios.com.br/SigepMasterJPA/AtendeClienteService/AtendeCliente?wsdl"
+    assert client.sigep_url == get_wsdl_path('AtendeCliente-test.wsdl')
     assert not client.sigep_verify
     assert client.username == "sigep"
     assert client.password == "XXXXXX"

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -2,7 +2,14 @@
 
 import pytest
 
-from correios.utils import capitalize_phrase, RangeSet, rreplace, to_decimal, to_integer
+from correios.utils import (
+    capitalize_phrase,
+    get_wsdl_path,
+    RangeSet,
+    rreplace,
+    to_decimal,
+    to_integer,
+)
 
 phrase = 'FOo bAr BAZ qux'
 
@@ -96,3 +103,9 @@ def test_to_decimal_precision(v, p, r):
 ))
 def test_to_integer(v, r):
     assert to_integer(v) == r
+
+
+def test_get_wsdl_file_path():
+    filepath = get_wsdl_path('fake')
+
+    assert 'correios/wsdls/fake' in filepath