add : GOD's preshared key added.

Author: sadrasabouri

mafia.py

@@ -1,11 +1,13 @@
 from sys import argv
-from random import randrange, shuffle
+from random import randrange, shuffle, sample
 from flask import Flask, render_template, url_for, request
 from flask_httpauth import HTTPBasicAuth
 from mafia_params import *
 
 app = Flask(__name__)
 auth = HTTPBasicAuth()
+auth_GOD = HTTPBasicAuth()
+preshared_key = ""
 id = 0
 nPlayers = 0
 roles = []
@@ -26,12 +28,17 @@ def index():
     image_name = ""
     ip = str(request.remote_addr)
 
+    if id == 0:
+        print("_" * 20 + "GOD's password" + "_" * 20)
+        print(preshared_key)
+        print("_" * 54)
+    
     if ip in ip2role_index_name.keys():
         role = ip2role_index_name[ip][0]
         image_name = ip2role_index_name[ip][0] + "_" + str(ip2role_index_name[ip][1])
     else:
         if id > nPlayers:
-            return "Numbers of players out of range!"   #TODO:well defined Error Page
+            return render_template("404.html", is_farsi=True)
         role = roles[id]
         ip2role_index_name[ip] = (role, str(randrange(1, nRoles[role] + 1)), username)
         image_name = role + "_" + str(ip2role_index_name[ip][1])
@@ -47,7 +54,14 @@ def index():
                             is_farsi=True)
 
 
+@auth_GOD.verify_password
+def verify_password_god(username, password):
+    if username == "GOD" and password == preshared_key:
+        return username
+
+
 @app.route('/GOD')
+@auth_GOD.login_required
 def GOD_PAGE():
     global ip2role_index_name
     return render_template("GOD.html", ip2role_index_name=ip2role_index_name)
@@ -80,6 +94,9 @@ def help_me():
         help_me()
     roles = ordered_roles[:nPlayers]
     shuffle(roles)
+    chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ123456789!@#$%^&*()"
+    for i in range(4):
+        preshared_key += chars[randrange(0, len(chars))]
     app.run(host="0.0.0.0", \
             port=5000, \
             debug=True)