28 lines
982 B
Python
28 lines
982 B
Python
from flask import Blueprint, render_template, session, redirect, url_for
|
|
from models import User, Finding
|
|
|
|
profile_bp = Blueprint('profile', __name__, url_prefix='/profile')
|
|
|
|
@profile_bp.route('/')
|
|
def my_findings():
|
|
# Check if user is logged in
|
|
if not session.get('loggedin'):
|
|
return redirect(url_for('login.login'))
|
|
|
|
user_id = session.get('id')
|
|
user = User.query.get(user_id)
|
|
if not user:
|
|
return redirect(url_for('login.login'))
|
|
|
|
# Get all findings by this user, exclude content_preview
|
|
findings = Finding.query.filter_by(found_by=user_id).all()
|
|
|
|
return render_template('profile.html', user=user, findings=findings)
|
|
|
|
@profile_bp.route('/get/<int:id>', methods=['GET'])
|
|
def view_profile(id):
|
|
user = User.query.get(id)
|
|
if not user:
|
|
return "User not found. Please try again later.",
|
|
findings = Finding.query.filter_by(found_by=id).all()
|
|
return render_template('view_profile.html', user=user, findings=findings) |