diff options
Diffstat (limited to 'lib/server.py')
| -rw-r--r-- | lib/server.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/lib/server.py b/lib/server.py index 1218531..663fb4d 100644 --- a/lib/server.py +++ b/lib/server.py @@ -39,9 +39,13 @@ class Server(object): return "HELLO WORLD!" @self.app.route('/im/api/<pb_classname>', methods=['POST']) def pb(pb_classname): - 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() + x_forwarded_headers = request.headers.getlist("X-Forwarded-For") + if x_forwarded_headers: + host = x_forwarded_headers[0] + regex = re.compile(r'[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+') + forwarded_ip = regex.search(host).group() + else: + forwarded_ip = None return self._response_post( pb_classname, request.form.to_dict(), |