Passa auth para o packtrack.

PostmonAPI · Jun 29, 2017 · 68905b5 · 68905b5
1 parent 7479964
commit 68905b5
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/PackTracker.py b/PackTracker.py
@@ -8,10 +8,11 @@
 from database import MongoDb as Database
 
 
-def correios(track, backend=None):
+def correios(track, backend=None, auth=None):
     if backend is None:
         backend = os.getenv('ECT_BACKEND')
-    encomenda = packtrack.Correios.track(track, backend=backend)
+    encomenda = packtrack.Correios.track(track, backend=backend, auth=auth)
+
     if not encomenda:
         raise ValueError(u"Encomenda nao encontrada.")
     if not encomenda.status:

diff --git a/PostmonServer.py b/PostmonServer.py
@@ -192,8 +192,15 @@ def cidade(sigla_uf, nome):
 def track_pack(provider, track):
     response.headers['Access-Control-Allow-Origin'] = '*'
     if provider == 'ect':
+        auth = (
+            request.headers.get('x-correios-usuario'),
+            request.headers.get('x-correios-senha'),
+        )
+        if auth == (None, None):
+            auth = None
+
         try:
-            historico = PackTracker.correios(track)
+            historico = PackTracker.correios(track, auth=auth)
         except (AttributeError, ValueError):
             message = "404 Pacote %s nao encontrado" % track
             logger.exception(message)