Class CepTracker

CepTracker.py:12–81

class CepTracker():

    def __init__(self):
        self.log_path = 'log.yaml'
        self._config_log_file()
        self.logger = logging.getLogger(__name__)

        self.url = 'http://m.correios.com.br/movel/buscaCepConfirma.do'

    def _config_log_file(self):
        with open(self.log_path, 'rt') as f:
            config = yaml.load(f.read())
        logging.config.dictConfig(config)

    def _request(self, cep):
        response = requests.post(self.url, data={
            'cepEntrada': cep,
            'tipoCep': '',
            'cepTemp': '',
            'metodo': 'buscarCep'
        })
        try:
            response.raise_for_status()
        except requests.exceptions.HTTPError as ex:
            logging.error('Erro request site Correios', exc_info=True)
            raise ex
        return response.text

    def _get_infos_(self, cep):
        from lxml.html import fromstring
        response = self._request(cep)
        html = fromstring(response)
        registro_csspattern = '.caixacampobranco, .caixacampoazul'
        registros = html.cssselect(registro_csspattern)

        resultado = []
        for item in registros:
            item_csspattern = '.resposta, .respostadestaque'
            resultado.append([a.text for a in item.cssselect(item_csspattern)])

        return resultado

    def track(self, cep):
        itens = self._get_infos_(cep)
        result = []

        for item in itens:

            data = dict()
            data["v_date"] = datetime.now()

            for label, value in zip(item[0::2], item[1::2]):

                label = label.lower().strip(' :')
                value = re.sub('\s+', ' ', value.strip())

                if 'localidade' in label:
                    cidade, estado = value.split('/', 1)