Verificando se localidade existe e se não existe disparando exceção L…

…ocalidade Não Existe.

open_cnl/open_cnl.py

@@ -38,13 +38,8 @@ def buscar_localidade(self, prefixo, sufixo, prefixo_de_referencia=None,
             localidade_de_referencia = self._buscar_localidade(
                 prefixo_de_referencia, sufixo_de_referencia)
 
-        if not localidade_de_referencia:
-            return None
-
         # Buscar localidade para comparação.
         localidade = self._buscar_localidade(prefixo, sufixo)
-        if not localidade:
-            return None
 
         if localidade[5] == localidade_de_referencia[5]:
             # Código Nacional de Localidade igual em ambas as localidades.
@@ -66,4 +61,11 @@ def _buscar_localidade(self, prefixo, sufixo):
             AND CAST(numero_da_faixa_inicial as INTEGER) <= ?
             AND CAST(numero_da_faixa_final as INTEGER) >= ?;
         """, (prefixo, int(sufixo), int(sufixo))).fetchone()
+
+        if not localidade:
+            raise LocalidadeNaoEncontrada
+
         return localidade
+
+class LocalidadeNaoEncontrada(Exception):
+    pass

tests.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 import unittest
-from open_cnl.open_cnl import OpenCNL
+from open_cnl.open_cnl import OpenCNL, LocalidadeNaoEncontrada
 
 caminho_da_base = './cnl_anatel.sqlite3'
 
@@ -45,5 +45,10 @@ def test_vc2_outro_prefixo(self):
             prefixo_mossoro, sufixo_mossoro)
         self.assertEqual(tarifa, 'VC2')
 
+    def test_localidade_nao_encontrada(self):
+        with self.assertRaises(LocalidadeNaoEncontrada):
+            tarifa = self.cnl.buscar_localidade('9977777', '545454')
+
+
 if __name__ == '__main__':
     unittest.main()