Merge pull request #2 from shuhikari/master

Adds new mode of reading municipio

pynfe/processamento/serializacao.py

@@ -35,7 +35,6 @@ class Serializacao(object):
     def exportar(self, destino, **kwargs):
         """Gera o(s) arquivo(s) de exportacao a partir da Nofa Fiscal eletronica
         ou lista delas."""
-
         raise NotImplementedError
 
     def importar(self, origem):
@@ -87,10 +86,9 @@ class SerializacaoXML(Serializacao):
         etree.SubElement(endereco, 'nro').text = emitente.endereco_numero
         etree.SubElement(endereco, 'xCpl').text = emitente.endereco_complemento
         etree.SubElement(endereco, 'xBairro').text = emitente.endereco_bairro
-        etree.SubElement(endereco, 'cMun').text = emitente.endereco_municipio
-        etree.SubElement(endereco, 'xMun').text = obter_municipio_por_codigo(
-                emitente.endereco_municipio, emitente.endereco_uf,
-                )
+        etree.SubElement(endereco, 'cMun').text = obter_codigo_por_municipio(
+            emitente.endereco_municipio, emitente.endereco_uf)
+        etree.SubElement(endereco, 'xMun').text = emitente.endereco_municipio
         etree.SubElement(endereco, 'UF').text = emitente.endereco_uf
         etree.SubElement(endereco, 'CEP').text = so_numeros(emitente.endereco_cep)
         etree.SubElement(endereco, 'cPais').text = emitente.endereco_pais
@@ -116,10 +114,9 @@ class SerializacaoXML(Serializacao):
         etree.SubElement(endereco, 'nro').text = cliente.endereco_numero
         etree.SubElement(endereco, 'xCpl').text = cliente.endereco_complemento
         etree.SubElement(endereco, 'xBairro').text = cliente.endereco_bairro
-        etree.SubElement(endereco, 'cMun').text = cliente.endereco_municipio
-        etree.SubElement(endereco, 'xMun').text = obter_municipio_por_codigo(
-                cliente.endereco_municipio, cliente.endereco_uf,
-                )
+        etree.SubElement(endereco, 'cMun').text = obter_codigo_por_municipio(
+            cliente.endereco_municipio, cliente.endereco_uf)
+        etree.SubElement(endereco, 'xMun').text = cliente.endereco_municipio
         etree.SubElement(endereco, 'UF').text = cliente.endereco_uf
         etree.SubElement(endereco, 'CEP').text = so_numeros(cliente.endereco_cep)
         etree.SubElement(endereco, 'cPais').text = cliente.endereco_pais

pynfe/utils/__init__.py

@@ -34,18 +34,21 @@ import flags
 
 # from geraldo.utils import memoize
 
+
 # @memoize
 def so_numeros(texto):
     """Retorna o texto informado mas somente os numeros"""
     return ''.join(filter(lambda c: ord(c) in range(48, 58), texto))
 
+
 # @memoize
 def obter_pais_por_codigo(codigo):
     # TODO
     if codigo == '1058':
         return 'Brasil'
 
-CAMINHO_DATA = os.path.join(os.path.dirname(os.path.abspath(__file__)), '..', 'data')
+CAMINHO_DATA = os.path.join(os.path.dirname(
+    os.path.abspath(__file__)), '..', 'data')
 CAMINHO_MUNICIPIOS = os.path.join(CAMINHO_DATA, 'MunIBGE')
 CARACTERS_ACENTUADOS = {
     ord(u'á'): u'a',
@@ -62,6 +65,7 @@ CARACTERS_ACENTUADOS = {
     ord(u'ç'): u'c',
 }
 
+
 # @memoize
 def normalizar_municipio(municipio):
     if not isinstance(municipio, unicode):
@@ -69,6 +73,7 @@ def normalizar_municipio(municipio):
 
     return municipio.lower().translate(CARACTERS_ACENTUADOS).upper()
 
+
 # @memoize
 def carregar_arquivo_municipios(uf, reverso=False):
     if isinstance(uf, basestring):
@@ -98,23 +103,25 @@ def carregar_arquivo_municipios(uf, reverso=False):
 
     return municipios_dict
 
+
 # @memoize
 def obter_codigo_por_municipio(municipio, uf):
     # TODO: fazer UF ser opcional
     municipios = carregar_arquivo_municipios(uf, True)
     return municipios[normalizar_municipio(municipio)]
 
+
 # @memoize
 def obter_municipio_por_codigo(codigo, uf, normalizado=False):
     # TODO: fazer UF ser opcional
     municipios = carregar_arquivo_municipios(uf)
-
-    municipio = municipios[unicode(codigo)]
+    municipio = municipios.get(unicode(codigo))
     if normalizado:
         return normalizar_municipio(municipio)
 
     return municipio
 
+
 # @memoize
 def obter_municipio_e_codigo(municipio_ou_codigo, uf):
     try:
@@ -126,16 +133,19 @@ def obter_municipio_e_codigo(municipio_ou_codigo, uf):
 
     return cod_municipio, municipio
 
+
 # @memoize
 def extrair_tag(root):
     return root.tag.split('}')[-1]
 
+
 def formatar_decimal(dec):
     if dec * 100 - int(dec * 100):
         return str(dec)
     else:
         return "%.2f" % dec
 
+
 def safe_str(str_):
     if not isinstance(str_, unicode):
         if isinstance(str_, str):