added cookies for user ver0.1

Naím Rodríguez · web-flow · commit 111176c635de · 2022-10-04T23:18:03.000+02:00
added cookies to identify users in chat room's
added the function to delete the cookies to get a new username
diff --git a/server.py b/server.py
@@ -1,4 +1,4 @@
-from flask import Flask, redirect, request, render_template, url_for, send_file
+from flask import Flask, redirect, request, render_template, url_for, send_file, make_response
 import secrets
 import os
 import time
@@ -7,11 +7,44 @@
 
 app = Flask(__name__)
 app.config['SECRET_KEY'] = secrets.token_hex(32)
+user_list = []
 
 # Home page
-@app.route("/")
+@app.get("/")
 def home():
-    return render_template("home.html")
+    # If there is no cookies need to get them
+    if not request.cookies.get('cookie_user'):
+        return render_template("home-post.html")
+    # If there is already cookies on the browser return the home-get.html
+    else:
+        return render_template("home-get.html")
+
+
+# to set the cookie for the user
+@app.post("/")
+def home_post():
+    # If user has been requested
+    if request.form['user']:
+        # If user is not elected already
+        if request.form['user'] not in user_list:
+            # Create a cookie for the user
+            resp = make_response(render_template("home-get.html"))
+            # Cookie for 20 minutes
+            tok = secrets.token_urlsafe(16)
+            resp.set_cookie("cookie_user", tok, max_age=60*20)
+            
+            # add user to session
+            user_list.append(request.form['user'])
+            print(user_list)
+            # redirect to the home-get page 
+            return resp
+        # user in user_list
+        else:
+            return "usuario ya escogido"
+    # user not requested
+    else:
+        return render_template("home-post.html")
+            
 
 # Create a room
 @app.route("/create/", methods=['GET'])
@@ -22,6 +55,7 @@ def create():
     new_file_room.close()
     return redirect(f"/room/{number}")
 
+
 # GET the join.html or POST to join a room
 @app.route("/join/", methods=['GET', 'POST'])
 def join():
@@ -30,6 +64,7 @@ def join():
     if request.method == 'POST':
         return redirect(f"/room/{request.form.get('room_key')}")
 
+
 # Get the messages from the room
 @app.get("/room/<string:number>/")
 def room_get(number):
@@ -41,15 +76,18 @@ def room_get(number):
     else:
         return "no existe esa sala"
 
+
 # Post a message on the room
 @app.post("/room/<string:number>/")
 def room_post(number):
     content_room = open(f"{os.getcwd()}/chats/room.{number}.txt", "a", encoding='utf-8')
     if request.form['message']:
         time_msg = f"[{time.localtime().tm_hour}:{time.localtime().tm_min}.{time.localtime().tm_sec}]"
-        content_room.write(f"{time_msg} - {request.form['message']}\n")
+        content_room.write(f"{time_msg} - *{request.cookies.get('cookie_user')}*: {request.form['message']}\n")
     return redirect(url_for("room_get", number=number))
 
+
+# Get or generate QR code for room
 @app.get("/room/<string:number>/qr/")
 def room_get_qr(number):
     # If the qr code of the room exists return it
@@ -62,6 +100,14 @@ def room_get_qr(number):
         return send_file(f"{os.getcwd()}/qrs/{number}.png")
 
 
+# Delete your cookies
+@app.route("/del-cookie/")
+def delete_cookies():
+    resp = make_response(render_template("home-post.html"))
+    resp.set_cookie("cookie_user", "", max_age=0)
+    return resp
+
+
 if __name__ == "__main__":
     os.chdir(os.getcwd())
     app.run(host="0.0.0.0", port=80, debug=True)
diff --git a/templates/home-get.html b/templates/home-get.html
@@ -0,0 +1,10 @@
+<!DOCTYPE html>
+<head>
+    <title>Secmeet Home</title>
+</head>
+
+<body>
+    <a href="create">Create</a>
+    <a href="join">Join</a>
+    <a href="del-cookie">Delete cookies</a>
+</body>
diff --git a/templates/home-post.html b/templates/home-post.html
@@ -0,0 +1,12 @@
+<!DOCTYPE html>
+<head>
+    <title>Set your user</title>
+</head>
+
+<body>
+    <form action="/" method="post">
+        <label>User for posting messages</label>
+        <input type="text" name="user" id="user">
+        <input type="submit" value="submit">
+    </form>
+</body>
