diff --git a/main.py b/main.py index ee731e6..85f9dd7 100755 --- a/main.py +++ b/main.py @@ -14,6 +14,7 @@ from werkzeug.middleware.proxy_fix import ProxyFix import hashlib import requests + app = Flask(__name__) FlaskUUID(app) qrcode = QRcode(app) @@ -65,7 +66,8 @@ def generate_next_url(type, ip_branch): @app.route('/') def show_qr_and_list(): # TODO: reject direct connections to server; allow access only via proxy - get_list = requests.get('http://137.184.126.24:8080/hr_attendance_extended/public/attendance/').json() + get_list = requests.get( + 'http://137.184.126.24:8080/hr_attendance_extended/public/attendance/').json() list_ips = get_list["list_ips"] ip_branch = request.environ.get('HTTP_X_REAL_IP', request.remote_addr) if ip_branch in list_ips: