Funcao helper para pegar municipio

pynfe/processamento/serializacao.py

@@ -6,7 +6,7 @@ except:
 
 from pynfe.entidades import Emitente, Cliente, Produto, Transportadora, NotaFiscal
 from pynfe.excecoes import NenhumObjetoEncontrado, MuitosObjetosEncontrados
-from pynfe.utils import etree, so_numeros, obter_municipio_por_codigo, obter_pais_por_codigo, obter_codigo_por_municipio
+from pynfe.utils import etree, so_numeros, obter_municipio_por_codigo, obter_pais_por_codigo, obter_municipio_e_codigo
 from pynfe.utils.flags import CODIGOS_ESTADOS, VERSAO_PADRAO
 
 class Serializacao(object):
@@ -352,18 +352,9 @@ class SerializacaoPipes(Serializacao):
 
     def _serializar_emitente(self, emitente):
 
-        try:
-            cod_municipio = int(emitente.endereco_municipio)
-        except ValueError:
-            cod_municipio = obter_codigo_por_municipio(
-                emitente.endereco_municipio,
-                emitente.endereco_uf
-            )
-
-        municipio = obter_municipio_por_codigo(
-            cod_municipio,
-            emitente.endereco_uf,
-            normalizado=True,
+        cod_municipio, municipio = obter_municipio_e_codigo(
+            emitente.endereco_municipio,
+            emitente.endereco_uf
         )
 
         serial_emitente_list = [
@@ -394,3 +385,5 @@ class SerializacaoPipes(Serializacao):
 
         return '|'.join(serial_emitente_list)
 
+    def _serializar_cliente(self, cliente):
+        pass 

pynfe/utils/__init__.py

@@ -110,6 +110,16 @@ def obter_municipio_por_codigo(codigo, uf, normalizado=False):
     return municipio
 
 @memoize
+def obter_municipio_e_codigo(municipio_ou_codigo, uf):
+    try:
+        cod_municipio = int(municipio_ou_codigo)
+    except ValueError:
+        cod_municipio = obter_codigo_por_municipio(municipio_ou_codigo, uf)
+
+    municipio = obter_municipio_por_codigo(cod_municipio, uf, normalizado=True)
+
+    return cod_municipio, municipio
+
+@memoize
 def extrair_tag(root):
     return root.tag.split('}')[-1]
-