From 60243520a987e19816a4d5302c1069b68efb69b2 Mon Sep 17 00:00:00 2001 From: pepperpepperpepper Date: Mon, 28 Sep 2015 23:03:43 -0700 Subject: added support for remote host --- lib/server.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'lib/server.py') diff --git a/lib/server.py b/lib/server.py index 9261730..1218531 100644 --- a/lib/server.py +++ b/lib/server.py @@ -2,7 +2,7 @@ from flask import Flask from flask import request, jsonify -import sys, os +import sys, os, re import cherrypy from paste.translogger import TransLogger @@ -39,11 +39,13 @@ class Server(object): return "HELLO WORLD!" @self.app.route('/im/api/', methods=['POST']) def pb(pb_classname): - ip = '127.0.0.1' #fixme + host = request.headers.getlist("X-Forwarded-For")[0] + regex = re.compile(r'[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+') + forwarded_ip = regex.search(host).group() return self._response_post( - pb_classname, + pb_classname, request.form.to_dict(), - remote_addr=ip + remote_addr=forwarded_ip ) @self.app.errorhandler(InvalidUsage) -- cgit v1.2.3-70-g09d2