diff --git a/app.py b/app.py
index 09e470b2adf9de20b1b59863d7fc424c5390355e..ae8055e5f6b35b7933543643c22ab3381d9a1e72 100644
--- a/app.py
+++ b/app.py
@@ -1,5 +1,5 @@
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():
- send("hello there", broadcast = False)
+def welcomeMessage(auth):
+ if auth['token'] == "password123":
+ send("hello there", broadcast = False)
+ else:
+ disconnect()
+
@socketio.on('message')
def handleMessage(msg):
diff --git a/front.html b/front.html
index 8ef555699fdad63c2d4a222b7cb9bd5f29fa6ca9..46e9ef436b36f8fefdbfc36a1963caea948d7c4d 100644
--- 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>