added spam removal

web/app.py

@@ -2,6 +2,7 @@ from flask import Flask, render_template, request
 from flask_bootstrap import Bootstrap
 import db
 import config
+import utils
 
 app = Flask(__name__)
 app.config.from_object(config.BaseConfig)
@@ -19,6 +20,14 @@ def hello_world():
 
 @app.route('/generate', methods=["POST"])
 def generate():
+    try:
+        ip = utils.get_ip(request)
+        pass
+    except:
+        ip = ''
+        pass
+
+    if ip != '5.135.188.148' or ip == '178.32.58.160':
         try:
             error = None
             identifier = my_db.get_identifier(request.form['name'])
@@ -27,8 +36,17 @@ def generate():
             identifier = ''
             error = 'Maximum entry limit reached - please contact Sean or Joe'
             pass
+    else:
+        error = None
+        identifier = 'F'
+
     return render_template('generate.html', brew_name=brew_name, brew_month=brew_month, identifier=identifier, error=error)
 
 
+@app.route('/getip')
+def getip():
+    return utils.get_ip(request)
+
+
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0')

web/utils.py

@@ -4,3 +4,10 @@ def remove_common_elements(a, b):
         if e in b:
             a.remove(e)
             b.remove(e)
+
+
+def get_ip(request):
+    if request.headers.getlist("X-Forwarded-For"):
+        return request.headers.getlist("X-Forwarded-For")[0]
+    else:
+        return request.remote_addr