Pesquisa de CEP

cep.py

#!/usr/bin/env python
# encoding: utf-8
import re
import sys

import requests

API_CEP = 'http://api.postmon.com.br/v1/cep/{}'


def imprime_resultado(resposta):
    cidade_uf = '{}-{}'.format(resposta['cidade'], resposta['estado'])
    tamanho_maximo = len(resposta['logradouro'])
    formatador = '{:<' + str(tamanho_maximo) + '}'
    print('\n')
    print('┌─────────────┬─{}'.format('─' * tamanho_maximo), end='┐\n')
    print('│ CEP:        │ {}'.format(formatador.format(resposta['cep'])), end='│\n')
    print('├─────────────┼─{}'.format('─' * tamanho_maximo), end='┤\n')
    print('│ CIDADE:     │ {}'.format(formatador.format(cidade_uf)), end='│\n')
    print('├─────────────┼─{}'.format('─' * tamanho_maximo), end='┤\n')
    print('│ BAIRRO:     │ {}'.format(formatador.format(resposta['bairro'])), end='│\n')
    print('├─────────────┼─{}'.format('─' * tamanho_maximo), end='┤\n')
    print('│ LOGRADOURO: │ {}'.format(formatador.format(resposta['logradouro'])), end='│\n')
    print('└─────────────┴─{}'.format('─' * tamanho_maximo), end='┘\n')


def valida_cep(cep):
    pattern = '[0-9]{5}-[0-9]{3}'
    pattern2 = '[0-9]{5}[0-9]{3}'
    if not any([re.match(pattern, cep), re.match(pattern2, cep)]):
        sys.exit('CEP Invalido. Favor informar um CEP valido!')
    return cep


def pesquisa_cep(cep):
    url = API_CEP.format(cep)
    resposta = requests.get(url)
    if resposta.status_code == 404:
        sys.exit('CEP não encontrado!')

    return resposta.json()


if __name__ == '__main__':
    try:
        cep = valida_cep(sys.argv[1])
    except IndexError:
        sys.exit('Informe o CEP que deseja utilizar.')

    resposta = pesquisa_cep(cep)
    imprime_resultado(resposta)