Update main.py

main.py

@@ -49,6 +49,7 @@ c.execute('''CREATE TABLE IF NOT EXISTS user_details
               certification TEXT,
               activity TEXT,
               country TEXT,
+              data_filled BOOLEAN DEFAULT FALSE,
               FOREIGN KEY(user_id) REFERENCES users(id))''')
 
 
@@ -120,7 +121,7 @@ def login():
         user = c.execute("SELECT * FROM users WHERE google_id=?", (google_id,)).fetchone()
         if user:
             access_token = create_access_token(identity=user[1], expires_delta=False)
-            return jsonify({"access_token": access_token}), 200
+            return jsonify({"access_token": access_token, "data_filled": user[2]}), 200
         else:
             return jsonify({"message": "User not found"}), 404
 
@@ -131,10 +132,11 @@ def login():
     user = c.execute("SELECT * FROM users WHERE username=?", (username,)).fetchone()
     if user and check_password_hash(user[2], password):
         access_token = create_access_token(identity=username, expires_delta=False)
-        return jsonify({"access_token": access_token}), 200
+        return jsonify({"access_token": access_token, "data_filled": user[2]}), 200
     else:
         return jsonify({"message": "Invalid username or password"}), 401
 
+
 @app.route('/user_details', methods=['POST'])
 @jwt_required()
 def add_user_details():
@@ -159,9 +161,9 @@ def add_user_details():
     user_id = user[0]
     try:
         c.execute("INSERT INTO user_details (user_id, first_name, last_name, school_name, bachelors_degree, "
-                  "masters_degree, certification, activity, country) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)",
+                  "masters_degree, certification, activity, country, data_filled) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
                   (user_id, first_name, last_name, school_name, bachelors_degree, masters_degree, certification,
-                   activity, country))
+                   activity, country, True))  
         conn.commit()
         return jsonify({"message": "User details added successfully"}), 201
     except sqlite3.IntegrityError:
@@ -364,5 +366,24 @@ def select_mentor():
         return jsonify({"message": "Failed to select mentor"}), 500
 
 
+@app.route('/users', methods=['GET'])
+def get_all_users():
+    try:
+        users = c.execute("SELECT * FROM users").fetchall()
+        users_list = [{"id": user[0], "username": user[1]} for user in users]
+        return jsonify({"users": users_list}), 200
+    except Exception as e:
+        return jsonify({"message": f"Failed to retrieve users: {str(e)}"}), 500
+
+@app.route('/users', methods=['DELETE'])
+def delete_all_users():
+    try:
+        c.execute("DELETE FROM users")
+        conn.commit()
+        return jsonify({"message": "All users deleted successfully"}), 200
+    except Exception as e:
+        return jsonify({"message": f"Failed to delete users: {str(e)}"}), 500
+
+
 if __name__ == '__main__':
     app.run(debug=True)