Merge pull request #67 from avelino/master

Fix pep8

Author: aleborba

.travis.yml

@@ -2,6 +2,6 @@ language: python
 
 before_install: pip install -r requirements.txt
 
-script: nosetests
+script: make test
 
 services: mongodb

CepTracker.py

@@ -1,63 +1,65 @@
-# encoding: utf-8
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
 from datetime import datetime
 
 import requests
 import re
 
+
 class CepTracker():
 
-	def __init__(self):
-		self.url = 'http://m.correios.com.br/movel/buscaCepConfirma.do'
-
-	def _request(self, cep):
-		response = requests.post(self.url, data={
-			'cepEntrada': cep,
-			'tipoCep': '',
-			'cepTemp': '',
-			'metodo': 'buscarCep'
-		})
-		response.raise_for_status()
-		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)
-					data['cidade'] = cidade.strip()
-					data['estado'] = estado.split('-')[0].strip()
-				elif 'logradouro' in label and ' - ' in value:
-					logradouro, complemento = value.split(' - ', 1)
-					data['logradouro'] = logradouro.strip()
-					data['complemento'] = complemento.strip(' -')
-				else:
-					data[label] = value
-
-			result.append(data)
-
-		return result
+    def __init__(self):
+        self.url = 'http://m.correios.com.br/movel/buscaCepConfirma.do'
+
+    def _request(self, cep):
+        response = requests.post(self.url, data={
+            'cepEntrada': cep,
+            'tipoCep': '',
+            'cepTemp': '',
+            'metodo': 'buscarCep'
+        })
+        response.raise_for_status()
+        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)
+                    data['cidade'] = cidade.strip()
+                    data['estado'] = estado.split('-')[0].strip()
+                elif 'logradouro' in label and ' - ' in value:
+                    logradouro, complemento = value.split(' - ', 1)
+                    data['logradouro'] = logradouro.strip()
+                    data['complemento'] = complemento.strip(' -')
+                else:
+                    data[label] = value
+
+            result.append(data)
+
+        return result

Makefile

@@ -0,0 +1,8 @@
+
+.PHONY: test
+test: pep8
+	nosetests
+
+.PHONY: pep8
+pep8:
+	@flake8 * --ignore=F403,F401 --exclude=requirements.txt,*.pyc,*.md,COPYING,Makefile,*.wsgi

PostmonServer.py

@@ -1,3 +1,5 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
 import bottle
 import json
 from bottle import route, run, response
@@ -6,104 +8,111 @@
 from correios import Correios
 from database import MongoDb as Database
 
+
 app_v1 = bottle.Bottle()
 jsonp_query_key = 'callback'
 
+
 def expired(record_date):
-	from datetime import datetime, timedelta
+    from datetime import datetime, timedelta
 
-	WEEKS = 26 #6 months
+    # 6 months
+    WEEKS = 26
 
-	now = datetime.now()
+    now = datetime.now()
 
-	return ( now - record_date['v_date'] >= timedelta(weeks=WEEKS))
+    return (now - record_date['v_date'] >= timedelta(weeks=WEEKS))
 
 
 def _get_info_from_source(cep):
-	tracker = CepTracker()
-	info = tracker.track(cep)
-	if len(info) == 0:
-		raise ValueError()
-	return info
+    tracker = CepTracker()
+    info = tracker.track(cep)
+    if len(info) == 0:
+        raise ValueError()
+    return info
 
 
 def format_result(result):
-	# checa se foi solicitada resposta em JSONP
-	js_func_name = bottle.request.query.get(jsonp_query_key)
+    # checa se foi solicitada resposta em JSONP
+    js_func_name = bottle.request.query.get(jsonp_query_key)
 
-	if js_func_name:
-		# se a resposta vai ser JSONP, o content type deve ser js e seu
-		# conteudo deve ser JSON
-		response.content_type = 'application/javascript'
-		result = json.dumps(result)
+    if js_func_name:
+        # se a resposta vai ser JSONP, o content type deve ser js e seu
+        # conteudo deve ser JSON
+        response.content_type = 'application/javascript'
+        result = json.dumps(result)
 
-		result = '%s(%s);' % (js_func_name, result)
-	return result
+        result = '%s(%s);' % (js_func_name, result)
+    return result
 
 
 @route('/cep/<cep:re:\d{5}-?\d{3}>')
 @app_v1.route('/cep/<cep:re:\d{5}-?\d{3}>')
 def verifica_cep(cep):
