diff --git a/pynfe/utils/__init__.py b/pynfe/utils/__init__.py index 0b3c2ab..0f972c1 100644 --- a/pynfe/utils/__init__.py +++ b/pynfe/utils/__init__.py @@ -71,10 +71,13 @@ def normalizar_municipio(municipio): # @memoize def carregar_arquivo_municipios(uf, reverso=False): - caminho_arquivo = os.path.join( - CAMINHO_MUNICIPIOS, - 'MunIBGE-UF%s.txt'%flags.CODIGOS_ESTADOS[uf.upper()], - ) + if isinstance(uf, basestring): + try: + uf = int(uf) + except ValueError: + uf = flags.CODIGOS_ESTADOS[uf.upper()] + + caminho_arquivo = os.path.join(CAMINHO_MUNICIPIOS, 'MunIBGE-UF%s.txt' % uf) # Carrega o conteudo do arquivo fp = codecs.open(caminho_arquivo, "r", "utf-8-sig")