From b5d975676630fc0dbba2b745fe55e4e2ec42813e Mon Sep 17 00:00:00 2001
From: JustIceO7 <oscarcao777@gmail.com>
Date: Fri, 31 Jan 2025 03:58:11 +0000
Subject: [PATCH] UPDATE: Added user_id into session

BUGFIX: Fixed typo
---
 web_server/blueprints/authentication.py | 6 +++++-
 web_server/blueprints/user.py           | 4 ++--
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/web_server/blueprints/authentication.py b/web_server/blueprints/authentication.py
index 383ce10..b95574f 100644
--- a/web_server/blueprints/authentication.py
+++ b/web_server/blueprints/authentication.py
@@ -3,6 +3,7 @@ from werkzeug.security import generate_password_hash, check_password_hash
 from flask_cors import cross_origin
 from database.database import Database
 from blueprints.utils import login_required, sanitizer
+from utils.user_utils import get_user_id
 from secrets import token_hex
 
 auth_bp = Blueprint("auth", __name__)
@@ -96,6 +97,8 @@ def signup():
         # Create session for new user, to avoid them having unnecessary state info
         session.clear()
         session["username"] = username
+        session["user_id"] = get_user_id(username)
+        print(f"Logged in as {username}. session: {session.get('username')}. user_id: {session.get('user_id')}", flush=True)
 
         return jsonify({
             "account_created": True,
@@ -175,7 +178,8 @@ def login():
         # Set up session to avoid having unncessary state information
         session.clear()
         session["username"] = username
-        print(f"Logged in as {username}. session: {session.get('username')}", flush=True)
+        session["user_id"] = get_user_id(username)
+        print(f"Logged in as {username}. session: {session.get('username')}. user_id: {session.get('user_id')}", flush=True)
 
         # User has been logged in, let frontend know that
         return jsonify({
diff --git a/web_server/blueprints/user.py b/web_server/blueprints/user.py
index b321a5e..f656f26 100644
--- a/web_server/blueprints/user.py
+++ b/web_server/blueprints/user.py
@@ -32,9 +32,9 @@ def user_unfollow(followed_username):
     """
     user_id = session.get("user_id")
     followed_id = get_user_id(followed_username)
-    response = unfollow(user_id, followed_id)
+    status = unfollow(user_id, followed_id)
 
-    status = True if response else False
+    status = True if status else False
     return jsonify({"status": status})
 
 @login_required