fixed some minor bugs

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(),