Add cidade route #56

Author: samuelgrigolato

PostmonServer.py

@@ -127,6 +127,21 @@ def uf(sigla):
         return
 
 
+@app_v1.route('/cidade/<sigla_uf>/<nome>')
+def cidade(sigla_uf, nome):
+    db = Database()
+    result = _get_cidade_info(db, sigla_uf, nome)
+    if result:
+        response.headers['Cache-Control'] = 'public, max-age=2592000'
+        return format_result(result)
+    else:
+        message = '404 A cidade %s (%s) informada ' \
+                  'nao pode ser localizada'
+        response.status = message % (sigla_uf, nome)
+        print response.status
+        return
+
+
 @app_v1.route('/rastreio/<provider>/<track>')
 def track_pack(provider, track):
     if provider == 'ect':

database.py

@@ -29,7 +29,8 @@ def get_one_uf(self, sigla, **kwargs):
         return self._db.ufs.find_one({'sigla': sigla}, **kwargs)
 
     def get_one_cidade(self, sigla_uf_nome_cidade, **kwargs):
-        return self._db.cidades.find_one({'sigla_uf_nome_cidade': sigla_uf_nome_cidade}, **kwargs)
+        spec = {'sigla_uf_nome_cidade': sigla_uf_nome_cidade}
+        return self._db.cidades.find_one(spec, **kwargs)
 
     def get_one_uf_by_nome(self, nome, **kwargs):
         return self._db.ufs.find_one({'nome': nome}, **kwargs)