-	cep = cep.replace('-','')
-	db = Database()
-	response.headers['Access-Control-Allow-Origin'] = '*'
-
-	result = db.get_one(cep, fields={ '_id': False })
-	if result and result.has_key('v_date') and not expired(result):
-		result.pop('v_date')
-	else:
-		try:
-			info = _get_info_from_source(cep)
-		except ValueError:
-			response.status = '404 O CEP %s informado nao pode ser localizado' % cep
-			return
-		except requests.exceptions.RequestException:
-			response.status = '503 Servico Temporariamente Indisponivel'
-			return
-		for item in info:
-			db.insert_or_update(item)
-		result = db.get_one(cep, fields={ '_id': False, 'v_date': False })
-
-	if result:
-
-		response.headers['Cache-Control'] = 'public, max-age=2592000'
-		return format_result(result)
-	else:
-		response.status = '404 O CEP %s informado nao pode ser localizado' % cep
-		return
+    cep = cep.replace('-', '')
+    db = Database()
+    response.headers['Access-Control-Allow-Origin'] = '*'
+
+    result = db.get_one(cep, fields={'_id': False})
+    if result and 'v_date' in result and not expired(result):
+        result.pop('v_date')
+    else:
+        try:
+            info = _get_info_from_source(cep)
+        except ValueError:
+            response.status = "404 O CEP {0} informado nao pode ser "
+            "localizado".format(cep)
+            return
+        except requests.exceptions.RequestException:
+            response.status = '503 Servico Temporariamente Indisponivel'
+            return
+        for item in info:
+            db.insert_or_update(item)
+        result = db.get_one(cep, fields={'_id': False, 'v_date': False})
+
+    if result:
+
+        response.headers['Cache-Control'] = 'public, max-age=2592000'
+        return format_result(result)
+    else:
+        response.status = '404 O CEP %s informado nao pode ser '
+        'localizado' % cep
+        return
 
 
 @app_v1.route('/rastreio/<provider>/<track>')
 def track_pack(provider, track):
-	if provider == 'ect':
-		try:
-			encomenda = Correios.encomenda(track)
+    if provider == 'ect':
+        try:
+            encomenda = Correios.encomenda(track)
 
-			resposta = dict()
-			result = []
+            resposta = dict()
+            result = []
 
-			for status in encomenda.status:
-				historico = dict()
-				
-				historico['data'] = status.data
-				historico['local'] = status.local
-				historico['situacao'] = status.situacao
-				historico['detalhes'] = status.detalhes
+            for status in encomenda.status:
+                historico = dict()
+                historico['data'] = status.data
+                historico['local'] = status.local
+                historico['situacao'] = status.situacao
+                historico['detalhes'] = status.detalhes
 
-				result.append(historico)
+                result.append(historico)
 
-			resposta['servico'] = provider
-			resposta['codigo'] = track
-			resposta['historico'] = result
+            resposta['servico'] = provider
+            resposta['codigo'] = track
+            resposta['historico'] = result
 
-			return format_result(resposta)
+            return format_result(resposta)
+
+        except AttributeError:
+            response.status = "404 O pacote {0} informado nao pode ser "
+            "localizado".format(track)
+    else:
+        response.status = '404 O Servico %s nao pode ser encontrado' % provider
 
-		except AttributeError:
-			response.status = '404 O pacote %s informado nao pode ser localizado' %track
-	else:
-		response.status = '404 O Servico %s nao pode ser encontrado' %provider
 
 bottle.mount('/v1', app_v1)
 
+
 def _standalone(port=9876):
     run(host='localhost', port=port)

README.md

@@ -25,7 +25,7 @@ Rodando testes
 ----------------
 Executar o comando:
 
-	nosetests
+	make test
 
 na raiz do projeto.
 

database.py

@@ -1,29 +1,32 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
 import pymongo
 
+
 class MongoDb(object):
 
-	_fields = [
-		'logradouro',
-		'bairro',
-		'cidade',
-		'estado',
-		'complemento'
-	]
+    _fields = [
+        'logradouro',
+        'bairro',
+        'cidade',
+        'estado',
+        'complemento'
+    ]
 
-	def __init__(self, address='localhost'):
-		self._client = pymongo.MongoClient(address)
-		self._db = self._client.postmon
+    def __init__(self, address='localhost'):
+        self._client = pymongo.MongoClient(address)
+        self._db = self._client.postmon
 
-	def get_one(self, cep, **kwargs):
-		return self._db.ceps.find_one({ 'cep': cep }, **kwargs)
+    def get_one(self, cep, **kwargs):
+        return self._db.ceps.find_one({'cep': cep}, **kwargs)
 
-	def insert_or_update(self, obj, **kwargs):
+    def insert_or_update(self, obj, **kwargs):
 
-		update = { '$set': obj }
-		empty_fields = set(self._fields) - set(obj)
-		update['$unset'] = dict((x, 1) for x in empty_fields)
+        update = {'$set': obj}
+        empty_fields = set(self._fields) - set(obj)
+        update['$unset'] = dict((x, 1) for x in empty_fields)
 
-		self._db.ceps.update({ 'cep': obj['cep'] }, update, upsert=True)
+        self._db.ceps.update({'cep': obj['cep']}, update, upsert=True)
 
-	def remove(self, cep):
-		self._db.ceps.remove({ 'cep': cep })
+    def remove(self, cep):
+        self._db.ceps.remove({'cep': cep})

requirements.txt

@@ -4,4 +4,5 @@ nose==1.2.1
 pymongo==2.4
 requests==0.14.2
 webtest==1.4.3
-packtrack==0.1.0
+packtrack==0.1.0
+flake8