Adiciona suporte à requisições JSONP

Author: cauethenorio

PostmonServer.py

@@ -7,6 +7,7 @@
 from database import MongoDb as Database
 
 app_v1 = bottle.Bottle()
+jsonp_query_key = 'callback'
 
 def expired(record_date):
 	from datetime import datetime, timedelta
@@ -26,6 +27,20 @@ def _get_info_from_source(cep):
 	return info
 
 
+def format_result(result):
+	# 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)
+
+		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):
@@ -52,7 +67,7 @@ def verifica_cep(cep):
 	if result:
 
 		response.headers['Cache-Control'] = 'public, max-age=2592000'
-		return result
+		return format_result(result)
 	else:
 		response.status = '404 O CEP %s informado nao pode ser localizado' % cep
 		return
@@ -81,7 +96,7 @@ def track_pack(provider, track):
 			resposta['codigo'] = track
 			resposta['historico'] = result
 
-			return json.dumps(resposta)
+			return format_result(resposta)
 
 		except AttributeError:
 			response.status = '404 O pacote %s informado nao pode ser localizado' %track

test/postmon_test.py

@@ -1,4 +1,6 @@
 # encoding: utf-8
+import json
+import re
 import unittest
 
 import webtest
@@ -132,6 +134,27 @@ def assertCep(self, cep):
 
 		self.assertNotIn('v_date', result)
 
+
+class PostmonWebJSONPTest(PostmonWebTest):
+	'''
+	Teste de requisições JSONP no servidor do Postmon
+	'''
+
+	def setUp(self):
+		self.jsonp_query_key = PostmonServer.jsonp_query_key
+		self.jsonp_func_name = 'func_name'
+		super(PostmonWebJSONPTest, self).setUp()
+
+	def get_cep(self, cep):
+		response = self.app.get('/cep/%s?%s=%s' %
+					(cep, self.jsonp_query_key, self.jsonp_func_name))
+
+		regexp = re.compile('^%s\((.*)\);$' % self.jsonp_func_name)
+		json_data = re.findall(regexp, response.body)[0]
+
+		return json.loads(json_data)
+
+
 class PostmonV1WebTest(PostmonWebTest):
 
 	'''