Merge pull request #68 from rochacbruno/master

Suporte a XML, executar direto com python PostmonServer.py, XML testes, suporte a MongoDb com auth.

Author: aleborba

PostmonServer.py

@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 import bottle
 import json
+import xmltodict
 from bottle import route, run, response
 from CepTracker import CepTracker
 import requests
@@ -36,6 +37,12 @@ def format_result(result):
     # checa se foi solicitada resposta em JSONP
     js_func_name = bottle.request.query.get(jsonp_query_key)
 
+    #checa se foi solicitado xml
+    format = bottle.request.query.get('format')
+    if format == 'xml':
+        response.content_type = 'application/xml'
+        return xmltodict.unparse({'result': result})
+
     if js_func_name:
         # se a resposta vai ser JSONP, o content type deve ser js e seu
         # conteudo deve ser JSON
@@ -71,7 +78,6 @@ def verifica_cep(cep):
         result = db.get_one(cep, fields={'_id': False, 'v_date': False})
 
     if result:
-
         response.headers['Cache-Control'] = 'public, max-age=2592000'
         return format_result(result)
     else:
@@ -110,9 +116,12 @@ def track_pack(provider, track):
     else:
         response.status = '404 O Servico %s nao pode ser encontrado' % provider
 
-
 bottle.mount('/v1', app_v1)
 
 
 def _standalone(port=9876):
-    run(host='localhost', port=port)
+    run(host='0.0.0.0', port=port)
+
+
+if __name__ == "__main__":
+    _standalone()

README.md

@@ -32,7 +32,31 @@ na raiz do projeto.
 Rodando a aplicação localmente na porta 9876
 --------------------------------
 
+	$ python PostmonServer.py
+
+ou
+
 	$ ipython -i PostmonServer.py
 	>> _standalone()
 
 Caso queira rodar em outra porta, basta passá-la como parametro no chamado do _standalone
+
+
+MongoDB com autenticação
+------------------------
+
+Se o seu MongoDB possui autenticação habilitada você deverá configurar o banco do postmon
+e também exportar duas variavéis de ambiente.
+
+```javascript
+mongo
+> use postmon
+> db.addUser('admin', '123456')
+```
+
+Agora que seu Mongo está com password exporte as variaveis de ambiente.
+
+```bash
+export POSTMON_DB_USER=admin
+export POSTMON_DB_PASSWORD=123456
+```

database.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
+import os
 import pymongo
 
 
@@ -15,6 +16,10 @@ class MongoDb(object):
 
     def __init__(self, address='localhost'):
         self._client = pymongo.MongoClient(address)
+        USERNAME = os.environ.get('POSTMON_DB_USER')
+        PASSWORD = os.environ.get('POSTMON_DB_PASSWORD')
+        if all((USERNAME, PASSWORD)):
+            self._client.postmon.authenticate(USERNAME, PASSWORD)
         self._db = self._client.postmon
 
     def get_one(self, cep, **kwargs):

requirements.txt

@@ -5,4 +5,5 @@ pymongo==2.4
 requests==0.14.2
 webtest==1.4.3
 packtrack==0.1.0
-flake8
+xmltodict
+flake8

test/postmon_test.py

@@ -168,3 +168,27 @@ class PostmonV1WebTest(PostmonWebTest):
     def get_cep(self, cep):
         response = self.app.get('/v1/cep/' + cep)
         return response.json
+
+
+class PostmonXMLTest(unittest.TestCase):
+    """ testa requisições XML """
+
+    def setUp(self):
+        self.app = webtest.TestApp(bottle.app())
+
+    def get_cep(self, cep):
+        response = self.app.get(
+            '/cep/%s?format=xml' % cep
+        )
+        return response
+
+    def test_xml_return(self):
+        import xmltodict
+        response = self.get_cep('06708070')
+        parsed = xmltodict.parse(response.body)
+        result = parsed.get('result')
+        self.assertEqual(result['bairro'], u'Parque S\xe3o George')
+        self.assertEqual(result['cidade'], u'Cotia')
+        self.assertEqual(result['cep'], u'06708070')
+        self.assertEqual(result['estado'], u'SP')
+        self.assertEqual(result['logradouro'], u'Avenida Eid Mansur')