30 lines
1.2 KiB
Python
30 lines
1.2 KiB
Python
from flask import Blueprint, render_template, request, redirect, url_for
|
|
from flask_login import login_required, current_user
|
|
from ..extensions import db
|
|
from ..models import User, Profile, Notification
|
|
|
|
bp = Blueprint("users", __name__, url_prefix="/users")
|
|
|
|
@bp.route("/<int:user_id>")
|
|
def profile(user_id):
|
|
user = User.query.get_or_404(user_id)
|
|
return render_template("users/profile.html", user=user)
|
|
|
|
@bp.route("/me/edit", methods=["GET", "POST"])
|
|
@login_required
|
|
def edit_profile():
|
|
profile = Profile.query.filter_by(user_id=current_user.id).first()
|
|
if request.method == "POST":
|
|
profile.bio = request.form.get("bio")
|
|
profile.grade = request.form.get("grade")
|
|
profile.class_name = request.form.get("class_name")
|
|
db.session.commit()
|
|
return redirect(url_for("users.profile", user_id=current_user.id))
|
|
return render_template("users/edit.html", profile=profile)
|
|
|
|
@bp.route("/me/notifications")
|
|
@login_required
|
|
def notifications():
|
|
items = Notification.query.filter_by(user_id=current_user.id).order_by(Notification.created_at.desc()).limit(50).all()
|
|
return render_template("users/notifications.html", items=items)
|