Add cidade_info attr into cep route return obj #56

Author: samuelgrigolato

IbgeTracker.py

@@ -57,8 +57,8 @@ def _get_info_cidades(self):
             infos.append({
                 'codigo_ibge_uf': celulas[0].text_content(),
                 'sigla_uf': celulas[1].text_content(),
-                'codigo_ibge_cidade': celulas[2].text_content(),
-                'nome_cidade': celulas[3].text_content(),
+                'codigo_ibge': celulas[2].text_content(),
+                'nome': celulas[3].text_content(),
                 'area_km2': celulas[4].text_content()
             })
         return infos
@@ -74,15 +74,15 @@ def _track_cidades(self, db):
         for info in infos:
             codigo_ibge_uf = info['codigo_ibge_uf']
             sigla_uf = info['sigla_uf']
-            nome_cidade = info['nome_cidade']
+            nome = info['nome']
             if codigo_ibge_uf not in siglas:
                 siglas[codigo_ibge_uf] = sigla_uf
 
             # a chave única de uma cidade não
             # pode ser só o nome, pois
             # existem cidades com mesmo nome
             # em estados diferentes
-            info['sigla_uf_nome_cidade'] = '%s_%s' % (sigla_uf, nome_cidade)
+            info['sigla_uf_nome_cidade'] = '%s_%s' % (sigla_uf, nome)
 
             db.insert_or_update_cidade(info)
 

PostmonServer.py

@@ -58,6 +58,19 @@ def _get_estado_info(db, sigla):
     return db.get_one_uf(sigla, fields={'_id': False, 'sigla': False})
 
 
+def _get_cidade_info(db, sigla_uf, nome_cidade):
+    sigla_uf = sigla_uf.upper()
+    sigla_uf_nome_cidade = '%s_%s' % (sigla_uf, nome_cidade)
+    fields = {
+        '_id': False,
+        'sigla_uf': False,
+        'codigo_ibge_uf': False,
+        'sigla_uf_nome_cidade': False,
+        'nome': False
+    }
+    return db.get_one_cidade(sigla_uf_nome_cidade, fields=fields)
+
+
 @route('/cep/<cep:re:\d{5}-?\d{3}>')
 @app_v1.route('/cep/<cep:re:\d{5}-?\d{3}>')
 def verifica_cep(cep):
@@ -84,9 +97,14 @@ def verifica_cep(cep):
 
     if result:
         response.headers['Cache-Control'] = 'public, max-age=2592000'
-        estado_info = _get_estado_info(db, result['estado'])
+        sigla_uf = result['estado']
+        estado_info = _get_estado_info(db, sigla_uf)
         if estado_info:
             result['estado_info'] = estado_info
+        nome_cidade = result['cidade']
+        cidade_info = _get_cidade_info(db, sigla_uf, nome_cidade)
+        if cidade_info:
+            result['cidade_info'] = cidade_info
         return format_result(result)
     else:
         response.status = '404 O CEP %s informado nao pode ser '

database.py

@@ -28,6 +28,9 @@ def get_one(self, cep, **kwargs):
     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)
+
     def get_one_uf_by_nome(self, nome, **kwargs):
         return self._db.ufs.find_one({'nome': nome}, **kwargs)