add authentication

b522727a · Muhamad Nobel Fauzan · f8723ee2 · b522727a · b522727a
Commit b522727a authored 3 years ago by Muhamad Nobel Fauzan
--- a/app.py
+++ b/app.py
 from flask import Flask
-from flask_socketio import SocketIO, send
+from flask_socketio import SocketIO, send, disconnect
 from flask_sqlalchemy import SQLAlchemy
 from flask import request
 from difflib import get_close_matches
@@ -28,8 +28,12 @@ question = QAPair.query.with_entities(QAPair.question).all()
 questions = [a.question for a in question]
 @socketio.on('connect')
-def welcomeMessage():
+def welcomeMessage(auth):
-    send("hello there", broadcast = False)
+    if auth['token'] == "password123":
+        send("hello there", broadcast = False)
+    else:
+        disconnect()
 @socketio.on('message')
 def handleMessage(msg):

--- a/front.html
+++ b/front.html
@@ -7,7 +7,9 @@
    <body>
        <script type="text/javascript">
        $(document).ready(function() {
-            var socket = io.connect('http://127.0.0.1:5000');
+            let password = prompt("Please enter a password");
+            var socket = io.connect('http://127.0.0.1:5000', { auth : { token : password } });
            socket.on('message', function(msg) {
                $("#messages").append('<li>'+msg+'</li>')
@@ -18,6 +20,7 @@
                $("#messages").append('<li>'+$('#myMessage').val()+'</li>')
                $('#myMessage').val('');
            });
        });
        </script>