Add module docstrings to login routes for improved documentation and clarity; streamline login status check

This commit is contained in:
voldemort 2025-07-23 01:42:17 +07:00
parent c82e493ef1
commit c756361da0

View File

@ -1,3 +1,5 @@
"""Module to handle the login process."""
from flask import Blueprint, request, jsonify, session
from custom_functions.database.user_db import verify_user
@ -9,7 +11,7 @@ login_bp = Blueprint(
@login_bp.route("/login", methods=["POST"])
def login():
if request.method == "POST":
"""Handle the login process."""
data = request.get_json()
for required_field in ["username", "password"]:
if required_field not in data:
@ -23,23 +25,18 @@ def login():
"username"
].lower() # Stored securely in a signed cookie
return jsonify({"message": "Successfully logged in!"})
else:
return jsonify({"error": "Invalid username or password!"}), 401
@login_bp.route("/login/status", methods=["POST"])
def login_status():
try:
"""Check if the user is logged in."""
username = session.get("username")
if username:
return jsonify({"message": "True"})
else:
return jsonify({"message": "False"})
except:
return jsonify({"message": "False"})
return jsonify({"message": "True" if username else "False"})
@login_bp.route("/logout", methods=["POST"])
def logout():
"""Logout the user."""
session.pop("username", None)
return jsonify({"message": "Successfully logged out!"})